#include #include #include #include #include #include #include // 内部定数 #define D_SIZE_MAX 205 // 最大サイズ #define D_ON 1 // 汎用フラグ - ON #define D_OFF 0 // 汎用フラグ - OFF // 内部構造体 - マス情報 typedef struct Mass { int miColor; // 色 int miDx, miDy; // 代表マス int miDone; // 処理済フラグ } Mass; // 内部変数 static FILE *szpFpI; // 入力 static Mass sz2Mass[D_SIZE_MAX][D_SIZE_MAX]; // マス static int siW, siH; // 幅・高さ // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; #endif // 代表マス - 取得 int fGetDMass( int piTx // 対象マスX , int piTy // 対象マスY , int *pipDx // 代表マスX , int *pipDy // 代表マスY ) { // 自分が代表かチェック if (sz2Mass[piTx][piTy].miDx == piTx) { if (sz2Mass[piTx][piTy].miDy == piTy) { *pipDx = piTx; *pipDy = piTy; return 0; } } // 代表マス - 取得 fGetDMass(sz2Mass[piTx][piTy].miDx, sz2Mass[piTx][piTy].miDy, pipDx, pipDy); // 代表マス - 更新 sz2Mass[piTx][piTy].miDx = *pipDx; sz2Mass[piTx][piTy].miDy = *pipDy; return 0; } // 同一グループ化 int fSameGrp( int piX // 位置X , int piY // 位置Y ) { // 代表マス - 取得 int liDx, liDy; fGetDMass(piX, piY, &liDx, &liDy); // 同色チェック - 上下左右 if (sz2Mass[piX][piY].miColor == sz2Mass[piX + 1][piY].miColor) { sz2Mass[piX + 1][piY].miDx = liDx; sz2Mass[piX + 1][piY].miDy = liDy; } if (sz2Mass[piX][piY].miColor == sz2Mass[piX - 1][piY].miColor) { sz2Mass[piX - 1][piY].miDx = liDx; sz2Mass[piX - 1][piY].miDy = liDy; } if (sz2Mass[piX][piY].miColor == sz2Mass[piX][piY + 1].miColor) { sz2Mass[piX][piY + 1].miDx = liDx; sz2Mass[piX][piY + 1].miDy = liDy; } if (sz2Mass[piX][piY].miColor == sz2Mass[piX][piY - 1].miColor) { sz2Mass[piX][piY - 1].miDx = liDx; sz2Mass[piX][piY - 1].miDy = liDy; } return 0; } // 色セット int fSetColor( int piTx // 対象位置X , int piTy // 対象位置Y , int *pipDx // 代表マスX , int *pipDy // 代表マスY , int piColor // 色 ) { // 処理済 - チェック if (sz2Mass[piTx][piTy].miDone == D_ON) { return 0; } // 代表マス - 取得 int liDx, liDy; fGetDMass(piTx, piTy, &liDx, &liDy); // 代表マス - チェック if (liDx == *pipDx && liDy == *pipDy) { // 一致 // 色チェック if (sz2Mass[piTx][piTy].miColor == piColor) { // セット済 return 0; } sz2Mass[piTx][piTy].miColor = piColor; // セット // 上下左右へ fSetColor(piTx + 1, piTy, pipDx, pipDy, piColor); fSetColor(piTx - 1, piTy, pipDx, pipDy, piColor); fSetColor(piTx, piTy + 1, pipDx, pipDy, piColor); fSetColor(piTx, piTy - 1, pipDx, pipDy, piColor); // 処理済 - セット sz2Mass[piTx][piTy].miDone = D_ON; } else { // 不一致 // 色チェック if (sz2Mass[piTx][piTy].miColor != piColor) { // 範囲外 return 0; } // 代表マス - 更新 sz2Mass[*pipDx][*pipDy].miDx = liDx; sz2Mass[*pipDx][*pipDy].miDy = liDy; *pipDx = liDx; *pipDy = liDy; } return 0; } // 実行メイン int fMain( int piTNo // テスト番号 1~ ) { int i, j; char lc1Buf[1024], lc1Out[1024]; // データ - 初期化 memset(sz2Mass, 0, sizeof(sz2Mass)); // マス // 入力 - セット #ifdef D_TEST sprintf(lc1Buf, ".\\Test\\T%d.txt", piTNo); szpFpI = fopen(lc1Buf, "r"); sprintf(lc1Buf, ".\\Test\\A%d.txt", piTNo); szpFpA = fopen(lc1Buf, "r"); siRes = 0; #else szpFpI = stdin; #endif // 幅・高さ - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &siH, &siW); // マス - 初期化 for (i = 0; i <= siW + 1; i++) { for (j = 0; j <= siH + 1; j++) { sz2Mass[i][j].miColor = -1; sz2Mass[i][j].miDx = i; sz2Mass[i][j].miDy = j; } } // 色 - 取得 for (j = 1; j <= siH; j++) { for (i = 1; i <= siW; i++) { fscanf(szpFpI, "%d", &sz2Mass[i][j].miColor); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); } // 同一グループ化 for (i = 1; i <= siW; i++) { for (j = 1; j <= siH; j++) { fSameGrp(i, j); } } // クエリ数 - 取得 int liQCnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &liQCnt); // クエリ - 取得 for (i = 0; i < liQCnt; i++) { int liX, liY, liColor; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &liY, &liX, &liColor); // 代表マス - 取得 int liDx, liDy; fGetDMass(liX, liY, &liDx, &liDy); // 同色チェック if (sz2Mass[liDx][liDy].miColor == liColor) { continue; } // 色セット fSetColor(liDx, liDy, &liDx, &liDy, liColor); } // 色 - 更新 for (i = 1; i <= siW; i++) { for (j = 1; j <= siH; j++) { // 代表マス - 取得 int liDx, liDy; fGetDMass(i, j, &liDx, &liDy); // 色 - 更新 sz2Mass[i][j].miColor = sz2Mass[liDx][liDy].miColor; } } // 結果 for (j = 1; j <= siH; j++) { // 結果 - セット sprintf(lc1Out, "%d", sz2Mass[1][j].miColor); for (i = 2; i <= siW; i++) { sprintf(lc1Buf, " %d", sz2Mass[i][j].miColor); strcat(lc1Out, lc1Buf); } strcat(lc1Out, "\n"); // 結果 - 表示 #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, lc1Out)) { siRes = -1; } #else printf("%s", lc1Out); #endif } // 残データ有無 #ifdef D_TEST lc1Buf[0] = '\0'; fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, "")) { siRes = -1; } #endif // テストファイルクローズ #ifdef D_TEST fclose(szpFpI); fclose(szpFpA); #endif // テスト結果 #ifdef D_TEST if (siRes == 0) { printf("OK %d\n", piTNo); } else { printf("NG %d\n", piTNo); } #endif return 0; } int main() { #ifdef D_TEST int i; for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) { fMain(i); } #else fMain(0); #endif return 0; }