#341878 (C) No.674 n連勤

提出ソース
結果

問題	No.674 n連勤
ユーザー	asugen0402
提出日時	2019-04-24 16:40:05
言語	C (gcc 12.3.0)
結果	AC
実行時間	34 ms / 2,000 ms
コード長	12,318 bytes
コンパイル時間	386 ms
コンパイル使用メモリ	33,408 KB
実行使用メモリ	5,248 KB
最終ジャッジ日時	2024-11-08 02:22:34
合計ジャッジ時間	1,559 ms
ジャッジサーバーID （参考情報）	judge5 / judge1
このコードへのチャレンジ
（要ログイン）
テストケース

テストケース表示
入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	1 ms 5,248 KB
testcase_02	AC	1 ms 5,248 KB
testcase_03	AC	1 ms 5,248 KB
testcase_04	AC	1 ms 5,248 KB
testcase_05	AC	1 ms 5,248 KB
testcase_06	AC	1 ms 5,248 KB
testcase_07	AC	1 ms 5,248 KB
testcase_08	AC	1 ms 5,248 KB
testcase_09	AC	1 ms 5,248 KB
testcase_10	AC	1 ms 5,248 KB
testcase_11	AC	2 ms 5,248 KB
testcase_12	AC	3 ms 5,248 KB
testcase_13	AC	17 ms 5,248 KB
testcase_14	AC	18 ms 5,248 KB
testcase_15	AC	17 ms 5,248 KB
testcase_16	AC	33 ms 5,248 KB
testcase_17	AC	34 ms 5,248 KB
testcase_18	AC	28 ms 5,248 KB
testcase_19	AC	18 ms 5,248 KB
権限があれば一括ダウンロードができます
ソースコード

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_TREE_MAX		60000									// 木の最大データ数
#define D_TREE_WCNT		2										// 木の方向数
#define D_TREE_LEFT		0										// 木の方向 - 左側
#define D_TREE_RIGHT	1										// 木の方向 - 右側

// 内部構造体 - 木構造
typedef struct Tree {
	long long mlKey;											// キー(日付)
	int mi1Height[D_TREE_WCNT];									// 木の高さ
	struct Tree *mzp1Child[D_TREE_WCNT];						// 子
} Tree;

// 内部変数
static FILE *szpFpI;											// 入力
static Tree sz1Tree[D_TREE_MAX];								// 木の実データ
static int siTCnt;												// 木の実データ数
static Tree *szpTop;											// 先頭の木データ
static Tree *szpFree;											// 空きデータ(削除時に発生)

// プロトタイプ宣言
int fTreeDel(Tree **, long long);								// 削除

// 内部変数 - テスト用
#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;
}

// 木データ - 作成
Tree *
fTreeMake(
	long long plKey				// <I> キー
)
{
	// 対象の木データ
	Tree *lzpTree;
	if (szpFree == NULL) {								// 空き領域なし
		lzpTree = &(sz1Tree[siTCnt]);
		siTCnt++;
	}
	else {												// 空き領域あり
		lzpTree = szpFree;
		szpFree = szpFree->mzp1Child[D_TREE_LEFT];
	}

	// データセット
	memset(lzpTree, 0, sizeof(Tree));		// 初期化
	lzpTree->mlKey = plKey;					// キー

	return lzpTree;
}

// 木データ - 比較 - キー昇順
int
fTreeCmp(
	Tree *pzpTree				// <I> 木データ
	, long long plKey			// <I> キー
)
{
	// キー昇順
	if (plKey < pzpTree->mlKey) {			// 左側
		return -1;
	}
	else if (plKey > pzpTree->mlKey) {		// 右側
		return 1;
	}

	return 0;
}

// 木データ - 検索
Tree *
fTreeSrh(
	long long plKey			// <I> キー
)
{
	// 先頭の木データ
	Tree *lzpNow = szpTop;

	// 検索
	while (1) {

		// データ有無
		if (lzpNow == NULL) {
			return NULL;
		}

		// 比較
		int liRet = fTreeCmp(lzpNow, plKey);
		if (liRet == 0) {								// 一致
			return lzpNow;
		}

		// 子へ移動
		if (liRet < 0) {								// 左側
			lzpNow = lzpNow->mzp1Child[D_TREE_LEFT];
		}
		else {											// 右側
			lzpNow = lzpNow->mzp1Child[D_TREE_RIGHT];
		}
	}

	return NULL;
}

