BZOJ 1060: [ZJOI2007]时态同步

BZOJ 1060

orzzyy  跪烂

如果要满足题目的要求，则以每一个非叶节点  到其子树中所有叶子节点距离相同

这很显然

然后考虑到我们只能加大一条边的权值而不能减小

所以我们只能所有路径加长到最长的那一条

所以我们只要贪心就好了

具体实现看代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <cstring>
#include <ctime>
#include <vector>
#define inf 1e9
#define eps 1e-9
#define iter multiset<ll>::iterator
#define ll long long
#define maxn 20010
#define For(i,j,k) for(ll i=j;i<=k;i++)
#define Dow(i,j,k) for(ll i=k;i>=j;i--)
using namespace std;
inline void read(ll &tx){ ll x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} tx=x*f; }
inline void write(ll x){ if (x<0) putchar('-'),x=-x; if (x>=10) write(x/10); putchar(x%10+'0'); }
inline void writeln(ll x){write(x);puts("");}
ll f[1000001],poi[1000001],nxt[1000001],v[1000001],cnt,mx[1000001],n,s,x,y,z;
ll ans;
inline void add(ll x,ll y,ll z){poi[++cnt]=y;nxt[cnt]=f[x];v[cnt]=z;f[x]=cnt;}
inline void dfs(ll x,ll fa)
{
for(ll i=f[x];i;i=nxt[i])
{
if(poi[i]==fa) continue;
dfs(poi[i],x);
mx[x]=max(mx[x],mx[poi[i]]+v[i]);
}
for(ll i=f[x];i;i=nxt[i])
{
if(poi[i]==fa) continue;
ans+=(ll)(mx[x]-mx[poi[i]]-v[i]);
}
}
int main()
{
read(n);read(s);
For(i,1,n-1)
{
read(x);read(y);read(z);
add(x,y,z);add(y,x,z);
}
dfs(s,0);
writeln(ans);
}