C++の練習を兼ねて, AtCoder Grand Contest 015 の 問題C (Nuske vs Phantom Thnook) を解いてみた.
■感想.
1. 問題C は, 解答方針が, 全く見えなかったので, 解答を参照して, 実装して何とかAC版となった.
2. 辺数に関する 二次元累積和 の 参照方法 が 独特であり, 実装時に, 苦労した.
3. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイトAGC 015 解説をご覧下さい.
■C++版プログラム(問題C／AC版).
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; #define repex(i, a, b, c) for(int i = a; i < b; i += c) #define repx(i, a, b) repex(i, a, b, 1) #define rep(i, n) repx(i, 0, n) #define repr(i, a, b) for(int i = a; i >= b; i--) const int MAX = 2020; int board[MAX][MAX], bCUM[MAX][MAX], rCUM[MAX][MAX], cCUM[MAX][MAX]; int main(){ // 1. 入力情報. int N, M, Q; scanf("%d %d %d", &N, &M, &Q); rep(i, N){ char S[2020]; scanf("%s", S); rep(j, M) board[i][j] = S[j] - '0'; } // 2. 二次元累積和(黒マス). repx(i, 1, N + 1){ repx(j, 1, M + 1){ bCUM[i][j] = board[i - 1][j - 1] + bCUM[i - 1][j] + bCUM[i][j - 1] - bCUM[i - 1][j - 1]; } } // repx(i, 0, N + 1){ // repx(j, 0, M + 1) printf("%d ", bCUM[i][j]); // printf("\n"); // } // 3. 二次元累積和(黒マスが縦方向に連続する箇所の数). repx(i, 1, N + 1){ repx(j, 1, M + 1){ cCUM[i][j] = cCUM[i - 1][j] + cCUM[i][j - 1] - cCUM[i - 1][j - 1]; if(i > 1 && board[i - 2][j - 1] && board[i - 1][j - 1]) cCUM[i][j]++; } } // repx(i, 0, N + 1){ // repx(j, 0, M + 1) printf("%d ", cCUM[i][j]); // printf("\n"); // } // 4. 二次元累積和(黒マスが横方向に連続する箇所の数). repx(i, 1, N + 1){ repx(j, 1, M + 1){ rCUM[i][j] = rCUM[i - 1][j] + rCUM[i][j - 1] - rCUM[i - 1][j - 1]; if(j > 1 && board[i - 1][j - 2] && board[i - 1][j - 1]) rCUM[i][j]++; } } // repx(i, 0, N + 1){ // repx(j, 0, M + 1) printf("%d ", rCUM[i][j]); // printf("\n"); // } // 5. クエリ回答. rep(i, Q){ // 5-1. 矩形領域. int x1, y1, x2, y2; scanf("%d %d %d %d", &x1, &y1, &x2, &y2); // 5-2. 黒マスの数. int vCount = bCUM[x2][y2] - bCUM[x1 - 1][y2] - bCUM[x2][y1 - 1] + bCUM[x1 - 1][y1 - 1]; // 5-3. 辺数. int eCount = 0; eCount += cCUM[x2][y2] - cCUM[x1][y2] - cCUM[x2][y1 - 1] + cCUM[x1][y1 - 1]; eCount += rCUM[x2][y2] - rCUM[x1 - 1][y2] - rCUM[x2][y1] + rCUM[x1 - 1][y1]; // 5-4. 出力. int ans = vCount - eCount; printf("%d\n", ans); } return 0; } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
[入力例] 3 4 4 1101 0110 1101 1 1 3 4 1 1 3 1 2 2 3 4 1 2 2 4 [出力例] 3 2 2 2 ※AtCoderテストケースより [入力例] 5 5 6 11010 01110 10101 11101 01010 1 1 5 5 1 2 4 5 2 3 3 4 3 3 3 3 3 1 3 5 1 1 3 4 [出力例] 3 2 1 1 3 2 ※AtCoderテストケースより [入力例] 12 10 7 1011010100 0101111010 1001001001 1111010010 0010100101 0001010000 1101010111 1000100111 1000110110 0110001000 1011000010 1000111101 1 2 5 7 2 3 7 8 1 1 12 10 4 5 6 7 3 5 8 9 7 10 10 10 5 7 7 9 [出力例] 4 7 19 3 7 1 2 |
■参照サイト
AtCoder Grand Contest 015