#include #include #include #include #include #include #include // 内部定数 #define D_ARRAY_MAX 2005 // 最大配列数 #define D_NUM_KIND 2 // 数値種類 #define D_PTN_MAX 5 // 最大パターン数 #define D_NUM_CNT 3 // 1パターンの数字数 // 内部変数 static FILE *szpFpI; // 入力 static int si1Array2[D_ARRAY_MAX]; // 2段目 static int siACnt; // 配列数 static int si3Ptn[D_NUM_KIND][D_PTN_MAX][D_NUM_CNT]; // 1段目のパターン[数値種類][パターン数][数字] static int si1PCnt[D_NUM_KIND]; // 1段目のパターン数 static int si2Array1[D_ARRAY_MAX][D_ARRAY_MAX]; // 1段目[チェック位置] static int siKCnt; // 1段目の種類数 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; static int siTNo; #endif // 出力 int fOut( char *pcpLine // 1行 ) { char lc1Buf[1024]; #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, pcpLine)) { siRes = -1; } #else printf("%s", pcpLine); #endif return 0; } // パターン数 - セット int fSetPCnt( int piANo // チェック位置 1~ ) { int i, j; // チェック中 if (piANo <= siACnt) { int liVal2 = si1Array2[piANo]; for (i = 0; i < si1PCnt[liVal2]; i++) { // 一致チェック for (j = 0; j < D_NUM_CNT; j++) { int liVal1 = si2Array1[piANo - 1][piANo + j - 1]; if (liVal1 >= 0 && liVal1 != si3Ptn[liVal2][i][j]) { break; } } if (j < D_NUM_CNT) { continue; } // セット memcpy(si2Array1[piANo], si2Array1[piANo - 1], sizeof(si2Array1[0])); for (j = 0; j < D_NUM_CNT; j++) { si2Array1[piANo][piANo + j - 1] = si3Ptn[liVal2][i][j]; } // 下位へ fSetPCnt(piANo + 1); } return 0; } // 両端のチェック if (si2Array1[siACnt][0] != si2Array1[siACnt][siACnt]) { return 0; } if (si2Array1[siACnt][1] != si2Array1[siACnt][siACnt + 1]) { return 0; } // 1段目の種類数 siKCnt++; return 0; } // 実行メイン int fMain( ) { int i; char lc1Buf[1024]; // 配列数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &siACnt); // 2段目 - 取得 for (i = 1; i <= siACnt; i++) { fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &si1Array2[i]); } // 1段目のパターン - セット - 0 si3Ptn[0][0][0] = 0; si3Ptn[0][0][1] = 0; si3Ptn[0][0][2] = 0; si3Ptn[0][1][0] = 1; si3Ptn[0][1][1] = 0; si3Ptn[0][1][2] = 0; si3Ptn[0][2][0] = 1; si3Ptn[0][2][1] = 1; si3Ptn[0][2][2] = 1; si1PCnt[0] = 3; // 1段目のパターン - セット - 1 si3Ptn[1][0][0] = 0; si3Ptn[1][0][1] = 0; si3Ptn[1][0][2] = 1; si3Ptn[1][1][0] = 0; si3Ptn[1][1][1] = 1; si3Ptn[1][1][2] = 0; si3Ptn[1][2][0] = 0; si3Ptn[1][2][1] = 1; si3Ptn[1][2][2] = 1; si3Ptn[1][3][0] = 1; si3Ptn[1][3][1] = 0; si3Ptn[1][3][2] = 1; si3Ptn[1][4][0] = 1; si3Ptn[1][4][1] = 1; si3Ptn[1][4][2] = 0; si1PCnt[1] = 5; // パターン数 - セット fSetPCnt(1); return 0; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(si2Array1, -1, sizeof(si2Array1)); // 1段目 siKCnt = 0; // 1段目の種類数 // 入力 - セット #ifdef D_TEST sprintf(lc1Buf, ".\\Test\\T%d.txt", siTNo); szpFpI = fopen(lc1Buf, "r"); sprintf(lc1Buf, ".\\Test\\A%d.txt", siTNo); szpFpA = fopen(lc1Buf, "r"); siRes = 0; #else szpFpI = stdin; #endif // 実行メイン liRet = fMain(); // 出力 sprintf(lc1Buf, "%d\n", siKCnt); fOut(lc1Buf); // 残データ有無 #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", siTNo); } else { printf("NG %d\n", siTNo); } #endif return 0; } // プログラム開始 int main() { #ifdef D_TEST int i; for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) { siTNo = i; fOne(); } #else fOne(); #endif return 0; }