1. 程式人生 > >2018.12.28-dtoj-3648-寢室管理

2018.12.28-dtoj-3648-寢室管理

題目描述:

T64有一個好朋友,叫T128。T128是寄宿生,並且最近被老師叫過去當宿管了。宿管可不是一件很好做的工作,碰 巧T128有一個工作上的問題想請T64幫忙解決。T128的寢室條件不是很好,所以沒有很多錢來裝修。禮間寢室僅由n -1條雙向道路連線,而且任意兩間寢室之間都可以互達。最近,T128被要求對一條路徑上的所有寢室進行管理,這 條路徑不會重複經過某個點或某條邊。但他不記得是哪條路徑了。他只記得這條路徑上有不少於k個寢室。於是, 他想請T64幫忙數一下,有多少條這樣的路徑滿足條件。嗯…還有一個問題。由於最近有一些熊孩子不準晚上講話 很不爽,他們決定修築一條“情報通道”,如果通道建成,寢室就變成了一個N個點N條邊的無向圖。並且,經過“ 情報通道”的路徑也是合法的。T128心想:通道建成之前,T64還有一個高效的演算法幫我數路徑條數,但是通道建 成之後,他還有辦法嗎?對,T64手忙腳亂,根本數不清有多少條路徑。於是他找到了你

演算法標籤:點分治

思路:

當m==n-1時是裸的點分治,考慮再加一條邊,統計對答案的貢獻,對於加邊後構成的環,依此列舉斷掉每一條邊的貢獻,怎麼斷不會算重的問題看程式碼吧。

以下程式碼:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
const int N=1e5+5,M=2e5+5;
int k,m,w,r,c,n,t[M],l[M],h[N],f[N],p[N],s[N],d[N];
bool b[N];ll a[N],q; int get(int x){return !f[x]?x:f[x]=get(f[x]);} void insert(int x,int y){ t[++c]=y,l[c]=h[x],h[x]=c; t[++c]=x,l[c]=h[y],h[y]=c; } void add(int x,int y=1){ for(;x<=n;x+=x&-x) if(p[x]!=c) p[x]=c,a[x]=y; else a[x]+=y; } ll sum(int x){
if(x<0) return 0; ll y=0; for(;x;x-=x&-x) if(p[x]==c) y+=a[x]; return y; } #define z t[i] void root(int x,int y){ int c=0;s[x]=1; for(int i=h[x];i;i=l[i]){ if(b[z] || y==z) continue; root(z,x); s[x]+=s[z]; c=max(c,s[z]); } c=max(c,m-s[x]); if(c<w) w=c,r=x; } void dfs1(int x,int y,int w){ s[x]=1;q+=sum(n)-sum(k-w-1); for(int i=h[x];i;i=l[i]){ if(b[z] || y==z) continue; dfs1(z,x,w+1);s[x]+=s[z]; } } void dfs2(int x,int y,int w){ add(w+1); for(int i=h[x];i;i=l[i]) if(!b[z] && y!=z) dfs2(z,x,w+1); } void solve(int x){ ++c;b[x]=1;add(1); for(int i=h[x];i;i=l[i]) if(!b[z]) dfs1(z,x,1),dfs2(z,x,1); for(int i=h[x];i;i=l[i]){ if(b[z]) continue; w=1e9;m=s[z]; root(z,x);solve(r); } } void dfs3(int x){ add(d[x]); for(int i=h[x];i;i=l[i]){ if(z==f[x]) continue; f[z]=x;d[z]=d[x]+1; dfs3(z); } } void dfs4(int x,int y){ add(d[x],-1); for(int i=h[x];i;i=l[i]) if(z!=y && !b[z]) dfs4(z,x); } void dfs5(int x,int y,int w){ q+=sum(n)-sum(k-w-1); for(int i=h[x];i;i=l[i]) if(z!=y && !b[z]) dfs5(z,x,w+1); } int main(){ int x,y,u,v,g=0,h; scanf("%d%d%d",&n,&m,&k); while(m--){ scanf("%d%d",&x,&y); u=get(x),v=get(y); if(u==v) g=x,h=y; else f[u]=v,insert(x,y); } c=0;m=n,w=1e9; root(1,0); solve(r); if(!g) return !printf("%lld\n",q); memset(b,0,n+1); ++c;f[g]=0;d[g]=1;dfs3(g); for(int i=h;i;i=f[i]) b[i]=1; for(int i=h,j=1;i!=g;i=f[i],++j) dfs4(i,0),dfs5(i,0,j); printf("%lld\n",q); return 0; }
View Code