結果

問題 No.827 総神童数
ユーザー asugen0402asugen0402
提出日時 2019-05-29 11:31:07
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 145 ms / 2,000 ms
コード長 5,619 bytes
コンパイル時間 1,818 ms
コンパイル使用メモリ 33,128 KB
実行使用メモリ 18,656 KB
最終ジャッジ日時 2023-10-17 18:41:27
合計ジャッジ時間 4,732 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
8,136 KB
testcase_01 AC 3 ms
8,136 KB
testcase_02 AC 2 ms
8,136 KB
testcase_03 AC 3 ms
8,136 KB
testcase_04 AC 2 ms
8,136 KB
testcase_05 AC 2 ms
8,136 KB
testcase_06 AC 3 ms
8,136 KB
testcase_07 AC 2 ms
8,136 KB
testcase_08 AC 2 ms
8,136 KB
testcase_09 AC 141 ms
18,656 KB
testcase_10 AC 45 ms
12,020 KB
testcase_11 AC 4 ms
8,228 KB
testcase_12 AC 19 ms
9,156 KB
testcase_13 AC 110 ms
17,732 KB
testcase_14 AC 5 ms
8,348 KB
testcase_15 AC 42 ms
12,320 KB
testcase_16 AC 111 ms
15,616 KB
testcase_17 AC 134 ms
16,576 KB
testcase_18 AC 54 ms
12,692 KB
testcase_19 AC 145 ms
12,908 KB
testcase_20 AC 102 ms
12,796 KB
testcase_21 AC 74 ms
12,640 KB
testcase_22 AC 114 ms
12,848 KB
testcase_23 AC 3 ms
8,140 KB
testcase_24 AC 127 ms
12,904 KB
testcase_25 AC 94 ms
12,752 KB
testcase_26 AC 128 ms
12,908 KB
testcase_27 AC 85 ms
12,688 KB
testcase_28 AC 83 ms
12,684 KB
testcase_29 AC 62 ms
10,548 KB
testcase_30 AC 18 ms
8,248 KB
testcase_31 AC 49 ms
10,464 KB
testcase_32 AC 46 ms
10,456 KB
testcase_33 AC 139 ms
12,908 KB
testcase_34 AC 125 ms
12,912 KB
testcase_35 AC 48 ms
10,468 KB
testcase_36 AC 72 ms
10,584 KB
testcase_37 AC 96 ms
12,748 KB
testcase_38 AC 134 ms
12,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <float.h>
#include <limits.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

// 内部定数
#define D_ON			1										// 汎用フラグ - ON
#define D_OFF			0										// 汎用フラグ - OFF
#define D_MOD			1000000007								// 除数(10の9乗+7)
#define D_VTX_MAX		200000									// 最大頂点数
#define D_EDGE_MAX		200000									// 最大辺数
#define D_FACT_MAX		D_VTX_MAX								// 最大階乗数

// 内部構造体 - 辺情報
typedef struct Edge {
	int miVNo;													// 接続先頂点
	struct Edge *mzpNext;										// 次の辺情報
} Edge;

// 内部構造体 - 頂点情報
typedef struct Vtx {
	Edge *mzpEdge;												// 辺
	int miDone;													// 処理済フラグ
} Vtx;

// 内部変数
static FILE *szpFpI;											// 入力
static Vtx sz1Vtx[D_VTX_MAX];									// 頂点
static int siVCnt;												// 頂点数
static Edge sz1Edge[D_EDGE_MAX * 2];							// 辺
static int siECnt;												// 辺数
static int si1ModFact[D_FACT_MAX];								// 階乗リスト
static int si1ModFactR[D_FACT_MAX];								// 階乗リスト(逆元)
static int siSum;												// 合計

// 内部変数 - テスト用
#ifdef D_TEST
	static int siRes;
	static FILE *szpFpA;
	static int siTNo;
#endif

