连通块

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

Sample Output

1
1
2
3
2

做的第一道并查集，第一次看还以为直接递归标记就能过，再一看要TLE
一开始一共出了三处错误
b的值没有不断更新
在连通了两个不同的连通块时，没有合并集合
(x-1)*n+y没有-1

#include
#include
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;
}