#315033 (C) No.307 最近色塗る問題多くない？

提出ソース

結果

問題	No.307 最近色塗る問題多くない？
ユーザー	asugen0402
提出日時	2019-02-08 11:36:46
言語	C (gcc 12.3.0)
結果	WA
実行時間	-
コード長	6,082 bytes
コンパイル時間	453 ms
コンパイル使用メモリ	31,828 KB
実行使用メモリ	4,508 KB
最終ジャッジ日時	2023-09-09 23:29:50
合計ジャッジ時間	3,264 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,380 KB
testcase_01	AC	1 ms 4,380 KB
testcase_02	AC	1 ms 4,376 KB
testcase_03	AC	1 ms 4,380 KB
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	AC	12 ms 4,380 KB
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	AC	16 ms 4,376 KB
testcase_28	WA	-
testcase_29	AC	9 ms 4,380 KB
testcase_30	WA	-
testcase_31	AC	23 ms 4,376 KB
testcase_32	AC	21 ms 4,380 KB
testcase_33	WA	-
testcase_34	AC	22 ms 4,376 KB
testcase_35	WA	-

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

ソースコード

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_SIZE_MAX		205										// 最大サイズ
#define D_ON			1										// 汎用フラグ - ON
#define D_OFF			0										// 汎用フラグ - OFF

// 内部構造体 - マス情報
typedef struct Mass {
	int miColor;												// 色
	int miDx, miDy;												// 代表マス
	int miDone;													// 処理済フラグ
} Mass;

// 内部変数
static FILE *szpFpI;											// 入力
static Mass sz2Mass[D_SIZE_MAX][D_SIZE_MAX];					// マス
static int siW, siH;											// 幅・高さ

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

// 代表マス - 取得
int
fGetDMass(
	int piTx				// <I> 対象マスX
	, int piTy				// <I> 対象マスY
	, int *pipDx			// <O> 代表マスX
	, int *pipDy			// <O> 代表マスY
)
{
	// 自分が代表かチェック
	if (sz2Mass[piTx][piTy].miDx == piTx) {
		if (sz2Mass[piTx][piTy].miDy == piTy) {
			*pipDx = piTx;
			*pipDy = piTy;
			return 0;
		}
	}

	// 代表マス - 取得
	fGetDMass(sz2Mass[piTx][piTy].miDx, sz2Mass[piTx][piTy].miDy, pipDx, pipDy);

	// 代表マス - 更新
	sz2Mass[piTx][piTy].miDx = *pipDx;
	sz2Mass[piTx][piTy].miDy = *pipDy;

	return 0;
}

// 同一グループ化
int
fSameGrp(
	int piX					// <I> 位置X
	, int piY				// <I> 位置Y
)
{
	// 代表マス - 取得
	int liDx, liDy;
	fGetDMass(piX, piY, &liDx, &liDy);

	// 同色チェック - 上下左右
	if (sz2Mass[piX][piY].miColor == sz2Mass[piX + 1][piY].miColor) {
		sz2Mass[piX + 1][piY].miDx = liDx;
		sz2Mass[piX + 1][piY].miDy = liDy;
	}
	if (sz2Mass[piX][piY].miColor == sz2Mass[piX - 1][piY].miColor) {
		sz2Mass[piX - 1][piY].miDx = liDx;
		sz2Mass[piX - 1][piY].miDy = liDy;
	}
	if (sz2Mass[piX][piY].miColor == sz2Mass[piX][piY + 1].miColor) {
		sz2Mass[piX][piY + 1].miDx = liDx;
		sz2Mass[piX][piY + 1].miDy = liDy;
	}
	if (sz2Mass[piX][piY].miColor == sz2Mass[piX][piY - 1].miColor) {
		sz2Mass[piX][piY - 1].miDx = liDx;
		sz2Mass[piX][piY - 1].miDy = liDy;
	}

	return 0;
}

