#include #include #include #include #include #include #include // 内部定数 #define D_ON 1 // 汎用フラグ - ON #define D_OFF 0 // 汎用フラグ - OFF #define D_VTX_MAX 100000 // 最大頂点数 #define D_EDGE_MAX 100000 // 最大辺数 // 内部構造体 - 辺情報 typedef struct Edge { int miVNo; // 接続先頂点 struct Edge *mzpNext; // 次の辺情報 } Edge; // 内部構造体 - 頂点情報 typedef struct Vtx { int miDone; // 処理済フラグ char mcChar; // 文字 Edge *mzpEdge; // 辺 int miCCnt; // c数 long long mlWCnt; // w数 } 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 slPCnt; // パターン数 // 内部変数 - テスト用 #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 fAddPCntUp( int piVNo // 頂点 0~ ) { int liRet; // 処理済フラグ if (sz1Vtx[piVNo].miDone != D_OFF) { return -1; } sz1Vtx[piVNo].miDone = D_ON; // 辺でループ Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge; while (lzpEdge != NULL) { // 下位へ liRet = fAddPCntUp(lzpEdge->miVNo); if (liRet == 0) { sz1Vtx[piVNo].miCCnt += sz1Vtx[lzpEdge->miVNo].miCCnt; sz1Vtx[piVNo].mlWCnt += sz1Vtx[lzpEdge->miVNo].mlWCnt; } // 次の辺へ lzpEdge = lzpEdge->mzpNext; } // 自分 if (sz1Vtx[piVNo].mcChar == 'c') { sz1Vtx[piVNo].miCCnt++; } else { slPCnt += sz1Vtx[piVNo].mlWCnt; sz1Vtx[piVNo].mlWCnt += sz1Vtx[piVNo].miCCnt; } return 0; } // パターン数 - 加算 - 上->下 int fAddPCntDown( int piVNo // 頂点 0~ , int piCCnt // 上位のc数合計 , long long plWCnt // 上位のw数合計 ) { // 処理済フラグ if (sz1Vtx[piVNo].miDone != D_OFF) { return -1; } sz1Vtx[piVNo].miDone = D_ON; // 自分 if (sz1Vtx[piVNo].mcChar == 'c') { piCCnt++; } else { slPCnt += plWCnt; plWCnt += piCCnt; } // 辺でループ Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge; while (lzpEdge != NULL) { // c数合計・w数合計 - 加算 if (sz1Vtx[lzpEdge->miVNo].miDone == D_OFF) { piCCnt += sz1Vtx[lzpEdge->miVNo].miCCnt; plWCnt += sz1Vtx[lzpEdge->miVNo].mlWCnt; } // 次の辺へ lzpEdge = lzpEdge->mzpNext; } // 辺でループ lzpEdge = sz1Vtx[piVNo].mzpEdge; while (lzpEdge != NULL) { // c数合計・w数合計 int liCCnt = piCCnt - sz1Vtx[lzpEdge->miVNo].miCCnt; long long llWCnt = plWCnt - sz1Vtx[lzpEdge->miVNo].mlWCnt; // 下位へ fAddPCntDown(lzpEdge->miVNo, liCCnt, llWCnt); // 次の辺へ lzpEdge = lzpEdge->mzpNext; } return 0; } // 実行メイン int fMain( ) { int i; char lc1Buf[1024]; // 頂点数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &siVCnt); // 文字 - 取得 for (i = 0; i < siVCnt; i++) { fscanf(szpFpI, "%c", &sz1Vtx[i].mcChar); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); // 辺 - 取得 for (i = 0; i < siVCnt - 1; i++) { int liVNo1, liVNo2; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &liVNo1, &liVNo2); liVNo1--; liVNo2--; // 辺 - 追加 fAddEdge(liVNo1, liVNo2); fAddEdge(liVNo2, liVNo1); } // パターン数 - 加算 - 下->上 fAddPCntUp(0); // 処理済フラグ - 初期化 for (i = 0; i < siVCnt; i++) { sz1Vtx[i].miDone = D_OFF; } // パターン数 - 加算 - 上->下 fAddPCntDown(0, 0, 0); return 0; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1Vtx, 0, sizeof(sz1Vtx)); // 頂点 siECnt = 0; // 辺数 slPCnt = 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", slPCnt); // 結果 - 出力 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; }