結果
問題 | No.778 クリスマスツリー |
ユーザー | asugen0402 |
提出日時 | 2019-04-26 16:16:50 |
言語 | C (gcc 12.3.0) |
結果 |
AC
|
実行時間 | 108 ms / 2,000 ms |
コード長 | 5,632 bytes |
コンパイル時間 | 1,039 ms |
コンパイル使用メモリ | 31,872 KB |
実行使用メモリ | 14,816 KB |
最終ジャッジ日時 | 2024-10-14 01:58:50 |
合計ジャッジ時間 | 1,683 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
5,504 KB |
testcase_01 | AC | 3 ms
5,504 KB |
testcase_02 | AC | 3 ms
5,504 KB |
testcase_03 | AC | 4 ms
5,376 KB |
testcase_04 | AC | 4 ms
5,504 KB |
testcase_05 | AC | 3 ms
5,376 KB |
testcase_06 | AC | 63 ms
14,776 KB |
testcase_07 | AC | 52 ms
8,568 KB |
testcase_08 | AC | 108 ms
11,576 KB |
testcase_09 | AC | 102 ms
8,520 KB |
testcase_10 | AC | 104 ms
8,448 KB |
testcase_11 | AC | 101 ms
8,576 KB |
testcase_12 | AC | 100 ms
8,524 KB |
testcase_13 | AC | 63 ms
8,528 KB |
testcase_14 | AC | 64 ms
14,816 KB |
ソースコード
#include <float.h> #include <limits.h> #include <math.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <time.h> // 内部定数 #define D_VTX_MAX 200000 // 最大頂点数 #define D_EDGE_MAX 200000 // 最大辺数 #define D_SEGT_CNT 262144 // セグメントツリーデータ数(2の累乗) // 内部構造体 - 辺情報 typedef struct Edge { int miVNo; // 接続先頂点 struct Edge *mzpNext; // 次の辺情報 } Edge; // 内部構造体 - 頂点情報 typedef struct Vtx { Edge *mzpEdge; // 辺 } Vtx; // 内部変数 static FILE *szpFpI; // 入力 static Vtx sz1Vtx[D_VTX_MAX]; // 頂点 static int siVCnt; // 頂点数 static Edge sz1Edge[D_EDGE_MAX]; // 辺 static int siECnt; // 辺数 static int si1SegT[D_SEGT_CNT * 2]; // セグメントツリー static int siSCNo; // セグメントツリー - 子の開始位置 static long long slPCnt; // パターン数 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; static int siTNo; #endif // 出力 int fOut( char *pcpLine // <I> 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 // <I> 頂点 - 元 0~ , int piVTNo // <I> 頂点 - 先 0~ ) { sz1Edge[siECnt].miVNo = piVTNo; sz1Edge[siECnt].mzpNext = sz1Vtx[piVFNo].mzpEdge; sz1Vtx[piVFNo].mzpEdge = &sz1Edge[siECnt]; siECnt++; return 0; } // セグメントツリー - 子の開始位置 - セット int fSegTSetCSNo( int piCnt // <I> 子データ数 ) { siSCNo = 1; while (siSCNo < piCnt) { siSCNo *= 2; } return 0; } // セグメントツリー - 子の処理 int fSegTCProc( int piNo1 // <I> 位置1 0~ , int piNo2 // <I> 位置2 0~ ) { return si1SegT[piNo1] + si1SegT[piNo2]; } // セグメントツリー - 取得 int fSegTGetRngMain( int piGetS // <I> 取得範囲 - 開始 0~ , int piGetE // <I> 取得範囲 - 終了 0~ , int piNNo // <I> 現在位置 1~ , int piNowS // <I> 現在範囲 - 開始 0~ , int piNowE // <I> 現在範囲 - 終了 0~ ) { // 内包チェック if (piGetS <= piNowS && piNowE <= piGetE) { return si1SegT[piNNo]; } // 中間位置 int liCenter = (piNowS + piNowE) / 2; int liRet = 0; // 左側 if (piGetS <= liCenter) { liRet = fSegTGetRngMain(piGetS, piGetE, piNNo * 2, piNowS, liCenter); } // 右側 if (piGetE >= liCenter + 1) { liRet += fSegTGetRngMain(piGetS, piGetE, piNNo * 2 + 1, liCenter + 1, piNowE); } return liRet; } int fSegTGetRng( int piGetS // <I> 取得範囲 - 開始 0~ , int piGetE // <I> 取得範囲 - 終了 0~ ) { return fSegTGetRngMain(piGetS, piGetE, 1, 0, siSCNo - 1); } // セグメントツリー - 更新 int fSegTUpOne( int piUpNo // <I> 更新位置 0~ , int piVal // <I> 更新値 ) { // 子番号 int liCNo1 = siSCNo + piUpNo; int liCNo2; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } // 更新 si1SegT[liCNo1] = piVal; // 親を更新 while (1) { // 親番号 int liPNo = liCNo1 / 2; if (liPNo < 1) { break; } // 更新 si1SegT[liPNo] = fSegTCProc(liCNo1, liCNo2); // 次の子番号 liCNo1 = liPNo; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } } return 0; } // パターン数 - セット int fSetPCnt( int piVNo // <I> 頂点 0~ ) { // パターン数 - 加算 slPCnt += fSegTGetRng(0, piVNo); // 頂点 - 登録 fSegTUpOne(piVNo, 1); // 辺でループ Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge; while (lzpEdge != NULL) { // 下位へ fSetPCnt(lzpEdge->miVNo); // 次の辺へ lzpEdge = lzpEdge->mzpNext; } // 頂点 - 削除 fSegTUpOne(piVNo, 0); return 0; } // 実行メイン int fMain( ) { int i; char lc1Buf[1024]; // 頂点数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d", &siVCnt); // 辺 - 取得 for (i = 1; i < siVCnt; i++) { int liVNo; fscanf(szpFpI, "%d", &liVNo); // 辺 - 追加 fAddEdge(liVNo, i); } // セグメントツリー - 子の開始位置 - セット fSegTSetCSNo(siVCnt); // パターン数 - セット fSetPCnt(0); return 0; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1Vtx, 0, sizeof(sz1Vtx)); // 頂点 siECnt = 0; // 辺数 memset(si1SegT, 0, sizeof(si1SegT)); // セグメントツリー 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; }