// 木データ - 検索 - 指定値 or 指定値に近い値
Tree *
fTreeSrhRng(
	int piMode					// <I> [-1]以下 or [1]以上
	, long long plKey			// <I> キー
)
{
	// 先頭の木データ
	Tree *lzpNow = szpTop;

	// 検索
	Tree *lzpNear = NULL;
	while (1) {

		// データ有無
		if (lzpNow == NULL) {
			return lzpNear;
		}

		// 比較
		int liRet = fTreeCmp(lzpNow, plKey);

		// 一致
		if (liRet == 0) {								// 一致
			return lzpNow;
		}

		// 子へ移動
		if (liRet < 0) {								// 左側

			// 対象値に近い値更新
			if (piMode > 0) {								// 以上
				lzpNear = lzpNow;
			}

			lzpNow = lzpNow->mzp1Child[D_TREE_LEFT];
		}
		else {											// 右側

			// 対象値に近い値更新
			if (piMode < 0) {								// 以下
				lzpNear = lzpNow;
			}

			lzpNow = lzpNow->mzp1Child[D_TREE_RIGHT];
		}
	}

	return NULL;
}

// 木データ - 検索 - 一番端
Tree *
fTreeSrhTip(
	Tree *pzpTree				// <I> 検索開始位置
	, int piWay					// <I> 移動方向
)
{
	// データ有無
	if (pzpTree == NULL) {
		return NULL;
	}

	// 行けるところまで移動
	while (pzpTree->mzp1Child[piWay] != NULL) {
		pzpTree = pzpTree->mzp1Child[piWay];
	}

	return pzpTree;
}

// 木データ - 高さ取得
int
fTreeGetHeight(
	Tree *pzpTree				// <I> 対象の木情報
)
{
	// データ有無
	if (pzpTree == NULL) {
		return 0;
	}

	if (pzpTree->mi1Height[D_TREE_LEFT] >= pzpTree->mi1Height[D_TREE_RIGHT]) {
		return pzpTree->mi1Height[D_TREE_LEFT] + 1;
	}
	else {
		return pzpTree->mi1Height[D_TREE_RIGHT] + 1;
	}
}

// 木データ - 右回転(親は子の右下へ・子は親の左上へ)
int
fTreeRttR(
	Tree **pzppTree				// <I> 回転対象
)
{
	// 現在の子
	Tree *lzpChild = (*pzppTree)->mzp1Child[D_TREE_LEFT];

	// 右回転
	(*pzppTree)->mzp1Child[D_TREE_LEFT] = lzpChild->mzp1Child[D_TREE_RIGHT];	// 親の左側 = 子の右側
	(*pzppTree)->mi1Height[D_TREE_LEFT] = lzpChild->mi1Height[D_TREE_RIGHT];	// 親の高さ(左) = 子の高さ(右)
	lzpChild->mzp1Child[D_TREE_RIGHT] = *pzppTree;								// 子の右側 = 親
	lzpChild->mi1Height[D_TREE_RIGHT] = fTreeGetHeight(*pzppTree);				// 子の高さ(右) - 親の高さ
	*pzppTree = lzpChild;														// 親 = 子

	return 0;
}

// 木データ - 左回転(親は子の左下へ・子は親の右上へ)
int
fTreeRttL(
	Tree **pzppTree				// <I> 回転対象
)
{
	// 現在の子
	Tree *lzpChild = (*pzppTree)->mzp1Child[D_TREE_RIGHT];

	// 左回転
	(*pzppTree)->mzp1Child[D_TREE_RIGHT] = lzpChild->mzp1Child[D_TREE_LEFT];	// 親の右側 = 子の左側
	(*pzppTree)->mi1Height[D_TREE_RIGHT] = lzpChild->mi1Height[D_TREE_LEFT];	// 親の高さ(右) = 子の高さ(左)
	lzpChild->mzp1Child[D_TREE_LEFT] = *pzppTree;								// 子の左側 = 親
	lzpChild->mi1Height[D_TREE_LEFT] = fTreeGetHeight(*pzppTree);				// 子の高さ(左) - 親の高さ
	*pzppTree = lzpChild;														// 親 = 子

	return 0;
}

