A - Artwork Gym - 101550A 并查集

题目描述：
有一个n*m的格子，有q个查询，每次涂一条线的格子，问你每次后联通块的数目

分析：
可以倒着想，先把q次操作涂满，然后求出联通块的数目，每次删掉格子后，如果这个格子还是黑色，不处理，如果是白色
1）如果四周都是黑色，联通块数目加一
2）如果有白色，每次合并，第一次不同没事，以后每次不同联通块数目减一

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <set>
#include <vector>
using namespace std;
const int maxn = 1000 + 10 ;
int num[maxn][maxn];
int far[maxn*maxn],ran[maxn*maxn];
bool vis[maxn][maxn];
int dx[4]= {0,0,1,-1};
int dy[4]= {1,-1,0,0};
void init()
{
    for(int i=0; i<maxn*maxn; i++)
    {
        far[i] = i;
        ran[i] = 0;
    }
}
int find_(int x)
{
    if(x==far[x]) return far[x];
    return far[x]=find_(far[x]);
}
void unite(int x,int y)
{
    x = find_(x);
    y = find_(y);
    if(x==y) return ;
    if(ran[x] < ran[y])
    {
        far[x]=y;
    }
    else
    {
        far[y]=x;
        if(ran[x]==ran[y]) ran[x]++;
    }
}
bool same(int x,int y)
{
    return find_(x)==find_(y);
}
int n,m,q;
int ans[maxn*10];
struct node
{
    int x1,y1,x2,y2;
} s[maxn*10];
void solve1(int x,int y1,int y2)
{
    for(int i = min(y1,y2); i <= max(y1,y2); i++)
        num[x][i]++;
}
void solve2(int y,int x1,int x2)
{
    for(int i = min(x1,x2); i <= max(x1,x2); i++)
        num[i][y]++;
}
bool is_ok(int x,int y)
{
    if(x >= 0&&x < n&&y >= 0&&y < m) return true;
    return false;
}
void dfs(int u,int v,int c)
{
    vis[u][v]=1;
    if(!same(u*m+v,c)) unite(u*m+v,c);
    for(int i = 0; i < 4; i++)
    {
        int x = dx[i] + u;
        int y = dy[i] + v;
        if(is_ok(x,y)&&!vis[x][y]&&!num[x][y])
        {
            dfs(x,y,c);
        }
    }
}
void solve(int x,int y,int q1)
{
    num[x][y]--;
    if(num[x][y])
    {
        return ;
    }
    else
    {
        bool flag = false;
        int is_first = 0;
        for(int i = 0; i < 4; i++)
        {
            int x1 = x + dx[i];
            int y1 = y + dy[i];
            if(is_ok(x1,y1))
            {
                if(!num[x1][y1])
                {
                    flag = true;
                    if(!same(x1*m+y1,x*m+y))
                    {
                        unite(x1*m+y1,x*m+y);
                        is_first++;
                        if(is_first==1) continue;
                        else ans[q1]--;

                    }
                }
            }
        }
        if(!flag) ans[q1]++;
    }
}
int main()
{
    scanf("%d%d%d",&m,&n,&q);
    init();
    for(int i = 0; i < q; i++)
    {
        scanf("%d%d%d%d",&s[i].y1,&s[i].x1,&s[i].y2,&s[i].x2);
        s[i].y1--;
        s[i].x1--;
        s[i].y2--;
        s[i].x2--;
        if(s[i].x1==s[i].x2)
        {
            solve1(s[i].x1,s[i].y1,s[i].y2);
        }
        else
        {
            solve2(s[i].y1,s[i].x1,s[i].x2);
        }
    }
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < m; j++)
        {
            if(!vis[i][j]&&!num[i][j])
            {
                ans[q-1]++;
                dfs(i,j,i*m+j);
            }
        }
    }
    for(int i = q-1; i >= 1; i--)
    {
        int x1 = s[i].x1,y1 = s[i].y1,x2 = s[i].x2,y2 = s[i].y2;
        ans[i-1] = ans[i];
        if(x1==x2)
        {
            for(int j = min(y1,y2); j <= max(y1,y2); j++)
            {
                solve(x1,j,i-1);
            }
        }
        else
        {
            for(int j = min(x1,x2); j <= max(x1,x2); j++)
            {
                solve(j,y1,i-1);
            }
        }
    }
    for(int i = 0 ; i < q; i++)
    {
        printf("%d\n",ans[i]);
    }
    return 0;
}
//