BZOJ 3648 寝室管理

【题解】

　　GDOI2016 Day2T3

　　如果给出的数据是一棵树那么皆大欢喜直接点分治就好了，用树状数组维护大于x的数的个数。如果是一棵基环树，我们先断掉环上的一条边，然后跑点分治；再加上经过这条边的方案数，其实就是从断掉的那条边的一个端点开始，把环遍历一遍，更新贡献的方式跟点分治过程中的差不多，具体看注释2333。

　　

 #include<cstdio>
 #include<algorithm>
 #define rg register
 #define N 100010
 #define LL long long
 using namespace std;
 int n,m,k,tot,cnt,top,cir,root;
 int last[N],dep[N],bit[N<<],siz[N],mxsiz[N],st[N],c[N];
 bool cut[N<<],v[N];
 LL ans=;
 struct edge{
     int to,pre,dis;
 }e[N<<];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     return k*f;
 }
 inline void add(int x,int y){for(;x>;x-=(x&-x))bit[x]+=y;}
 inline int query(int x){
     if(x<) x=; int ret=;
     for(;x<=n+cir;x+=(x&-x)) ret+=bit[x]; return ret;
 }
 void findcircle(int x,int fa){
     if(cir) return;
     if(v[x]){
         c[cir=]=x;
         for(rg int i=top;st[i]!=x;c[++cir]=st[i--]);
     }
     v[x]=; st[++top]=x;
     for(rg int i=last[x],to;i;i=e[i].pre)
         if((to=e[i].to)!=fa) findcircle(to,x);
     v[x]=; top--;
 }
 void getroot(int x,int fa){
     siz[x]=; mxsiz[x]=;
     for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]&&to!=fa&&!cut[i]){
         getroot(to,x); siz[x]+=siz[to];
         mxsiz[x]=max(mxsiz[x],siz[to]);
     }
     mxsiz[x]=max(mxsiz[x],cnt-mxsiz[x]);
     if(!root||mxsiz[x]<mxsiz[root]) root=x;
 }
 void getdep(int x,int fa,int d){
     st[++top]=d;
     for(rg int i=last[x],to;i;i=e[i].pre)
     if(!v[to=e[i].to]&&to!=fa&&!cut[i]) getdep(to,x,d+);
 }
 void dfs(int x){
     int l=; v[x]=;
     for(rg int i=last[x],to;i;i=e[i].pre)
     if(!v[to=e[i].to]&&!cut[i]){
         l=top+; getdep(to,x,);
         for(rg int j=l;j<=top;j++) ans+=query(k-st[j]-);
         for(rg int j=l;j<=top;j++) add(st[j],);
     }
     ans+=query(k-);
     while(top) add(st[top--],-);
     for(rg int i=last[x],to;i;i=e[i].pre)
     if(!v[to=e[i].to]&&!cut[i]&&siz[to]>=k){
         cnt=siz[to]; root=; getroot(to,x); dfs(root);
     }
 }
 int main(){
     n=read(); m=read(); k=read();
     for(rg int i=;i<=m;i++){
         int u=read(),v=read();
         e[++tot]=(edge){v,last[u]}; last[u]=tot;
         e[++tot]=(edge){u,last[v]}; last[v]=tot;
     }
     if(m<n){
         cnt=n; root=; getroot(,); dfs(root);
     }
     else{
         findcircle(,);
         for(rg int i=last[c[]],to;i;i=e[i].pre)if((to=e[i].to)==c[cir]){
             cut[i]=; break;
         }
         for(rg int i=last[c[cir]],to;i;i=e[i].pre)if((to=e[i].to)==c[]){
             cut[i]=; break;
         }
         top=; cnt=n; root=;
         getroot(,); dfs(root);
         top=;
         for(rg int i=;i<=n;i++)v[i]=;
         for(rg int i=;i<=cir+n;i++) bit[i]=;
         for(rg int i=;i<=cir;i++) v[c[i]]=;
         for(rg int i=;i<=cir;i++){
             v[c[i]]=;
             getdep(c[i],,);
             v[c[i]]=;
             for(rg int j=;j<=top;j++)ans+=query(k-(cir-i+)-st[j]);//cir-i+1是绕一圈到环上起点的距离
             while(top) add(st[top--]+i,);//+i就是加上到环上起点的距离
         }
     }
     printf("%lld\n",ans);
     return ;
 }