#342231 (C) No.763 Noelちゃんと木遊び

提出ソース

結果

問題	No.763 Noelちゃんと木遊び
コンテスト
ユーザー	asugen0402
提出日時	2019-04-26 13:14:20
言語	C (gcc 13.3.0)
結果	WA
実行時間	-
コード長	7,027 bytes
記録記録タグの例: 初AC ショートコード純ショートコード純主流ショートコード最速実行時間
コンパイル時間	249 ms
コンパイル使用メモリ	34,048 KB
実行使用メモリ	12,548 KB
最終ジャッジ日時	2024-11-24 05:16:08
合計ジャッジ時間	2,523 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

このコードへのチャレンジ
（要ログイン）

ファイルパターン	結果
other	AC * 1 WA * 20

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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									// 最大辺数
#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				// <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～
	, int piERNo				// <I> 逆側の辺 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					// <I> 頂点 0～
	, Edge *pzpEdge				// <I> 辺
)
{
	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					// <I> 子データ数
)
{
	siSCNo = 1;
	while (siSCNo < piCnt) {
		siSCNo *= 2;
	}

	return 0;
}

// セグメントツリー - 子の処理
int
fSegTCProc(
	int piNo1					// <I> 位置１ 0～
	, int piNo2					// <I> 位置２ 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					// <I> 更新位置 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;
}

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

yukicoder

結果

ソースコード