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 100010inline 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;}