#include #include #include #include #include #include #include // 内部定数 #define D_ICE_MAX 100 // 最大アイス数 // 内部変数 static FILE *szpFpI; // 入力 static int siICnt; // 1箱のアイス数 static int siNCnt; // 必要なアイス数 static char sc1Ice[D_ICE_MAX]; // アイスの当たりはずれ static int si1ICnt[D_ICE_MAX]; // アイス数[位置] // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; #endif // アイス数[位置] - セット int fSetICnt( int piINo // 位置 0~ ) { int liRet; // 無限 if (piINo >= siICnt * 2) { return -1; } // 本数 int liCnt = 1; // 1本当たり if (sc1Ice[piINo] > '0') { liRet = fSetICnt(piINo + liCnt); if (liRet != 0) { si1ICnt[piINo] = siNCnt; return -1; } // 本数追加 liCnt += si1ICnt[piINo + liCnt]; } // 2本当たり if (sc1Ice[piINo] > '1') { liRet = fSetICnt(piINo + liCnt); if (liRet != 0) { si1ICnt[piINo] = siNCnt; return -1; } // 本数追加 liCnt += si1ICnt[piINo + liCnt]; } // 本数 - セット si1ICnt[piINo] = liCnt; return 0; } // 実行メイン int fMain( int piTNo // テスト番号 1~ ) { int liRet; char lc1Buf[1024], lc1Out[1024]; // データ - 初期化 memset(si1ICnt, 0, sizeof(si1ICnt)); // アイス数[位置] // 入力 - セット #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 // 1箱のアイス数・必要なアイス数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &siICnt, &siNCnt); // アイスの当たりはずれ - 取得 fgets(sc1Ice, sizeof(sc1Ice), szpFpI); memcpy(&sc1Ice[siICnt], sc1Ice, siICnt); // アイス数[位置] - セット int liINo = 0; int liSNo; while (liINo < siICnt) { liRet = fSetICnt(liINo); if (liRet != 0) { liSNo = -1; break; } // 次の位置 liSNo = liINo; liINo += si1ICnt[liINo]; } // 周期 - セット int liISum = 0; int liBSum = 0; if (liSNo >= 0) { liINo = liSNo; do { liISum += si1ICnt[liINo]; liBSum++; liINo += si1ICnt[liINo]; liINo %= siICnt; } while (liINo != liSNo); } // 購入 int liICnt = 0; int liBCnt = 0; liINo = 0; while (liICnt < siNCnt) { // 周期 if (liINo == liSNo) { int liUCnt = (siNCnt - liICnt) / liISum; liICnt += liISum * liUCnt; liBCnt += liBSum * liUCnt; if (liICnt >= siNCnt) { break; } } // 1本購入 liICnt += si1ICnt[liINo]; liBCnt++; liINo += si1ICnt[liINo]; liINo %= siICnt; } // 結果 - セット sprintf(lc1Out, "%d\n", liBCnt); // 結果 - 表示 #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; }