結果

問題 No.307 最近色塗る問題多くない?
ユーザー koyumeishikoyumeishi
提出日時 2015-09-21 23:09:36
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 184 ms / 4,000 ms
コード長 1,478 bytes
コンパイル時間 773 ms
コンパイル使用メモリ 81,484 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-21 21:43:23
合計ジャッジ時間 3,799 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;

int dx[] = {0,0,1,-1,0,0};
int* dy = dx+2;

int bfs(vector<vector<int>>& A, int r, int c, int x){
	int h = A.size();
	int w = A[0].size();

	int tmp = A[r][c];

	int ret = 1;

	queue<int> q;
	q.push(r*w+c);
	A[r][c] = x;

	while(q.size()){
		int pos = q.front();
		q.pop();
		r = pos/w;
		c = pos%w;
		for(int k=0; k<4; k++){
			int r_ = r + dy[k];
			int c_ = c + dx[k];
			if(r_<0 || r_>=h || c_<0 || c_>=w) continue;
			if(A[r_][c_] != tmp) continue;
			A[r_][c_] = x;
			ret++;
			q.push(r_*w + c_);
		}
	}

	return ret;
}


#include <ctime>

int main(){
	auto start = clock();
	int h,w;
	cin >> h >> w;
	vector<vector<int>> A(h, vector<int>(w));
	
	for(int i=0; i<h; i++){
		for(int j=0; j<w; j++){
			cin >> A[i][j];
		}
	}

	bool filled = false;
	int color = -1;

	int q;
	cin >> q;
	for(int i=0; i<q; i++){
		int r,c,x;
		cin >> r >> c >> x;
		r--; c--;

		color = x;

		if(A[r][c] == x) continue;
		if(filled || bfs(A, r,c, x) == h*w){
			filled = true;
		}
	}

	if(filled){
		for(int i=0; i<h; i++){
			for(int j=0; j<w; j++){
				printf("%d%s", color, (j==w-1)?"\n":" ");
			}
		}
	}else{
		for(int i=0; i<h; i++){
			for(int j=0; j<w; j++){
				printf("%d%s", A[i][j], (j==w-1)?"\n":" ");
			}
		}
	}

	auto end = clock();
	//cerr << end-start << endl;

	return 0;
}
0