【貪心】BZOJ1060(ZJOI2007)[時態同步]題解
阿新 • • 發佈:2019-02-09
題目概述
有一棵有邊權的有根樹,現在從根開始傳播JZ的神犇之力,經過一條邊的時間是該邊的邊權。為了JZ神犇之力的平衡,要求神犇之力傳播到所有葉子節點的時間相同。有一種操作是讓某條邊的邊權增加 ,求最少的運算元使得神犇之力平衡。
解題報告
肯定是要把所有葉子節點的時間都變成最長時間葉子節點的時間,而且為了少花費,操作必然是越上面越好,所以只需要先DFS一遍記錄一下 表示 子樹中葉子的的最長時間,然後再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;
}