#include #include #include #include #include #include #include // 内部定数 #define D_VTX_MAX 100000 // 最大頂点数 #define D_EDGE_MAX D_VTX_MAX // 最大辺数 #define D_LEN_MAX 20 // 最大距離情報数 // 内部構造体 - 辺情報 typedef struct Edge { int miVNo; // 接続先頂点 struct Edge *mzpNext; // 次の辺情報 } Edge; // 内部構造体 - 距離情報 typedef struct Len { int miVNo; // 頂点 int miTax; // 税合計 } Len; // 内部構造体 - 頂点情報 typedef struct Vtx { Edge *mzpEdge; // 辺 int miTax; // 税 int miHeight; // 高さ 1~ Len mz1Len[D_LEN_MAX]; // 距離情報 int miLCnt; // 距離情報数 } Vtx; // 内部変数 static FILE *szpFpI; // 入力 static Vtx sz1Vtx[D_VTX_MAX]; // 頂点 static int siVCnt; // 頂点数 static Edge sz1Edge[D_EDGE_MAX * 2]; // 辺 static int siECnt; // 辺数 static long long slSum; // 合計額 // 内部変数 - テスト用 #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~ ) { sz1Edge[siECnt].miVNo = piVTNo; sz1Edge[siECnt].mzpNext = sz1Vtx[piVFNo].mzpEdge; sz1Vtx[piVFNo].mzpEdge = &sz1Edge[siECnt]; siECnt++; return 0; } // 高さ - セット int fSetHeight( int piVNo // 頂点 0~ , int piHeight // 高さ 1~ ) { // セット済 if (sz1Vtx[piVNo].miHeight > 0) { return 0; } sz1Vtx[piVNo].miHeight = piHeight; // 辺でループ Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge; while (lzpEdge != NULL) { // 下位へ fSetHeight(lzpEdge->miVNo, piHeight + 1); // 次の辺へ lzpEdge = lzpEdge->mzpNext; } return 0; } // 距離情報 - セット - 1つ目 int fSetLen1( ) { int i; for (i = 1; i < siVCnt; i++) { // 上の頂点 - 取得 int liVNo; Edge *lzpEdge = sz1Vtx[i].mzpEdge; while (lzpEdge != NULL) { liVNo = lzpEdge->miVNo; // 高さ - チェック if (sz1Vtx[liVNo].miHeight == sz1Vtx[i].miHeight - 1) { break; } // 次の辺へ lzpEdge = lzpEdge->mzpNext; } // 距離情報 - セット sz1Vtx[i].mz1Len[0].miVNo = liVNo; sz1Vtx[i].mz1Len[0].miTax = sz1Vtx[i].miTax; sz1Vtx[i].miLCnt = 1; } return 0; } // 距離情報 - セット - 2つ目以降 int fSetLen2( int piLNo // 距離情報 1~ ) { int i; int liSCnt = 0; for (i = 1; i < siVCnt; i++) { // データ有無 if (sz1Vtx[i].miLCnt < piLNo) { continue; } // 上の頂点 int liVNo = sz1Vtx[i].mz1Len[piLNo - 1].miVNo; // データ有無 if (sz1Vtx[liVNo].miLCnt < piLNo) { continue; } // 距離情報 - セット sz1Vtx[i].mz1Len[piLNo].miVNo = sz1Vtx[liVNo].mz1Len[piLNo - 1].miVNo; sz1Vtx[i].mz1Len[piLNo].miTax = sz1Vtx[i].mz1Len[piLNo - 1].miTax + sz1Vtx[liVNo].mz1Len[piLNo - 1].miTax; sz1Vtx[i].miLCnt++; liSCnt++; } return liSCnt; } // 頂点移動 int fMove( int *pipVNo // 頂点 , int piMove // 移動量 , int piHCnt // 人数 ) { int i; // 移動 while (piMove > 0) { // 移動先 - 取得 int liMove; for (i = sz1Vtx[*pipVNo].miLCnt - 1; i >= 0; i--) { liMove = 1 << i; if (piMove >= liMove) { break; } } // 移動 piMove -= liMove; slSum += sz1Vtx[*pipVNo].mz1Len[i].miTax * piHCnt; *pipVNo = sz1Vtx[*pipVNo].mz1Len[i].miVNo; } return 0; } // 実行メイン int fMain( ) { int i, j, liRet; char lc1Buf[1024]; // 頂点数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &siVCnt); // 辺 - 取得 for (i = 0; i < siVCnt - 1; i++) { int liVNo1, liVNo2; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &liVNo1, &liVNo2); // 辺 - 追加 fAddEdge(liVNo1, liVNo2); fAddEdge(liVNo2, liVNo1); } // 税 - 取得 for (i = 0; i < siVCnt; i++) { fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &sz1Vtx[i].miTax); } // 高さ - セット fSetHeight(0, 1); // 距離情報 - セット - 1つ目 fSetLen1(); // 距離情報 - セット - 2つ目以降 for (i = 1; ; i++) { liRet = fSetLen2(i); if (liRet < 1) { break; } } // 行商データ数 - 取得 int liDCnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &liDCnt); // 行商データ - 取得 for (i = 0; i < liDCnt; i++) { int liVNo1, liVNo2, liHCnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d%d", &liVNo1, &liVNo2, &liHCnt); // 高さを揃える if (sz1Vtx[liVNo1].miHeight < sz1Vtx[liVNo2].miHeight) { fMove(&liVNo2, sz1Vtx[liVNo2].miHeight - sz1Vtx[liVNo1].miHeight, liHCnt); } else if (sz1Vtx[liVNo1].miHeight > sz1Vtx[liVNo2].miHeight) { fMove(&liVNo1, sz1Vtx[liVNo1].miHeight - sz1Vtx[liVNo2].miHeight, liHCnt); } // 頂点を一致させる while (liVNo1 != liVNo2) { // 移動先 - 取得 for (j = sz1Vtx[liVNo1].miLCnt - 1; j >= 0; j--) { if (sz1Vtx[liVNo1].mz1Len[j].miVNo != sz1Vtx[liVNo2].mz1Len[j].miVNo) { break; } } if (j < 0) { j = 0; } // 移動 fMove(&liVNo1, 1 << j, liHCnt); fMove(&liVNo2, 1 << j, liHCnt); } // 合計額 - 加算 slSum += sz1Vtx[liVNo1].miTax * liHCnt; } return 0; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1Vtx, 0, sizeof(sz1Vtx)); // 頂点 siECnt = 0; // 辺数 slSum = 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 // 実行メイン liRet = fMain(); // 結果 - セット sprintf(lc1Buf, "%lld\n", slSum); // 結果 - 出力 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; }