Description
为了增强幼儿园小朋友的数数能力,小虎的老师给了一个家庭游戏作业。他让小虎拿一块空的围棋盘,随机地在一些方格中放些棋子(有黑白两种颜色),如果一个方格和它的上、下、左、右四个方格之一都有相同颜色的棋子,则认为两格子是相连通的。这期间,要求小虎不断统计共有多少个连通块。
如下图是一个5×9的棋盘,其中“.”表示空格,“*”表示黑棋子,“@”表示白棋子。则有4块连通的棋子块。
.........
..**..@..
.**@@.@@.
..*@..*..
.........
哥哥大虎在一边看一边想,如果棋盘是N×N的,共放了M个棋子,如何用计算机解决这个问题呢?
Input
第1行两个整数:N,m(1≤N≤500,l≤M≤N×N)。
接下来有M行,,每行三个正整数:C,X,Y(0≤C≤1,l≤X,Y≤N),分别表示依次放入棋子的颜色(0表示白色,1表示黑色)、要放入格子的横坐标和格子的纵坐标。
Output
共M行。第i行一个整数,表示放入第i个棋子后,当前有多少个棋子连通块。
Sample Input
3 5
1 1 1
1 1 2
0 2 2
1 3 1
1 2 1
1 1 1
1 1 2
0 2 2
1 3 1
1 2 1
Sample Output
1
1
2
3
2
做的第一道并查集,第一次看还以为直接递归标记就能过,再一看要TLE
一开始一共出了三处错误
b的值没有不断更新
在连通了两个不同的连通块时,没有合并集合
(x-1)*n+y没有-1
1
2
3
2
做的第一道并查集,第一次看还以为直接递归标记就能过,再一看要TLE
一开始一共出了三处错误
b的值没有不断更新
在连通了两个不同的连通块时,没有合并集合
(x-1)*n+y没有-1
#include<stdio.h>
#include<string.h>
int p[250002],cb[502][502];
int find(int x)
{
return p[x]==x?x:p[x]=find(p[x]);
}
int main()
{
freopen("in.txt","r",stdin);
int n,m,i,j,cnt,q,x,y,a,b,mark,step[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(cb,-1,sizeof(cb));
for(i=1;i<250002;i++)
p[i]=i;
cnt=0;
for(i=1;i<=m;i++)
{
mark=0;
scanf("%d%d%d",&q,&x,&y);
cb[x][y]=q;
for(j=0;j<4;j++)
{
int x1,y1;
x1=x+step[j][0];
y1=y+step[j][1];
if(x1>0&&x1<=n&&y1>0&&y1<=n&&cb[x][y]==cb[x1][y1])
{
b=find((x-1)*n+y);
a=find((x1-1)*n+y1);
if(a!=b&&!mark)
{
p[b]=a;
mark=1;
}
else if(a!=b&&mark)
{
cnt--;
p[a]=p[b];
}
}
if(j==3&&!mark)
cnt++;
}
printf("%d\n",cnt);
}
}
return 0;
}