結果

問題 No.697 池の数はいくつか
ユーザー ohreitetsu
提出日時 2018-06-10 19:15:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,014 bytes
コンパイル時間 991 ms
コンパイル使用メモリ 82,608 KB
実行使用メモリ 239,064 KB
最終ジャッジ日時 2024-11-25 13:15:00
合計ジャッジ時間 68,943 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int H,W;
int ans=0;
vector<pair<int,int>> indexList;
bool check(vector<vector<int>> &A,int I,int J,int &index)
{
	if (A[I][J]==1){
		A[I][J]=index;
		if (A[I-1][J]==1){
			return check(A,I-1,J,index);
		}
		if (A[I][J-1]==1){
			return check(A,I,J-1,index);
		}
		if (A[I][J+1]==1){
			return check(A,I,J+1,index);
		}
		if (A[I+1][J]==1){
			return check(A,I+1,J,index);
		}
	}
	return false;
}
void setIndex(vector<vector<int>> &A,int index)
{
	bool loop=true;
	pair<int,int> Index;
	int i,j;
	while (loop){
		loop=false;
		for (int x=0;x<indexList.size();x++){
			Index=indexList[x];
			i=Index.first;
			j=Index.second;
			if (A[i][j]>0){
				if (A[i][j]==index || A[i-1][j]==index ||  A[i][j-1]==index || A[i][j+1]==index || A[i+1][j]==index){
					if (A[i][j]>0 && A[i][j]!=index ){
						A[i][j]=index;
						loop=true;
					}
					if (A[i-1][j]>0 && A[i-1][j]!=index ){
						A[i-1][j]=index;
						loop=true;
					}
					if (A[i][j-1]>0 && A[i][j-1]!=index ){
						A[i][j-1]=index;
						loop=true;
					}
					if (A[i][j+1]>0 && A[i][j+1]!=index ){
						A[i][j+1]=index;
						loop=true;
					}
					if (A[i+1][j]>0 && A[i+1][j]!=index ){
						A[i+1][j]=index;
						loop=true;
					}
				}
			}
		}
	}

}
int main(int argc, char* argv[])
{
	cin>>H>>W;
	int n0=0,n1=0;
	vector<vector<int>> A(H+2,vector<int>(W+2));
	queue<pair<int,int>> q;

	int i,j;
	for (i=1;i<=H;i++){
		for (j=1;j<=W;j++){
			cin>>A[i][j];
			if (A[i][j]==0){
				n0++;
			}else{
				q.push(make_pair(i,j));
				indexList.push_back(make_pair(i,j));
				n1++;
			}
		}
	}
	if (n0==H*W){
		cout<<0<<endl;
		return 0;
	}
	if (n1==H*W){
		cout<<1<<endl;
		return 0;
	}
	int index=2;
	int I=0,J=0;
	while(true){
		bool loop=false;
		if (q.empty()){
			break;
		}
		pair<int,int> p;
		p=q.front();
		q.pop();
		I=p.first;
		J=p.second;
		if (A[I][J]==1){
			check(A,I,J,index);
			setIndex(A,index);
			index++;
			ans++;
		}

	}
	cout<<ans<<endl;
	return 0;
}
0