1. 程式人生 > >【貪心】BZOJ1060(ZJOI2007)[時態同步]題解

【貪心】BZOJ1060(ZJOI2007)[時態同步]題解

題目概述

有一棵有邊權的有根樹,現在從根開始傳播JZ的神犇之力,經過一條邊的時間是該邊的邊權。為了JZ神犇之力的平衡,要求神犇之力傳播到所有葉子節點的時間相同。有一種操作是讓某條邊的邊權增加 1 ,求最少的運算元使得神犇之力平衡。

解題報告

肯定是要把所有葉子節點的時間都變成最長時間葉子節點的時間,而且為了少花費,操作必然是越上面越好,所以只需要先DFS一遍記錄一下 MAXi 表示 i 子樹中葉子的的最長時間,然後再DFS一遍,能操作就操作即可。

示例程式

#include<cstdio>
#include<cctype>
using namespace
std; typedef long long LL; const int maxn=500000; int n,ro;LL dis[maxn+5],MAX[maxn+5],ans; int E,lnk[maxn+5],son[(maxn<<1)+5],nxt[(maxn<<1)+5],w[(maxn<<1)+5]; inline char readc(){ static char buf[100000],*l=buf,*r=buf; if (l==r) r=(l=buf)+fread(buf,1,100000,stdin); if (l==r) return
EOF;return *l++; } #define Eoln(x) ((x)==10||(x)==13||(x)==EOF) inline int readi(int &x){ int tot=0,f=1;char ch=readc(),lst='+'; while (!isdigit(ch)) {if (ch==EOF) return EOF;lst=ch;ch=readc();} if (lst=='-') f=-f; while (isdigit(ch)) tot=(tot<<3)+(tot<<1)+(ch^48),ch=readc(); return
x=tot*f,Eoln(ch); } #define Add(x,y,z) son[++E]=(y),w[E]=(z),nxt[E]=lnk[x],lnk[x]=E void Dfs(int x,int pre=0){ for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=pre){ dis[son[j]]=dis[x]+w[j];Dfs(son[j],x); if (MAX[son[j]]>MAX[x]) MAX[x]=MAX[son[j]]; } if (dis[x]>MAX[x]) MAX[x]=dis[x]; } void Solve(int x,int pre=0,LL tag=0){ for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=pre){ LL now=MAX[ro]-MAX[son[j]]-tag; ans+=now;Solve(son[j],x,tag+now); } } int main(){ freopen("program.in","r",stdin); freopen("program.out","w",stdout); for (int i=(readi(n),readi(ro),1),x,y,z;i<n;i++) readi(x),readi(y),readi(z),Add(x,y,z),Add(y,x,z); return Dfs(ro),Solve(ro),printf("%lld\n",ans),0; }