本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作。
本文作者:ljh2000
作者博客:http://www.cnblogs.com/ljh2000-jump/
转载请注明出处,侵权必究,保留最终解释权!
题目链接:BZOJ2117
正解:动态树分治
解题报告:
考虑维护一棵动态树分治的结构树,因为是树高严格$log$的,所以就可以暴力往上跳了。
先建出这棵树,然后维护树分治结构中的所有点到分治重心的距离,以及到分治重心的分治结构父亲的距离。
对于每个点,二分答案,然后计算与他距离$<=mid$的点的个数,往上跳的时候加加减减,容斥一下计算答案。
注意动态树分治的结构与原树大不相同!计算贡献、距离什么的都要仔细考虑!
//It is made by ljh2000 //有志者,事竟成,破釜沉舟,百二秦关终属楚;苦心人,天不负,卧薪尝胆,三千越甲可吞吴。 #include <algorithm> #include <iostream> #include <cstring> #include <vector> #include <cstdio> #include <string> #include <queue> #include <cmath> #include <ctime> #define lc root<<1 #define rc root<<1|1 #define rep(i,j,k) for(int i=j;i<=k;i++) #define reg(i,x) for(int i=first[x];i;i=next[i]) using namespace std; typedef long long LL; const int MAXN = 200011; const int MAXM = 400011; int n,k,ecnt,first[MAXN],to[MAXM],next[MAXM],w[MAXM],S,minl,size[MAXN],tot,father[MAXN],dis[MAXN],dis2[MAXN]; bool vis[MAXN]; vector<int>D1[MAXN],D2[MAXN]; inline void link(int x,int y,int z){ next[++ecnt]=first[x]; first[x]=ecnt; to[ecnt]=y; w[ecnt]=z; } inline int getint(){ int w=0,q=0; char c=getchar(); while((c<'0'||c>'9') && c!='-') c=getchar(); if(c=='-') q=1,c=getchar(); while (c>='0'&&c<='9') w=w*10+c-'0',c=getchar(); return q?-w:w; } namespace jump{ int f[MAXN][20],deep[MAXN],g[MAXN][20]; inline void dfs(int x,int fa){ reg(i,x) { int v=to[i]; if(v==fa) continue; deep[v]=deep[x]+1; g[v][0]=w[i]; f[v][0]=x; dfs(v,x); } } inline void Init(){ deep[1]=1; dfs(1,0); for(int j=1;j<=19;j++) for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1],g[i][j]=g[i][j-1]+g[f[i][j-1]][j-1];} inline int getdis(int x,int y){ if(x==y) return 0; if(deep[x]<deep[y]) swap(x,y);//!!! int tot=0,t=0; while((1<<t)<=deep[x]) t++; t--; for(int i=t;i>=0;i--) if(deep[x]-(1<<i)>=deep[y]) tot+=g[x][i],x=f[x][i]; if(x==y) return tot; for(int i=t;i>=0;i--) if(f[x][i]!=f[y][i]) tot+=g[x][i],tot+=g[y][i],x=f[x][i],y=f[y][i]; tot+=g[x][0]; tot+=g[y][0]; return tot; } } inline void calcw(int x,int fa){ size[x]=1; tot++; for(int i=first[x];i;i=next[i]){ int v=to[i]; if(v==fa || vis[v]) continue; calcw(v,x); size[x]+=size[v]; } } inline void dp(int x,int fa){ int maxs=0; for(int i=first[x];i;i=next[i]){ int v=to[i]; if(v==fa || vis[v]) continue; dp(v,x); maxs=max(maxs,size[v]); } maxs=max(maxs,tot-size[x]); if(maxs<minl) minl=maxs,S=x; } inline void getrt(int &x){ tot=0; calcw(x,0); S=0; minl=n+1; dp(x,0); x=S; } inline void dfs(int x,int fa,int nowrt,int lei){ D2[nowrt].push_back(dis[x]); dis2[x]=dis[x]; dis[x]=lei; D1[nowrt].push_back(lei); for(int i=first[x];i;i=next[i]) { int v=to[i]; if(v==fa || vis[v]) continue; dfs(v,x,nowrt,lei+w[i]); } } inline int solve(int x){ getrt(x); vis[x]=1; dfs(x,0,x,0); if(tot==1) return x; for(int i=first[x];i;i=next[i]) { int v=to[i]; if(vis[v]) continue; father[solve(v)]=x;//!!! } return x; } inline bool calc(int x,int val){ int tot=0,savval=val,savx=x,last=0; while(x) { val=savval-jump::getdis(x,savx); //if(val<0) return false;//!!! if(last) { tot-=upper_bound(D2[last].begin(),D2[last].end(),val)-D2[last].begin(); } tot+=upper_bound(D1[x].begin(),D1[x].end(),val)-D1[x].begin(); if(tot>k) return true; /*if(father[x]) { //printf("---%d ",upper_bound(D2[x].begin(),D2[x].end(),val)-D2[x].begin()); tot-=upper_bound(D2[x].begin(),D2[x].end(),val)-D2[x].begin(); }*/ last=x; x=father[x]; } return false; } inline void getans(int x){ int l=0,r=(1<<30)-1,mid; int ans=0; while(l<=r) { mid=(l+r)>>1; if(calc(x,mid)) r=mid-1,ans=mid; else l=mid+1; } printf("%d ",ans); } inline void work(){ n=getint(); k=getint(); int x,y,z; for(int i=1;i<n;i++) { x=getint(); y=getint(); z=getint(); link(x,y,z); link(y,x,z); } jump::Init(); solve(1); for(int i=1;i<=n;i++) sort(D1[i].begin(),D1[i].end()),sort(D2[i].begin(),D2[i].end()); for(int i=1;i<=n;i++) getans(i); } int main() { #ifndef ONLINE_JUDGE freopen("2117.in","r",stdin); freopen("2117.out","w",stdout); #endif work(); return 0; } //有志者,事竟成,破釜沉舟,百二秦关终属楚;苦心人,天不负,卧薪尝胆,三千越甲可吞吴。