E. Tree Shuffling
题意:
给你一棵有根树。根为1.
每个结点都有一个值
b
[
i
]
b[i]
b[i] = 0/1。
有一个目标值
c
[
i
]
c[i]
c[i] = 0/1
现在提供操作:
你可以选择一个
u
u
u结点作为根,在这棵子树下,同时修改k个结点,使得它们的值变为你想要的。
花费为
a
[
u
]
a[u]
a[u]
求:
最小的花费
思路:
- 首先可以知道,无解的情况就是 0->1 的 个数 和 1->0 的个数不相等
- 首先可以知道对于一个结点 i i i ,假如结点 p p p (u的爸爸) 花费更少。那么以 i i i为子树的点,可以到爸爸时再消去。
- 用2的贪心思想就可以达到最优。
AC
/*
皮卡丘冲鸭!
へ /|
/\7 ∠_/
/ │ / /
│ Z _,< / /`ヽ
│ ヽ / 〉
Y ` / /
イ● 、 ● ⊂⊃〈 /
() へ | \〈
>ー 、_ ィ │ //
/ へ / ノ<| \\
ヽ_ノ (_/ │//
7 |/
>―r ̄ ̄`ー―_
*/
#include <iostream>
#include <bits/stdc++.h>
#define For(i,x,y) for(int i=(x); i<=(y); i++)
#define fori(i,x,y) for(int i=(x); i<(y); i++)
#define rep(i,y,x) for(int i=(y); i>=(x); i--)
#define mst(x,a) memset(x,a,sizeof(x))
#define pb push_back
#define sz(a) (int)a.size()
#define ALL(x) x.begin(),x.end()
#define mp make_pair
#define fi first
#define se second
#define db double
#define debug(a) cout << #a << ": " << a << endl
using namespace std;
typedef long long ll;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
typedef pair<int,int>pa;
typedef pair<ll,ll>pai;
typedef pair<db,db> pdd;
const int N = 2e5+10;
const int M = 1e5;
const int maxn=1e6+10;
const db eps = 1e-8;
const db pi = acos(-1.0);
template<class T> bool uin(T &a, T b) { return a > b ? (a = b, true) : false; }
template<class T> bool uax(T &a, T b) { return a < b ? (a = b, true) : false; }
ll c[N], b[N];
int n;
vector<int>e[N];
ll ans;
ll dfs(int u, int fa){
int pos=0, neg= 0;
if(b[u]==1)pos++;
if(b[u]==-1)neg++;
for(auto v: e[u]){
if(v == fa)continue;
uin(c[v],c[u]);
int r = dfs(v,u);
if(r>0)pos+=r;
if(r<0)neg-=r;
}
ans+=min(pos,neg)*2*c[u];
return pos-neg;
}
int main()
{
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin>>n;
int x,y;
for(int i= 1; i <= n; i++ ){
cin>>c[i]>>x>>y;
b[i] = x-y;
}
for(int i = 1; i <= n-1; i ++ ){
cin>>x>>y;
e[x].pb(y);
e[y].pb(x);
}
if(dfs(1,-1))ans = -1;
cout<<ans<<endl;
return 0;
}