历届试题 合根植物（bfs）

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+100;
const int M=1e6+10;
vector<int> v[M];
//bool g[M][M];
int a[N][N];
bool st[N][N];
int n,m;
int dx[5]={0,0,1,-1};
int dy[5]={1,-1,0,0};
struct P{
	int x,y;
};
bool check(int x,int y){
	for(int i=0;i<v[x].size();i++){
		if(v[x][i]==y) return true;
	}
	return false;
}
void bfs(P u){
	queue<P> q;
	q.push(u);
	st[u.x][u.y]=true;
	while(q.size()){
		P b=q.front();
		q.pop();
		for(int i=0;i<4;i++){
			int xx=b.x+dx[i];
			int yy=b.y+dy[i];
			if(xx<0||xx>=n||yy<0||yy>=m) continue;
			if(st[xx][yy]) continue;
			if(!check(a[b.x][b.y],a[xx][yy])) continue;
			//if(!g[a[b.x][b.y]][a[xx][yy]]) continue;
			st[xx][yy]=true;
			P t={xx,yy};
			q.push(t);	
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	//cout<<n<<" "<<m<<endl; 
	int num=1;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			a[i][j]=num;
			num++;
		}
	}
	int k;scanf("%d",&k);
	//cout<<k<<endl;
	for(int i=0;i<k;i++){
		int x,y;scanf("%d%d",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	int ans=0;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			if(!st[i][j]){
				ans++;
				P t={i,j};
				bfs(t);
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}