#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 // 最大辺数 #define D_SEGT_CNT 131072 // セグメントツリーデータ数(2の累乗) // 内部構造体 - 辺情報 typedef struct Edge { int miVNo; // 接続先頂点 struct Edge *mzpPrev; // 前の辺情報 struct Edge *mzpNext; // 次の辺情報 struct Edge *mzpRev; // 逆側の辺情報 } Edge; // 内部構造体 - 頂点情報 typedef struct Vtx { int miVNo; // 頂点 0~ Edge *mzpEdge; // 辺 int miECnt; // 辺数 char mcDel; // 削除フラグ } Vtx; // 内部変数 static FILE *szpFpI; // 入力 static Vtx sz1Vtx[D_VTX_MAX]; // 頂点 static int siVCnt; // 頂点数 static Edge sz1Edge[D_EDGE_MAX * 2]; // 辺 static int siECnt; // 辺数 static Vtx *szp1SegT[D_SEGT_CNT * 2]; // セグメントツリー static int siSCNo; // セグメントツリー - 子の開始位置 // 内部変数 - テスト用 #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 piERNo // 逆側の辺 0~ ) { // 対象辺 Edge *lzpNew = &sz1Edge[siECnt]; Edge *lzpNow = sz1Vtx[piVFNo].mzpEdge; siECnt++; // セット lzpNew->miVNo = piVTNo; if (lzpNow == NULL) { // 新規 lzpNew->mzpPrev = lzpNew; lzpNew->mzpNext = lzpNew; sz1Vtx[piVFNo].mzpEdge = lzpNew; } else { // 追加 lzpNow->mzpPrev->mzpNext = lzpNew; lzpNew->mzpNext = lzpNow; lzpNew->mzpPrev = lzpNow->mzpPrev; lzpNow->mzpPrev = lzpNew; } lzpNew->mzpRev = &sz1Edge[piERNo]; // 辺数 sz1Vtx[piVFNo].miECnt++; return 0; } // 辺 - 削除 int fDelEdge( int piVNo // 頂点 0~ , Edge *pzpEdge // 辺 ) { if (pzpEdge == pzpEdge->mzpNext) { pzpEdge->mzpNext = NULL; } else { pzpEdge->mzpPrev->mzpNext = pzpEdge->mzpNext; pzpEdge->mzpNext->mzpPrev = pzpEdge->mzpPrev; } if (sz1Vtx[piVNo].mzpEdge == pzpEdge) { sz1Vtx[piVNo].mzpEdge = pzpEdge->mzpNext; } // 辺数 sz1Vtx[piVNo].miECnt--; return 0; } // セグメントツリー - 子の開始位置 - セット int fSegTSetCSNo( int piCnt // 子データ数 ) { siSCNo = 1; while (siSCNo < piCnt) { siSCNo *= 2; } return 0; } // セグメントツリー - 子の処理 int fSegTCProc( int piNo1 // 位置1 0~ , int piNo2 // 位置2 0~ ) { if (szp1SegT[piNo1] == NULL) { return piNo2; } if (szp1SegT[piNo2] == NULL) { return piNo1; } if (szp1SegT[piNo1]->miECnt > szp1SegT[piNo2]->miECnt) { return piNo1; } else if (szp1SegT[piNo1]->miECnt < szp1SegT[piNo2]->miECnt) { return piNo2; } return piNo1; } // セグメントツリー - 親データセット int fSegTSetPData( ) { int i; // 最初の親番号範囲 int liPNo1 = siSCNo / 2; int liPNo2 = siSCNo - 1; // 作成 while (liPNo1 > 0) { for (i = liPNo1; i <= liPNo2; i++) { // 対象の子を取得 int liCNo = fSegTCProc(i * 2, i * 2 + 1); // セット szp1SegT[i] = szp1SegT[liCNo]; } // 次の親番号範囲 liPNo2 = liPNo1 - 1; liPNo1 /= 2; } return 0; } // セグメントツリー - 更新 int fSegTUpOne( int piUpNo // 更新位置 0~ ) { // 子番号 int liCNo1 = siSCNo + piUpNo; int liCNo2; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } // 親を更新 while (1) { // 親番号 int liPNo = liCNo1 / 2; if (liPNo < 1) { break; } // 対象の子を取得 int liCNo = fSegTCProc(liCNo1, liCNo2); // 更新 szp1SegT[liPNo] = szp1SegT[liCNo]; // 次の子番号 liCNo1 = liPNo; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } } return 0; } // 実行メイン int fMain( ) { int i; char lc1Buf[1024]; // 頂点数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &siVCnt); // 頂点 - 初期化 for (i = 0; i < siVCnt; i++) { sz1Vtx[i].miVNo = i; } // 辺 - 取得 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, siECnt + 1); fAddEdge(liVNo2, liVNo1, siECnt - 1); } // セグメントツリー - 子の開始位置 - セット fSegTSetCSNo(siVCnt); // セグメントツリー - 子データセット for (i = 0; i < siVCnt; i++) { szp1SegT[siSCNo + i] = &sz1Vtx[i]; } // セグメントツリー - 親データセット fSegTSetPData(); // 全辺 - 削除 while (1) { // 対象頂点 Vtx *lzpVtx = szp1SegT[1]; if (lzpVtx->miECnt < 1) { break; } // 対象頂点の全辺 - 削除 while (lzpVtx->mzpEdge != NULL) { Edge *lzpEdge = lzpVtx->mzpEdge; // 逆側の辺 - 削除 Edge *lzpRev = lzpEdge->mzpRev; fDelEdge(lzpEdge->miVNo, lzpRev); fSegTUpOne(lzpEdge->miVNo); // 辺 - 削除 fDelEdge(lzpVtx->miVNo, lzpEdge); } // 対象頂点 - 削除 lzpVtx->mcDel = D_ON; fSegTUpOne(lzpVtx->miVNo); } // 残った頂点数 int liCnt = 0; for (i = 0; i < siVCnt; i++) { if (sz1Vtx[i].mcDel == D_OFF) { liCnt++; } } return liCnt; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1Vtx, 0, sizeof(sz1Vtx)); // 頂点 siECnt = 0; // 辺数 memset(szp1SegT, 0, sizeof(szp1SegT)); // セグメントツリー // 入力 - セット #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, "%d\n", liRet); 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; }