RUPC2016 Day2 B SEARIGHT LIVE FES

問題

http://judge.u-aizu.ac.jp/onlinejudge/cdescription.jsp?cid=RitsCamp16Day2&pid=B

考察

外側の角の部分から上,横,下に分けて考える。

外側の人がa番目に存在する
内側の人がb番目に存在する

以上を前提とすると

• 外側の人が上(W-1番目まで)に存在する場合,a == bで判定

• 外側の人が横(W+1番目からW+H-2番目まで)に存在する場合,a-2 == bで判定

• 外側の人が下(W+H-2+1番目からW*2+H-2番目まで)に存在する場合,a-4 == bで判定

ソースコード

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.NoSuchElementException;

public class Main{
	int W,H,N,a,b;
	public void solve() {
		W = nextInt();
		H = nextInt();
		N = nextInt();

		int ans = 0;
		a = 1;
		b = 1;
		for(int i = 0;i < N;i++){
			int p = nextInt();
			if(p==0)a++;
			else b++;

			if(a <= W-1){
				if(a == b)ans++;
			}else if(a == W){
				continue;
			}else if(W + 1 <= a && a <= W + H - 2){
				if(a - 2 == b)ans++;
			}else if(a == W + H - 2 + 1){
				continue;
			}else if(a >= W + H){
				if(a - 4 == b)ans++;
			}
		}
		out.println(ans);
	}

	public static void main(String[] args) {
		out.flush();
		new Main().solve();
		out.close();
	}

	/* Input */
	private static final InputStream in = System.in;
	private static final PrintWriter out = new PrintWriter(System.out);
	private final byte[] buffer = new byte[2048];
	private int p = 0;
	private int buflen = 0;

	private boolean hasNextByte() {
		if (p < buflen)
			return true;
		p = 0;
		try {
			buflen = in.read(buffer);
		} catch (IOException e) {
			e.printStackTrace();
		}
		if (buflen <= 0)
			return false;
		return true;
	}

	public boolean hasNext() {
		while (hasNextByte() && !isPrint(buffer[p])) {
			p++;
		}
		return hasNextByte();
	}

	private boolean isPrint(int ch) {
		if (ch >= '!' && ch <= '~')
			return true;
		return false;
	}

	private int nextByte() {
		if (!hasNextByte())
			return -1;
		return buffer[p++];
	}

	public String next() {
		if (!hasNext())
			throw new NoSuchElementException();
		StringBuilder sb = new StringBuilder();
		int b = -1;
		while (isPrint((b = nextByte()))) {
			sb.appendCodePoint(b);
		}
		return sb.toString();
	}

	public int nextInt() {
		return Integer.parseInt(next());
	}

	public long nextLong() {
		return Long.parseLong(next());
	}

	public double nextDouble() {
		return Double.parseDouble(next());
	}
}