RUPC2016 Day2 B SEARIGHT LIVE FES
考察
外側の角の部分から上,横,下に分けて考える。
外側の人が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()); } }