結果

問題 No.307 最近色塗る問題多くない？
ユーザー FF256grhy
提出日時 2016-05-08 18:05:54
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 157 ms / 4,000 ms
コード長 2,721 bytes
コンパイル時間 244 ms
コンパイル使用メモリ 27,008 KB
実行使用メモリ 7,016 KB
最終ジャッジ日時 2024-11-21 22:16:53
合計ジャッジ時間 2,356 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:96:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   96 |         scanf("%d%d", &h, &w);
      |         ~~~~~^~~~~~~~~~~~~~~~
main.cpp:99:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   99 |                 scanf("%d", &a[i][j]);
      |                 ~~~~~^~~~~~~~~~~~~~~~
main.cpp:102:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  102 |         scanf("%d", &q);
      |         ~~~~~^~~~~~~~~~
main.cpp:104:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  104 |                 scanf("%d%d%d", &r[i], &c[i], &x[i]);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

// UnionFind のスニペットを作ったので試しに使ってみる

#include <stdio.h>

class UnionFind {
public:
	UnionFind(int size);
	int unite(int v, int w);
	int find(int v);
	int get_size_eq(int v);
	int get_size_part();
	virtual ~UnionFind();

private:
	int *parent;
	int *size_eq;
	int size_part;
};

UnionFind::UnionFind(int size) {
	parent = new int[size];
	size_eq = new int[size];
	for(int i = 0; i < size; i++) { parent[i] = -1; size_eq[i] = 1; }
	size_part = size;
}

int UnionFind::unite(int v, int w) {
	int fv = find(v);
	int fw = find(w);
	if(fv == fw) { return fv; }
	
	size_part--;
	int &sv = size_eq[fv];
	int &sw = size_eq[fw];
	
	if(sv < sw) {
		parent[fv] = fw;
		sw += sv;
		return fw;
	} else {
		parent[fw] = fv;
		sv += sw;
		return fv;
	}
}

int UnionFind::find(int v) {
	if(parent[v] == -1) { return v; }
	return parent[v] = find( parent[v] );
}

int UnionFind::get_size_eq(int v) {
	return size_eq[ find(v) ];
}

int UnionFind::get_size_part() {
	return size_part;
}

UnionFind::~UnionFind() {
	delete[] parent;
	delete[] size_eq;
}




int h, w, a[200][200], q, r[50000], c[50000], x[50000];
UnionFind *uf;
int neighbor[200 * 200], ptr;

int at(int i, int j) {
	return i * w + j;
}

void dfs(int i, int j, int f, int v) {
	if( ! ( 0 <= i && i < h && 0 <= j && j < w) ) { return; }
	if(a[i][j] == v) { return; }
	a[i][j] = v;
	
	int fij = uf -> find( at(i, j) );
	if(fij == f) {
		dfs(i - 1, j    , f, v);
		dfs(i + 1, j    , f, v);
		dfs(i    , j - 1, f, v);
		dfs(i    , j + 1, f, v);
	} else {
		neighbor[ptr] = fij;
		ptr++;
	}
	
	return;
}

int main() {
	scanf("%d%d", &h, &w);
	for(int i = 0; i < h; i++) {
	for(int j = 0; j < w; j++) {
		scanf("%d", &a[i][j]);
	}
	}
	scanf("%d", &q);
	for(int i = 0; i < q; i++) {
		scanf("%d%d%d", &r[i], &c[i], &x[i]);
		r[i]--;
		c[i]--;
	}
	
	
	uf = new UnionFind(h * w);
	
	for(int i = 0; i < h; i++) {
	for(int j = 0; j < w - 1; j++) {
		if(a[i][j] == a[i][j + 1]) { uf -> unite( at(i, j), at(i, j + 1) ); }
	}
	}
	for(int j = 0; j < w; j++) {
	for(int i = 0; i < h - 1; i++) {
		if(a[i][j] == a[i + 1][j]) { uf -> unite( at(i, j), at(i + 1, j) ); }
	}
	}
	
	
	for(int k = 0; k < q; k++) {
		int f = uf -> find( at(r[k], c[k]) );
		int i = f / w;
		int j = f % w;
		if(uf -> get_size_part() == 1) {
			a[i][j] = x[k];
		} else {
			if(a[i][j] % 2 != x[k] % 2) {
				ptr = 0;
				dfs(i, j, f, 2 * (k + 1) + x[k]);
				for(int l = 0; l < ptr; l++) {
					f = uf -> unite(f, neighbor[l]);
				}
				i = f / w;
				j = f % w;
				a[i][j] = x[k];
			}
		}
	}
	
	
	for(int i = 0; i < h; i++) {
	for(int j = 0; j < w; j++) {
		int f = uf -> find( at(i, j) );
		printf("%d ", a[f / w][f % w] );
	} printf("\n");
	}
	
	delete uf;
	
	return 0;
}
0