// 木データ - 追加・削除の共通処理
// 戻り値：[1]高さの変更あり [0]高さの変更なし
int
fTreeComAddDel(
	Tree **pzppNow				// <I> 現在の木情報
	, int piWay					// <I> 対象の方向
)
{
	// 高さの変更があるかチェック
	int liNew = fTreeGetHeight((*pzppNow)->mzp1Child[piWay]);
	if ((*pzppNow)->mi1Height[piWay] == liNew) {												// 変化なし
		return 0;
	}
	(*pzppNow)->mi1Height[piWay] = liNew;														// 更新

	// 高さが離れている場合、回転
	if ((*pzppNow)->mi1Height[D_TREE_LEFT] - (*pzppNow)->mi1Height[D_TREE_RIGHT] > 1) {			// 左側が高い
		fTreeRttR(pzppNow);																			// 右回転
	}
	else if ((*pzppNow)->mi1Height[D_TREE_RIGHT] - (*pzppNow)->mi1Height[D_TREE_LEFT] > 1) {	// 右側が高い
		fTreeRttL(pzppNow);																			// 左回転
	}

	return 1;
}

// 木データ - 追加
// 戻り値：[1]高さの変更あり [0]高さの変更なし [-1]追加なし
int
fTreeAdd(
	Tree **pzppNow				// <I> 現在の木情報
	, long long plKey			// <I> キー
)
{
	// 作成
	if (*pzppNow == NULL) {
		*pzppNow = fTreeMake(plKey);
		return 1;
	}

	// 比較
	int liRet = fTreeCmp(*pzppNow, plKey);

	// 一致
	if (liRet == 0) {
		return -1;
	}

	// 方向の判別
	int liWay;
	if (liRet < 0) {													// 左側
		liWay = D_TREE_LEFT;
	}
	else {																// 右側
		liWay = D_TREE_RIGHT;
	}

	// 下位へ
	liRet = fTreeAdd(&((*pzppNow)->mzp1Child[liWay]), plKey);
	if (liRet < 1) {													// 高さの変更なし or 追加なし
		return liRet;
	}

	// 追加・削除の共通処理
	return fTreeComAddDel(pzppNow, liWay);
}

// 木データ - 削除の共通処理
// 戻り値：[1]高さの変更あり [0]高さの変更なし
int
fTreeDelCom(
	Tree **pzppNow				// <I> 削除対象
)
{
	// 新しい子を取得
	Tree *lzpNew;
	if ((*pzppNow)->mzp1Child[D_TREE_LEFT] == NULL) {						// 左側なし
		if ((*pzppNow)->mzp1Child[D_TREE_RIGHT] == NULL) {						// 右側なし
			lzpNew = NULL;
		}
		else {																	// 右側あり
			lzpNew = (*pzppNow)->mzp1Child[D_TREE_RIGHT];
		}
	}
	else {																	// 左側あり
		if ((*pzppNow)->mzp1Child[D_TREE_RIGHT] == NULL) {						// 右側なし
			lzpNew = (*pzppNow)->mzp1Child[D_TREE_LEFT];
		}
		else {																	// 右側あり

			// 両側がある場合、左側の最大値で削除を更新する
			// 現在値(回転によりアドレス変更される場合を考慮し、保持しておく)
			Tree *lzpNow = *pzppNow;

			// 左側の最大値を取得
			Tree *lzpMax = fTreeSrhTip(lzpNow->mzp1Child[D_TREE_LEFT], D_TREE_RIGHT);
			long long llKey = lzpMax->mlKey;

			// 左側の最大値を削除
			fTreeDel(&szpTop, llKey);												// 回転する場合を考慮し、引数は先頭から

			// 左側の最大値で更新
			lzpNow->mlKey = llKey;

			return 0;
		}
	}

	// 自分を空き領域へ移動
	(*pzppNow)->mzp1Child[D_TREE_LEFT] = szpFree;
	szpFree = *pzppNow;

	// 新しい子で更新
	*pzppNow = lzpNew;

	return 1;
}

