結果

問題 No.439 チワワのなる木
ユーザー asugen0402
提出日時 2019-04-09 16:39:15
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 5,429 bytes
コンパイル時間 267 ms
コンパイル使用メモリ 32,384 KB
実行使用メモリ 14,304 KB
最終ジャッジ日時 2024-07-04 01:44:47
合計ジャッジ時間 2,131 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 WA * 17
権限があれば一括ダウンロードができます

ソースコード

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_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				// <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
fAddPCntUp(
	int piVNo					// <I> 頂点 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					// <I> 頂点 0～
	, int piCCnt				// <I> 上位のc数合計
	, long long plWCnt			// <I> 上位の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;
}

// １回実行
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;
}
0