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);
}