結果

問題 No.833 かっこいい電車
コンテスト
ユーザー asugen0402
提出日時 2019-05-29 17:05:30
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 81 ms / 2,000 ms
コード長 8,562 bytes
コンパイル時間 827 ms
コンパイル使用メモリ 34,304 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2024-07-02 03:55:14
合計ジャッジ時間 3,389 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

// 内部定数
#define D_SEGT_CNT		131072									// セグメントツリーデータ数(2の累乗)
#define D_SEGT_KIND		3										// セグメントツリーデータ種類数
#define D_SEGT_SUM		0										// セグメントツリーデータ種類 - 範囲取得 - かっこよさ合計
#define D_SEGT_MIN		1										// セグメントツリーデータ種類 - 範囲更新 - 最小車両
#define D_SEGT_MAX		2										// セグメントツリーデータ種類 - 範囲更新 - 最大車両

// 内部構造体 - セグメントツリー情報
typedef struct SegT {
	long long ml1Val[D_SEGT_KIND];								// 値
} SegT;

// 内部変数
static FILE *szpFpI;											// 入力
static SegT sz1SegT[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
fSegTSetCSNo(
	int piCnt					// <I> 子データ数
)
{
	siSCNo = 1;
	while (siSCNo < piCnt) {
		siSCNo *= 2;
	}

	return 0;
}

// セグメントツリー - 親データセット
int
fSegTSetPData(
)
{
	int i;

	// 最初の親番号範囲
	int liPNo1 = siSCNo / 2;
	int liPNo2 = siSCNo - 1;

	// 作成
	while (liPNo1 > 0) {
		for (i = liPNo1; i <= liPNo2; i++) {

			// 対象値を取得
			long long llVal = sz1SegT[i * 2].ml1Val[D_SEGT_SUM] + sz1SegT[i * 2 + 1].ml1Val[D_SEGT_SUM];

			// セット
			sz1SegT[i].ml1Val[D_SEGT_SUM] = llVal;
		}

		// 次の親番号範囲
		liPNo2 = liPNo1 - 1;
		liPNo1 /= 2;
	}

	return 0;
}

// セグメントツリー - 範囲取得
long long
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 sz1SegT[piNNo].ml1Val[D_SEGT_SUM];
	}

	// 中間位置
	int liCenter = (piNowS + piNowE) / 2;

	// 初期値
	long long llVal = 0;

	// 左側
	if (piGetS <= liCenter) {
		llVal = fSegTGetRngMain(piGetS, piGetE, piNNo * 2, piNowS, liCenter);
	}

	// 右側
	if (piGetE >= liCenter + 1) {
		llVal += fSegTGetRngMain(piGetS, piGetE, piNNo * 2 + 1, liCenter + 1, piNowE);
	}

	return llVal;
}
long long
fSegTGetRng(
	int piGetS					// <I> 取得範囲 - 開始 0～
	, int piGetE				// <I> 取得範囲 - 終了 0～
)
{
	return fSegTGetRngMain(piGetS, piGetE, 1, 0, siSCNo - 1);
}

// セグメントツリー - １データ加算
int
fSegTUpOne(
	int piUpNo					// <I> 更新位置 0～
)
{
	// 子番号
	int liCNo1 = siSCNo + piUpNo;
	int liCNo2;
	if (liCNo1 % 2 == 0) {
		liCNo2 = liCNo1 + 1;
	}
	else {
		liCNo2 = liCNo1 - 1;
	}

	// 加算
	sz1SegT[liCNo1].ml1Val[D_SEGT_SUM]++;

	// 親を更新
	while (1) {

		// 親番号
		int liPNo = liCNo1 / 2;
		if (liPNo < 1) {
			break;
		}

		// 対象値を取得
		long long llVal = sz1SegT[liCNo1].ml1Val[D_SEGT_SUM] + sz1SegT[liCNo2].ml1Val[D_SEGT_SUM];

		// 更新
		sz1SegT[liPNo].ml1Val[D_SEGT_SUM] = llVal;

		// 次の子番号
		liCNo1 = liPNo;
		if (liCNo1 % 2 == 0) {
			liCNo2 = liCNo1 + 1;
		}
		else {
			liCNo2 = liCNo1 - 1;
		}
	}

	return 0;
}

