CF811E Vladik and Entertaining Flags（线段树+并查集）

这题不错，jy神犇一语中的，道破此题本质。%%%
然而维护的信息还是比较需要细节的。然而写完bzoj1018这题简直就是小菜一碟~因为n很小，所以我们考虑用线段树维护每一列的信息。区间合并的时候我们需要知道合在一起的两列的联通情况，因此对于每个节点我们需要记录l[n],r[n]，区间左右端点的连通信息。为了查询我们还需要知道区间连通块个数。因此再维护一个tot。考虑区间合并，我们可以用并查集来O(n)的实现，别忘了再重新维护合并后的区间左右端点的联通情况。总的复杂度O(n(mlogm+qlogm))

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,q,mp[11][N],fa[N*11],num=0;
struct data{
    int l[11],r[11],tot;
};
struct node{
    data x;
}tree[N<<2];
inline int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
data merge(int mid,data l,data r){
    data res;res.tot=l.tot+r.tot;
    for(int i=1;i<=n;++i){
        fa[l.r[i]]=l.r[i];fa[l.l[i]]=l.l[i];
        fa[r.l[i]]=r.l[i];fa[r.r[i]]=r.r[i];
    }
    for(int i=1;i<=n;++i)
        if(mp[i][mid]==mp[i][mid+1]){
            int xx=find(l.r[i]),yy=find(r.l[i]);
            if(xx!=yy) fa[xx]=yy,res.tot--;
        }
    for(int i=1;i<=n;++i) res.l[i]=find(l.l[i]),res.r[i]=find(r.r[i]);
    return res;
}
void build(int p,int l,int r){
    if(l==r){
        data& x=tree[p].x;x.tot=n;
        for(int i=1;i<=n;++i)
            if(mp[i][l]==mp[i-1][l]) x.l[i]=x.r[i]=num,x.tot--;
            else x.l[i]=x.r[i]=++num;return;
    }int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
    tree[p].x=merge(mid,tree[p<<1].x,tree[p<<1|1].x);
}
data ask(int p,int l,int r,int x,int y){
    if(x<=l&&r<=y) return tree[p].x;
    int mid=l+r>>1;
    if(y<=mid) return ask(p<<1,l,mid,x,y);
    if(x>mid) return ask(p<<1|1,mid+1,r,x,y);
    return merge(mid,ask(p<<1,l,mid,x,y),ask(p<<1|1,mid+1,r,x,y));
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();q=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j) mp[i][j]=read();
    build(1,1,m);
    while(q--){
        int x=read(),y=read();
        data res=ask(1,1,m,x,y);
        printf("%d\n",res.tot);
    }
    return 0;
}