// 色セット
int
fSetColor(
	int piTx				// <I> 対象位置X
	, int piTy				// <I> 対象位置Y
	, int *pipDx			// <IO> 代表マスX
	, int *pipDy			// <IO> 代表マスY
	, int piColor			// <I> 色
)
{
	// 処理済 - チェック
	if (sz2Mass[piTx][piTy].miDone == D_ON) {
		return 0;
	}

	// 代表マス - 取得
	int liDx, liDy;
	fGetDMass(piTx, piTy, &liDx, &liDy);

	// 代表マス - チェック
	if (liDx == *pipDx && liDy == *pipDy) {				// 一致

		// 色チェック
		if (sz2Mass[piTx][piTy].miColor == piColor) {		// セット済
			return 0;
		}
		sz2Mass[piTx][piTy].miColor = piColor;				// セット

		// 上下左右へ
		fSetColor(piTx + 1, piTy, pipDx, pipDy, piColor);
		fSetColor(piTx - 1, piTy, pipDx, pipDy, piColor);
		fSetColor(piTx, piTy + 1, pipDx, pipDy, piColor);
		fSetColor(piTx, piTy - 1, pipDx, pipDy, piColor);

		// 処理済 - セット
		sz2Mass[piTx][piTy].miDone = D_ON;
	}
	else {												// 不一致

		// 色チェック
		if (sz2Mass[piTx][piTy].miColor != piColor) {		// 範囲外
			return 0;
		}

		// 代表マス - 更新
		sz2Mass[*pipDx][*pipDy].miDx = liDx;
		sz2Mass[*pipDx][*pipDy].miDy = liDy;
		*pipDx = liDx;
		*pipDy = liDy;
	}

	return 0;
}

// 実行メイン
int
fMain(
	int piTNo					// <I> テスト番号 1～
)
{
	int i, j;
	char lc1Buf[1024], lc1Out[1024];

	// データ - 初期化
	memset(sz2Mass, 0, sizeof(sz2Mass));						// マス

	// 入力 - セット
#ifdef D_TEST
	sprintf(lc1Buf, ".\\Test\\T%d.txt", piTNo);
	szpFpI = fopen(lc1Buf, "r");
	sprintf(lc1Buf, ".\\Test\\A%d.txt", piTNo);
	szpFpA = fopen(lc1Buf, "r");
	siRes = 0;
#else
	szpFpI = stdin;
#endif

	// 幅・高さ - 取得
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d", &siH, &siW);

	// マス - 初期化
	for (i = 0; i <= siW + 1; i++) {
		for (j = 0; j <= siH + 1; j++) {
			sz2Mass[i][j].miColor = -1;
			sz2Mass[i][j].miDx = i;
			sz2Mass[i][j].miDy = j;
		}
	}

	// 色 - 取得
	for (j = 1; j <= siH; j++) {
		for (i = 1; i <= siW; i++) {
			fscanf(szpFpI, "%d", &sz2Mass[i][j].miColor);
		}
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	}

	// 同一グループ化
	for (i = 1; i <= siW; i++) {
		for (j = 1; j <= siH; j++) {
			fSameGrp(i, j);
		}
	}

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

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

		// 代表マス - 取得
		int liDx, liDy;
		fGetDMass(liX, liY, &liDx, &liDy);

		// 同色チェック
		if (sz2Mass[liDx][liDy].miColor == liColor) {
			continue;
		}

		// 色セット
		fSetColor(liDx, liDy, &liDx, &liDy, liColor);
	}

	// 色 - 更新
	for (i = 1; i <= siW; i++) {
		for (j = 1; j <= siH; j++) {

			// 代表マス - 取得
			int liDx, liDy;
			fGetDMass(i, j, &liDx, &liDy);

			// 色 - 更新
			sz2Mass[i][j].miColor = sz2Mass[liDx][liDy].miColor;
		}
	}

	// 結果
	for (j = 1; j <= siH; j++) {

		// 結果 - セット
		sprintf(lc1Out, "%d", sz2Mass[1][j].miColor);
		for (i = 2; i <= siW; i++) {
			sprintf(lc1Buf, " %d", sz2Mass[i][j].miColor);
			strcat(lc1Out, lc1Buf);
		}
		strcat(lc1Out, "\n");

		// 結果 - 表示
#ifdef D_TEST
		fgets(lc1Buf, sizeof(lc1Buf), szpFpA);
		if (strcmp(lc1Buf, lc1Out)) {
			siRes = -1;
		}
#else
		printf("%s", lc1Out);
#endif
	}

	// 残データ有無
#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", piTNo);
	}
	else {
		printf("NG %d\n", piTNo);
	}
#endif

	return 0;
}

int
main()
{

#ifdef D_TEST
	int i;
	for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) {
		fMain(i);
	}
#else
	fMain(0);
#endif

	return 0;
}

yukicoder

結果

テストケース

ソースコード