E. Tree Shuffling(树上dp2000)

E. Tree Shuffling

题意:

给你一棵有根树。根为1.
每个结点都有一个值$b[i]$ = 0/1。
有一个目标值 $c[i]$ = 0/1
现在提供操作：
你可以选择一个$u$结点作为根，在这棵子树下，同时修改k个结点，使得它们的值变为你想要的。
花费为$a[u]$
求：
最小的花费

思路：

1. 首先可以知道，无解的情况就是 0->1 的 个数 和 1->0 的个数不相等
2. 首先可以知道对于一个结点$i$ ，假如结点 $p$ (u的爸爸) 花费更少。那么以$i$为子树的点，可以到爸爸时再消去。
3. 用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;
}