// 木データ - 削除
// 戻り値：[1]高さの変更あり [0]高さの変更なし [-1]削除なし
int
fTreeDel(
	Tree **pzppNow				// <I> 現在の木情報
	, long long plKey			// <I> キー
)
{
	// データ有無
	if (*pzppNow == NULL) {
		return -1;
	}

	// 比較
	int liRet = fTreeCmp(*pzppNow, plKey);

	// 一致
	if (liRet == 0) {
		return fTreeDelCom(pzppNow);											// 削除
	}

	// 方向の判別
	int liWay;
	if (liRet < 0) {														// 左側
		liWay = D_TREE_LEFT;
	}
	else {																	// 右側
		liWay = D_TREE_RIGHT;
	}

	// 下位へ
	liRet = fTreeDel(&((*pzppNow)->mzp1Child[liWay]), plKey);
	if (liRet != 1) {														// 高さの変更あり以外
		return liRet;
	}

	// 追加・削除の共通処理
	return fTreeComAddDel(pzppNow, liWay);
}

// 実行メイン
int
fMain(
)
{
	int i, liWork;
	char lc1Buf[1024];

	// 仕事数 - 取得
	int liWCnt;
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d", &liWork, &liWCnt);

	// 仕事 - 取得
	long long llMax = 0;
	for (i = 0; i < liWCnt; i++) {
		long long llStart, llEnd;
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
		sscanf(lc1Buf, "%lld%lld", &llStart, &llEnd);
		llStart = llStart * 2;
		llEnd = llEnd * 2 + 1;

		// 開始日以前
		int liAdd = 1;
		Tree *lzpTree = fTreeSrhRng(-1, llStart);
		if (lzpTree != NULL) {
			if (lzpTree->mlKey == llStart) {				// 同日
				liAdd = 0;
			}
			else if (lzpTree->mlKey == llStart - 1) {		// 前日(終了日)
				liAdd = 0;
				fTreeDel(&szpTop, llStart - 1);
				lzpTree = fTreeSrhRng(-1, llStart - 2);
				llStart = lzpTree->mlKey;
			}
			else if (lzpTree->mlKey % 2 == 0) {				// 別の仕事内
				liAdd = 0;
				llStart = lzpTree->mlKey;
			}
		}
		if (liAdd == 1) {
			fTreeAdd(&szpTop, llStart);
		}

		// 終了日以降
		liAdd = 1;
		lzpTree = fTreeSrhRng(1, llEnd);
		if (lzpTree != NULL) {
			if (lzpTree->mlKey == llEnd) {					// 同日
				liAdd = 0;
			}
			else if (lzpTree->mlKey == llEnd + 1) {			// 翌日(開始日)
				liAdd = 0;
				fTreeDel(&szpTop, llEnd + 1);
				lzpTree = fTreeSrhRng(1, llEnd + 2);
				llEnd = lzpTree->mlKey;
			}
			else if (lzpTree->mlKey % 2 == 1) {				// 別の仕事内
				liAdd = 0;
				llEnd = lzpTree->mlKey;
			}
		}
		if (liAdd == 1) {
			fTreeAdd(&szpTop, llEnd);
		}

		// 今回の仕事内
		while (1) {
			lzpTree = fTreeSrhRng(1, llStart + 1);
			if (lzpTree->mlKey == llEnd) {
				break;
			}
			fTreeDel(&szpTop, lzpTree->mlKey);
		}

		// 今回の仕事期間
		long long llCnt = (llEnd - llStart + 1) / 2;
		if (llMax < llCnt) {
			llMax = llCnt;
		}

		// 出力
		sprintf(lc1Buf, "%lld\n", llMax);
		fOut(lc1Buf);
	}

	return 0;
}

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

	// データ - 初期化
	siTCnt = 0;													// 木の実データ数
	szpTop = NULL;												// 先頭の木データ
	szpFree = NULL;												// 空きデータ

	// 入力 - セット
#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();

	// 残データ有無
#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

結果

テストケース

ソースコード