// 出力
int
fOut(
	char *pcpLine				// <I> １行
)
{
	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
fGetModPower(
	int piBase					// <I> 基数
	, int piIdx					// <I> 指数
)
{
	// 平方リスト - 作成
	int li1Val[100];
	li1Val[0] = piBase;
	int liCnt = 1;
	int liIdx = 1;
	while (piIdx > liIdx) {
		li1Val[liCnt] = (int)((long long)li1Val[liCnt - 1] * (long long)li1Val[liCnt - 1] % (long long)D_MOD);
		liCnt++;
		liIdx += liIdx;
	}

	// べき乗 - 取得
	int liVal = 1;
	while (piIdx > 0) {
		if (piIdx >= liIdx) {
			piIdx -= liIdx;
			liVal = (int)((long long)liVal * (long long)li1Val[liCnt - 1] % (long long)D_MOD);
		}
		liCnt--;
		liIdx /= 2;
	}

	return liVal;
}

// 階乗リスト - 作成
int
fMakeModFact(
	int piMax					// <I> 最大値
)
{
	int i;

	si1ModFact[0] = 1;
	si1ModFact[1] = 1;
	for (i = 2; i <= piMax; i++) {
		si1ModFact[i] = (int)((long long)si1ModFact[i - 1] * (long long)i % (long long)D_MOD);
	}

	return 0;
}

// 階乗リスト(逆元) - 作成
int
fMakeModFactR(
	int piMax					// <I> 最大値
)
{
	int i;

	for (i = 0; i <= piMax; i++) {
		si1ModFactR[i] = fGetModPower(si1ModFact[i], D_MOD - 2);
	}

	return 0;
}

// nCk - 取得
int
fGetnCk(
	int piN						// <I> N
	, int piK					// <I> K
)
{
	int liCnt = (int)((long long)si1ModFact[piN] * (long long)si1ModFactR[piN - piK] % (long long)D_MOD);
	liCnt = (int)((long long)liCnt * (long long)si1ModFactR[piK] % (long long)D_MOD);

	return liCnt;
}

// 合計 - セット
int
fSetSum(
	int piVNo					// <I> 頂点 0～
	, int piH					// <I> 高さ 1～
)
{
	// 処理済フラグ
	if (sz1Vtx[piVNo].miDone != D_OFF) {
		return 0;
	}
	sz1Vtx[piVNo].miDone = D_ON;

	// 対象頂点の選択パターン数
	int liPCnt = fGetnCk(siVCnt, piH);

	// 上位のパターン数
	liPCnt = (int)((long long)liPCnt * (long long)si1ModFact[piH - 1] % (long long)D_MOD);

	// 対象以外のパターン数
	liPCnt = (int)((long long)liPCnt * (long long)si1ModFact[siVCnt - piH] % (long long)D_MOD);

	// 合計 - 加算
	siSum = (siSum + liPCnt) % D_MOD;

	// 辺でループ
	Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge;
	while (lzpEdge != NULL) {

		// 下位へ
		fSetSum(lzpEdge->miVNo, piH + 1);

		// 次の辺へ
		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 - 1; i++) {
		int liVNo1, liVNo2;
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
		sscanf(lc1Buf, "%d%d", &liVNo1, &liVNo2);
		liVNo1--;
		liVNo2--;

		// 辺 - 追加
		fAddEdge(liVNo1, liVNo2);
		fAddEdge(liVNo2, liVNo1);
	}

	// 階乗リスト - 作成
	fMakeModFact(siVCnt);
	fMakeModFactR(siVCnt);

	// 合計 - セット
	fSetSum(0, 1);

	return 0;
}

// １回実行
int
fOne(
)
{
	int liRet;
	char lc1Buf[1024];

	// データ - 初期化
	memset(sz1Vtx, 0, sizeof(sz1Vtx));							// 頂点
	siECnt = 0;													// 辺数
	siSum = 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, "%d\n", siSum);
	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;
}
0