#include #include #include #include #include #include // 内部定数 #define D_ON 1 // 汎用フラグ - ON #define D_OFF 0 // 汎用フラグ - OFF #define D_TOWN_MAX 100 // 最大町数 #define D_ROAD_MAX 1000 // 最大道路数 // 内部構造体 - 道路情報 typedef struct Road { int miTNo1; // 町1 int miTNo2; // 町2 int miPay; // 通行料金 } Road; // 内部変数 static FILE *szpFpI; // 入力 static int siTCnt; // 町数 static Road sz1Road[D_ROAD_MAX]; // 道路 static int siRCnt; // 道路数 static char sc1Exist[D_TOWN_MAX + 5]; // 犯人存在フラグ // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; #endif // 実行メイン int fMain( int piTNo // テスト番号 1~ ) { int i, j; char lc1Buf[1024], lc1Out[1024]; // データ - 初期化 memset(sc1Exist, D_ON, sizeof(sc1Exist)); // 犯人存在フラグ // 入力 - セット #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 // 町数・道路数・支払回数 - 取得 int liPCnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &siTCnt, &siRCnt, &liPCnt); // 道路 - 取得 for (i = 0; i < siRCnt; i++) { fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &sz1Road[i].miTNo1, &sz1Road[i].miTNo2, &sz1Road[i].miPay); } // 支払 - 取得 for (i = 0; i < liPCnt; i++) { int liPay; fscanf(szpFpI, "%d", &liPay); // ワーク - クリア char lc1Exist[D_TOWN_MAX + 5]; memset(lc1Exist, D_OFF, sizeof(lc1Exist)); // 道路数でループ for (j = 0; j < siRCnt; j++) { // 対象チェック if (sz1Road[j].miPay != liPay) { continue; } // 町1 → 町2 if (sc1Exist[sz1Road[j].miTNo1] == D_ON) { lc1Exist[sz1Road[j].miTNo2] = D_ON; } // 町2 → 町1 if (sc1Exist[sz1Road[j].miTNo2] == D_ON) { lc1Exist[sz1Road[j].miTNo1] = D_ON; } } // ワーク - コピー memcpy(sc1Exist, lc1Exist, sizeof(sc1Exist)); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); // 存在数・最小位置 int liECnt = 0; int liTNo; for (i = siTCnt; i >= 1; i--) { if (sc1Exist[i] == D_ON) { liECnt++; liTNo = i; } } sprintf(lc1Out, "%d\n", liECnt); #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, lc1Out)) { siRes = -1; } #else printf("%s", lc1Out); #endif // 存在町 sprintf(lc1Out, "%d", liTNo); for (i = liTNo + 1; i <= siTCnt; i++) { if (sc1Exist[i] == D_ON) { sprintf(lc1Buf, " %d", i); 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; }