#include #include #include #include #include #include #include // 内部定数 #define D_STA_MAX 200 // 最大駅数 #define D_HEAP_MAX D_STA_MAX * D_STA_MAX // 最大ヒープ数 // 内部構造体 - ヒープ情報 typedef struct Heap { int miTime; // 移動時間 int mi1SNo[D_STA_MAX]; // 通過駅 int miSCnt; // 通過駅数 } Heap; // 内部構造体 - 駅情報 typedef struct Sta { int mi1Time[D_STA_MAX]; // 移動時間[駅] Heap mzMin; // 最短データ } Sta; // 内部変数 static FILE *szpFpI; // 入力 static Sta sz1Sta[D_STA_MAX]; // 駅 static int siSCnt; // 駅数 static Heap sz1Heap[D_HEAP_MAX]; // ヒープ static int siHCnt; // ヒープ数 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; #endif // ヒープ - 比較 - 移動時間昇順 - 通過駅昇順 int fHeapCmp( Heap *pzpHeap1 // データ1 , Heap *pzpHeap2 // データ2 ) { // 移動時間昇順 if (pzpHeap1->miTime < pzpHeap2->miTime) { return -1; } else if (pzpHeap1->miTime > pzpHeap2->miTime) { return 1; } // 通過駅昇順 for (int liNo = 0; ; liNo++) { if (pzpHeap1->miSCnt <= liNo) { return -1; } else if (pzpHeap2->miSCnt <= liNo) { return 1; } if (pzpHeap1->mi1SNo[liNo] < pzpHeap2->mi1SNo[liNo]) { return -1; } else if (pzpHeap1->mi1SNo[liNo] > pzpHeap2->mi1SNo[liNo]) { return 1; } } return 0; } // ヒープ - 親子関係チェック // 戻り値:[>=0]:変更した子の配列番号 [-1]:変更なし int fHeapChk( int piPNo // 親の配列番号 0~ ) { int liRet; // 最小値 int liMNo = piPNo; // 左の子と比較 int liCNo = piPNo * 2 + 1; if (liCNo < siHCnt) { liRet = fHeapCmp(&sz1Heap[liMNo], &sz1Heap[liCNo]); if (liRet == 1) { liMNo = liCNo; } } // 右の子と比較 liCNo = piPNo * 2 + 2; if (liCNo < siHCnt) { liRet = fHeapCmp(&sz1Heap[liMNo], &sz1Heap[liCNo]); if (liRet == 1) { liMNo = liCNo; } } // 自分が最小値であるかチェック if (piPNo == liMNo) { return -1; } // 値の交換 Heap lzWork; memcpy(&lzWork, &sz1Heap[liMNo], sizeof(Heap)); memcpy(&sz1Heap[liMNo], &sz1Heap[piPNo], sizeof(Heap)); memcpy(&sz1Heap[piPNo], &lzWork, sizeof(Heap)); return liMNo; } // ヒープ - キュー追加 int fHeapEnqueue( Heap *pzpHeap // データ ) { int liRet; // 現在駅 int liSNo = pzpHeap->mi1SNo[pzpHeap->miSCnt - 1]; // 最短データ liRet = fHeapCmp(pzpHeap, &sz1Sta[liSNo].mzMin); if (liRet >= 0) { return 0; } memcpy(&sz1Sta[liSNo].mzMin, pzpHeap, sizeof(Heap)); // 末尾に追加 memcpy(&sz1Heap[siHCnt], pzpHeap, sizeof(Heap)); siHCnt++; // 親子関係チェック int liNo = siHCnt - 1; while (1) { // 親の配列番号 liNo = (liNo - 1) / 2; // 親子関係チェック liRet = fHeapChk(liNo); if (liRet < 0) { break; } } return 0; } // ヒープ - キュー取得 int fHeapDequeue( Heap *pzpRet // 取得先 ) { // データ数 if (siHCnt < 1) { return -1; } // 取得 memcpy(pzpRet, &sz1Heap[0], sizeof(Heap)); siHCnt--; // データ数 if (siHCnt < 1) { return 0; } // 末尾を先頭へ memcpy(&sz1Heap[0], &sz1Heap[siHCnt], sizeof(Heap)); // 親子関係チェック int liNo = 0; while (liNo >= 0) { liNo = fHeapChk(liNo); } return 0; } // 実行メイン int fMain( int piTNo // テスト番号 1~ ) { int i, liRet; char lc1Buf[1024], lc1Out[1024]; // データ - 初期化 memset(sz1Sta, 0, sizeof(sz1Sta)); // 駅 // 入力 - セット #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 liRCnt, liStart, liGoal; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d%d", &siSCnt, &liRCnt, &liStart, &liGoal); // 最短時間 - 初期化 for (i = 0; i < siSCnt; i++) { sz1Sta[i].mzMin.miTime = INT_MAX; } // 線路 - 取得 for (i = 0; i < liRCnt; i++) { int liSNo1, liSNo2, liTime; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &liSNo1, &liSNo2, &liTime); // 移動時間 - セット sz1Sta[liSNo1].mi1Time[liSNo2] = liTime; sz1Sta[liSNo2].mi1Time[liSNo1] = liTime; } // 出発駅 - 登録 Heap lzHeap; memset(&lzHeap, 0, sizeof(lzHeap)); lzHeap.mi1SNo[0] = liStart; lzHeap.miSCnt = 1; fHeapEnqueue(&lzHeap); // 算出 while (1) { // キュー取得 liRet = fHeapDequeue(&lzHeap); if (liRet != 0) { break; } // 現在駅 int liSNo = lzHeap.mi1SNo[lzHeap.miSCnt - 1]; // 隣接駅へ for (i = 0; i < siSCnt; i++) { int liTime = sz1Sta[liSNo].mi1Time[i]; if (liTime < 1) { continue; } // データコピー Heap lzHeap2; memcpy(&lzHeap2, &lzHeap, sizeof(lzHeap2)); // 隣接駅 - 追加 lzHeap2.mi1SNo[lzHeap2.miSCnt] = i; lzHeap2.miSCnt++; lzHeap2.miTime += liTime; // 登録 fHeapEnqueue(&lzHeap2); } } // 結果 - セット memcpy(&lzHeap, &sz1Sta[liGoal].mzMin, sizeof(lzHeap)); sprintf(lc1Out, "%d", lzHeap.mi1SNo[0]); for (i = 1; i < lzHeap.miSCnt; i++) { sprintf(lc1Buf, " %d", lzHeap.mi1SNo[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; }