#include #include #include #include #include #include #include // 内部定数 #define D_VTX_MAX 2000 // 最大頂点数 #define D_EDGE_MAX 2000 // 最大辺数 #define D_HEAP_MAX 10000 // 最大ヒープ数 // 内部構造体 - 辺情報 typedef struct Edge { int miVNo; // 接続先頂点 int miLen; // 距離 struct Edge *mzpNext; // 次の辺情報 } Edge; // 内部構造体 - 頂点情報 typedef struct Vtx { Edge *mzpEdge; // 辺 int miTCnt; // トラック数 int mi1Min[D_VTX_MAX]; // 最短距離 } Vtx; // 内部構造体 - ヒープ情報 typedef struct Heap { int miVNo; // 頂点 int miLen; // 距離 } Heap; // 内部変数 static FILE *szpFpI; // 入力 static Vtx sz1Vtx[D_VTX_MAX]; // 頂点 static int siVCnt; // 頂点数 static Edge sz1Edge[D_EDGE_MAX * 2]; // 辺 static int siECnt; // 辺数 static int siWNo; // レッカー車 0~ static Heap sz1Heap[D_HEAP_MAX]; // ヒープ static int siHCnt; // ヒープ数 // 内部変数 - テスト用 #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 fAddEdge( int piVFNo // 頂点 - 元 0~ , int piVTNo // 頂点 - 先 0~ , int piLen // 距離 ) { sz1Edge[siECnt].miVNo = piVTNo; sz1Edge[siECnt].miLen = piLen; sz1Edge[siECnt].mzpNext = sz1Vtx[piVFNo].mzpEdge; sz1Vtx[piVFNo].mzpEdge = &sz1Edge[siECnt]; siECnt++; return 0; } // ヒープ - 比較 - 距離昇順 int fHeapCmp( int piNo1 // 配列番号1 0~ , int piNo2 // 配列番号2 0~ ) { // 距離昇順 if (sz1Heap[piNo1].miLen < sz1Heap[piNo2].miLen) { return -1; } else if (sz1Heap[piNo1].miLen > sz1Heap[piNo2].miLen) { 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(liMNo, liCNo); if (liRet == 1) { liMNo = liCNo; } } // 右の子と比較 liCNo = piPNo * 2 + 2; if (liCNo < siHCnt) { liRet = fHeapCmp(liMNo, 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( int piVSNo // 開始頂点 0~ , int piVNNo // 現在 0~ , int piLen // 距離 ) { int liRet; // セット済 if (sz1Vtx[piVSNo].mi1Min[piVNNo] >= 0) { return 0; } // 末尾に追加 sz1Heap[siHCnt].miVNo = piVNNo; sz1Heap[siHCnt].miLen = piLen; 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; } // 実行メイン long long fMain( ) { int i, j, liRet; char lc1Buf[1024]; // 頂点数・辺数・レッカー車 - 取得 int liECnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &siVCnt, &liECnt, &siWNo); siWNo--; // トラック数 - 取得 for (i = 0; i < siVCnt; i++) { fscanf(szpFpI, "%d", &sz1Vtx[i].miTCnt); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); // 辺 - 取得 for (i = 0; i < liECnt; i++) { int liVNo1, liVNo2, liLen; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &liVNo1, &liVNo2, &liLen); liVNo1--; liVNo2--; // 辺 - 追加 fAddEdge(liVNo1, liVNo2, liLen); fAddEdge(liVNo2, liVNo1, liLen); } // 最短距離 - セット for (i = 0; i < siVCnt; i++) { // 初期化 memset(sz1Vtx[i].mi1Min, -1, sizeof(sz1Vtx[i].mi1Min)); // ヒープ - 初期値 siHCnt = 0; fHeapEnqueue(i, i, 0); // ヒープ - 取得 while (1) { Heap lzHeap; liRet = fHeapDequeue(&lzHeap); if (liRet != 0) { break; } // 最短距離 - セット if (sz1Vtx[i].mi1Min[lzHeap.miVNo] > 0) { continue; } sz1Vtx[i].mi1Min[lzHeap.miVNo] = lzHeap.miLen; // 辺でループ Edge *lzpEdge = sz1Vtx[lzHeap.miVNo].mzpEdge; while (lzpEdge != NULL) { // ヒープ - 追加 fHeapEnqueue(i, lzpEdge->miVNo, lzpEdge->miLen + lzHeap.miLen); // 次の辺へ lzpEdge = lzpEdge->mzpNext; } } } // 最小移動距離 - 取得 long long llMin = LLONG_MAX; for (i = 0; i < siVCnt; i++) { // 開始位置でループ long long llSum = 0; for (j = 0; j < siVCnt; j++) { llSum += (long long)sz1Vtx[i].mi1Min[j] * 2 * (long long)sz1Vtx[j].miTCnt; } // 開始位置でループ for (j = 0; j < siVCnt; j++) { // レッカー車を開始位置へ移動 long long llLen = llSum + sz1Vtx[siWNo].mi1Min[j]; // 片道減算 if (sz1Vtx[j].miTCnt > 0) { llLen -= sz1Vtx[j].mi1Min[i]; } // 最小移動距離 - 更新 if (llMin > llLen) { llMin = llLen; } } } return llMin; } // 1回実行 int fOne( ) { long long llRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1Vtx, 0, sizeof(sz1Vtx)); // 頂点 siECnt = 0; // 辺数 // 入力 - セット #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 // 実行メイン llRet = fMain(); // 出力 sprintf(lc1Buf, "%lld\n", llRet); 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; }