https://agc015.contest.atcoder.jp/tasks/agc015_c
この発想に至るまでが時間かかる気がする。
森の数=木の頂点数-木の辺数と言い換えられれば勝ちです。
あとは累積和適当に取れば通ります。
実装なんですがB,S,E,U,D,L,Rと7つも2010*2010の配列を作ってしまったことは反省するべきですね。
工夫すれば4つとかになるはずです。
int dx[4] = {1, 0, -1, 0}; int dy[4] = {0, 1, 0, -1}; int N, M, Q; int B[2010][2010]; //blocks int S[2010][2010]; //num of blocks; int E[2010][2010]; //num of edge int U[2010][2010], D[2010][2010], L[2010][2010], R[2010][2010]; void solve() { cin >> N >> M >> Q; rep(i, 0, N) { rep(j, 0, M) { char b; cin >> b; B[i][j] = (b == '1'); } } rep(i, 0, N) { rep(j, 0, M) { S[i + 1][j + 1] += B[i][j] + S[i][j + 1] + S[i + 1][j] - S[i][j]; int e = 0; rep(k, 0, 4) { int nx = i + dx[k], ny = j + dy[k]; if(0 <= nx && nx < N && 0 <= ny && ny < M) { e += B[nx][ny] * B[i][j]; } } E[i + 1][j + 1] += e + E[i][j + 1] + E[i + 1][j] - E[i][j]; } } rep(i, 0, N) { rep(j, 0, M) { int u = 0, d = 0; if(i != 0) { u = (B[i][j] && B[i - 1][j]); } if(i != N - 1) { d = (B[i][j] && B[i + 1][j]); } U[i][j + 1] += U[i][j] + u; D[i][j + 1] += D[i][j] + d; } } rep(j, 0, M) { rep(i, 0, N) { int l = 0, r = 0; if(j != 0) { l = (B[i][j] && B[i][j - 1]); } if(j != M - 1) { r = (B[i][j] && B[i][j + 1]); } L[i + 1][j] += L[i][j] + l; R[i + 1][j] += R[i][j] + r; } } while(Q--) { int a, b, c, d; cin >> a >> b >> c >> d; a--; b--; c--; d--; int e = (E[c + 1][d + 1] - E[a][d + 1] - E[c + 1][b] + E[a][b]); e -= U[a][d + 1] - U[a][b]; e -= D[c][d + 1] - D[c][b]; e -= L[c + 1][b] - L[a][b]; e -= R[c + 1][d] - R[a][d]; cout << (S[c + 1][d + 1] - S[a][d + 1] - S[c + 1][b] + S[a][b]) - e / 2 << "\n"; } }