// セグメントツリー - 範囲更新
int
fSegTUpRngMain(
	int piKind					// <I> 種類
	, int piUpS					// <I> 更新範囲 - 開始 0～
	, int piUpE					// <I> 更新範囲 - 終了 0～
	, int piUpVal				// <I> 更新値
	, int piNNo					// <I> 現在位置 1～
	, int piNowS				// <I> 現在範囲 - 開始 0～
	, int piNowE				// <I> 現在範囲 - 終了 0～
)
{
	// 範囲外チェック
	if (piUpE < piNowS || piNowE < piUpS) {
		return 0;
	}

	// 範囲内なら更新
	if (piUpS <= piNowS && piNowE <= piUpE) {
		sz1SegT[piNNo].ml1Val[piKind] = piUpVal;
		return 0;
	}

	// 更新済みの場合、子に展開しておく
	if (sz1SegT[piNNo].ml1Val[piKind] != 0) {
		sz1SegT[piNNo * 2].ml1Val[piKind] = sz1SegT[piNNo].ml1Val[piKind];
		sz1SegT[piNNo * 2 + 1].ml1Val[piKind] = sz1SegT[piNNo].ml1Val[piKind];
		sz1SegT[piNNo].ml1Val[piKind] = 0;
	}

	// 範囲の半分
	int liHalf = (piNowE - piNowS + 1) / 2;

	// 両側の子へ
	fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, piNNo * 2, piNowS, piNowS + liHalf - 1);
	fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, piNNo * 2 + 1, piNowE - liHalf + 1, piNowE);

	return 0;
}
int
fSegTUpRng(
	int piKind					// <I> 種類
	, int piUpS					// <I> 更新範囲 - 開始 0～
	, int piUpE					// <I> 更新範囲 - 終了 0～
	, int piUpVal				// <I> 更新値
)
{
	return fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, 1, 0, siSCNo - 1);
}

// セグメントツリー - １データ取得
int
fSegTGetOne(
	int piKind					// <I> 種類
	, int piGNo					// <I> 取得位置 0～
)
{
	int liRet;

	// 対象位置
	piGNo += siSCNo;

	// 初期値
	liRet = (int)sz1SegT[piGNo].ml1Val[piKind];

	// 親に値があれば更新
	while (piGNo > 1) {
		piGNo /= 2;										// 上位へ
		if (sz1SegT[piGNo].ml1Val[piKind] != 0) {		// 更新済み
			liRet = (int)sz1SegT[piGNo].ml1Val[piKind];
		}
	}

	return liRet;
}

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

	// 車両数・クエリ数 - 取得
	int liCCnt, liQCnt;
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d", &liCCnt, &liQCnt);

	// セグメントツリー - 子の開始位置 - セット
	fSegTSetCSNo(liCCnt);

	// かっこよさ - 取得
	for (i = 0; i < liCCnt; i++) {
		fscanf(szpFpI, "%lld", &sz1SegT[siSCNo + i].ml1Val[D_SEGT_SUM]);
	}
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);

	// セグメントツリー - 親データセット
	fSegTSetPData();

	// 最小車両・最大車両 - 初期化
	for (i = 0; i < liCCnt; i++) {
		sz1SegT[siSCNo + i].ml1Val[D_SEGT_MIN] = i + 1;
		sz1SegT[siSCNo + i].ml1Val[D_SEGT_MAX] = i + 1;
	}

	// クエリ - 取得
	for (i = 0; i < liQCnt; i++) {
		int liOpe, liCNo;
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
		sscanf(lc1Buf, "%d%d", &liOpe, &liCNo);

		// クエリ - 実行
		int liMin, liMax;
		switch (liOpe) {
		case 1:												// 車両結合
			liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1);
			if (liMax <= liCNo) {
				liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1);
				liMax = fSegTGetOne(D_SEGT_MAX, liCNo);
				fSegTUpRng(D_SEGT_MIN, liMin - 1, liMax - 1, liMin);
				fSegTUpRng(D_SEGT_MAX, liMin - 1, liMax - 1, liMax);
			}
			break;

		case 2:												// 車両分割
			liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1);
			if (liMax > liCNo) {
				liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1);
				fSegTUpRng(D_SEGT_MAX, liMin - 1, liCNo - 1, liCNo);
				liMax = fSegTGetOne(D_SEGT_MAX, liCNo);
				fSegTUpRng(D_SEGT_MIN, liCNo, liMax - 1, liCNo + 1);
			}
			break;

		case 3:												// かっこよさ増加
			fSegTUpOne(liCNo - 1);
			break;

		case 4:												// かっこよさ表示
			liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1);
			liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1);
			long long llSum = fSegTGetRng(liMin - 1, liMax - 1);
			sprintf(lc1Buf, "%lld\n", llSum);
			fOut(lc1Buf);
			break;
		}
	}

	return 0;
}

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

	// データ - 初期化
	memset(sz1SegT, 0, sizeof(sz1SegT));						// セグメントツリー

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