ACM ICPC 2017 Warmup Contest 1 A. Artwork(逆向+dfs+并查集)

链接：https://nanti.jisuanke.com/t/17410

分析：正向分析给跪了。。逆向考虑的话，先模拟一遍，记录下每个黑点被第一次涂黑的时间，然后按时间倒着来，每次把该时间变黑的那些块变白，然后ans++，然后考虑加了这一块以后连通了某些块（包括刚刚变白的这块），把通过连通减少的减去，更新答案即可。最后一个的答案可以用dfs搞一下，注意直接递归写会爆栈。。

复杂度为O（n*m*q）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<fstream>
#include<stack>
using namespace std;
const int maxn=1001,inf=1e9;
int p[maxn*maxn];
int Find(int x){return p[x]==x?x:p[x]=Find(p[x]);}
int n,m,q,color[maxn][maxn];
int ans[10005];
bool vis[maxn][maxn];
vector<int> t[10005];
void Print(){
            for(int i=1;i<=n;i++){
                for(int j=1;j<=m;j++){
                    int x=Find(i*maxn+j);
                    cout<<x/maxn<<','<<x%maxn<<' ';
                }
                cout<<endl;
            }
}
struct Statu{
    int x,y;
    Statu(int _x,int _y):x(_x),y(_y){}
};
void dfs(int x,int y,int fa){
    stack<Statu> sta;
    sta.push(Statu(x,y));
    while(!sta.empty()){
        x=sta.top().x;y=sta.top().y;
        p[x*maxn+y]=fa;
        sta.pop();
        if(x<n&&!color[x+1][y]&&!vis[x+1][y]){
            vis[x+1][y]=true;sta.push(Statu(x+1,y));
        }
        if(x>1&&!color[x-1][y]&&!vis[x-1][y]){
            vis[x-1][y]=true;sta.push(Statu(x-1,y));
        }
        if(y<m&&!color[x][y+1]&&!vis[x][y+1]){
            vis[x][y+1]=true;sta.push(Statu(x,y+1));
        }
        if(y>1&&!color[x][y-1]&&!vis[x][y-1]){
            vis[x][y-1]=true;sta.push(Statu(x,y-1));
        }
    }
}
int main(){
    scanf("%d%d%d",&n,&m,&q);
    int x1,y1,x2,y2;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++)
            p[i*maxn+j]=i*maxn+j;
    }
    for(int i=0;i<q;i++){
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        if(y1==y2){
            for(int j=x1;j<=x2;j++){
                if(!color[j][y1]){
                    t[i].push_back(j*maxn+y1);
                    color[j][y1]=1;
                }
            }
        }else{
            for(int j=y1;j<=y2;j++){
                if(!color[x1][j]){
                    t[i].push_back(x1*maxn+j);
                    color[x1][j]=1;
                }
            }
        }
    }
    ans[q]=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(color[i][j]||vis[i][j])continue;
            dfs(i,j,i*maxn+j);ans[q]++;
        }
    }
    for(int T=q-1;T>0;T--){
        ans[T]=ans[T+1];
        for(int i=0;i<t[T].size();i++){
            int x=t[T][i]/maxn,y=t[T][i]%maxn,count=0;
            color[x][y]=0;
            int Fa=t[T][i],H;
            ans[T]++;
            if(x>1&&!color[x-1][y]){
                H=Find((x-1)*maxn+y);
                if(H!=Fa){
                    ans[T]--;
                    p[H]=Fa;
                }
                count++;
            }
            if(x<n&&!color[x+1][y]){
                H=Find((x+1)*maxn+y);
                if(H!=Fa){
                    ans[T]--;
                    p[H]=Fa;
                }
                count++;
            }
            if(y>1&&!color[x][y-1]){
                H=Find(x*maxn+y-1);
                if(H!=Fa){
                    ans[T]--;
                    p[H]=Fa;
                }
                count++;
            }
            if(y<m&&!color[x][y+1]){
                H=Find(x*maxn+y+1);
                if(H!=Fa){
                    ans[T]--;
                    p[H]=Fa;
                }
                count++;
            }
//            cout<<x<<' '<<y<<endl;
//            Print();
        }
    }
    for(int i=1;i<=q;i++){
        printf("%d
",ans[i]);
//        if(ans[i]!=ANS[i])cout<<"WA"<<endl;
    }
        //printf("%d
",ans[i]);
    return 0;
}