E. Tree Shuffling (Round #646 (Div. 2))

题目
题意:给出一棵树n个结点的树,结点标号1-n,每个结点i具有(a_i)价值,(b_i)起始二进制位(0或1),(c_i)目标二进制位(0或1).
操作:对某一棵子树u的子集任意k个结点进行重组,花费为a[u]*k
问:是否可以使所有结点的二进制位都是目标二进制位,可以则输出最小花费。
解法
可以知道需要转换类型有两种:

  • 1变0
  • 0变1

条件:分析可知只有两种转换类型节点数相等,才可达到目标。
最小花费:如果根节点符合条件,则需看是否存在更优的花费,此外可以知道此时需要重组的对数已经固定。
贪心:如果根子树的某一结点符合条件,且该节点的花费比其所有父节点的花费小,
则可优先对该节点重组掉,因为此时花费比其所有父节点重组的花费小

#include<bits/stdc++.h>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <string>
#include <stdio.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <string.h>
#include <vector>
#include <stdlib.h>
#include <time.h>
using namespace std;
typedef  long long ll ;
#define int  ll
#define mod 1000000007
#define gcd __gcd
#define rep(i , j , n) for(int i = j ; i <= n ; i++)
#define red(i , n , j)  for(int i = n ; i >= j ; i--)
#define ME(x , y) memset(x , y , sizeof(x))
//int lcm(int a , int b){return a*b/gcd(a,b);}
//ll quickpow(ll a , ll b){ll ans=1;while(b){if(b&1)ans=ans*a%mod;b>>=1,a=a*a%mod;}return ans;}
//int euler1(int x){int ans=x;for(int i=2;i*i<=x;i++)if(x%i==0){ans-=ans/i;while(x%i==0)x/=i;}if(x>1)ans-=ans/x;return ans;}
//const int N = 1e7+9; int vis[n],prime[n],phi[N];int euler2(int n){ME(vis,true);int len=1;rep(i,2,n){if(vis[i]){prime[len++]=i,phi[i]=i-1;}for(int j=1;j<len&&prime[j]*i<=n;j++){vis[i*prime[j]]=0;if(i%prime[j]==0){phi[i*prime[j]]=phi[i]*prime[j];break;}else{phi[i*prime[j]]=phi[i]*phi[prime[j]];}}}return len}
#define INF  0x3f3f3f3f
#define PI acos(-1)
#define pii pair<int,int>
#define fi first
#define se second
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define pb push_back
#define mp make_pair
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 ;}
#define all(v) v.begin(),v.end()
#define size(v) (int)(v.size())
#define cin(x) scanf("%lld" , &x);
#define endl '
'
const double esp = 1e-6;
const int N = 1e6+9;
const int maxn = 2e5+9;
int a[maxn] , b[maxn] , c[maxn];
vector<int>g[maxn];
int ans ;

int dfs(int u , int pre){
	int x = 0 , y = 0 ;
	if(b[u] != c[u]){//统计两种转换
		if(b[u]){
			y++;
		}else{
			x++;
		}
	}
	rep(i , 0 , size(g[u])-1){
		int v = g[u][i];
		if(v == pre) continue;
		uin(a[v] , a[u]);//子树价值小于等于所有父亲
		int r = dfs(v , u);
		if(r > 0)x += r ;//加上子树所剩下的两种类型数量
		if(r < 0)y -= r ;
	}
	ans += 2*min(x , y) * a[u];//从下到上尽可能的重组.
	return x - y ;
}

void solve(){
	int n ;
	cin >> n ;
	rep(i , 1 , n){
		cin >> a[i] >> b[i] >> c[i];
	}
	rep(i , 1 , n-1){
		int u , v ;
		cin >> u >> v ;
		g[u].pb(v);g[v].pb(u);
	}
	if(dfs(1 , -1)) cout << -1 << endl;
	else{
		cout << ans << endl;
	}


}

signed main()
{
    //int _ ;cin>>_;while(_--)
    //while(~scanf("%lld%lld" , &n , &m) && n+m)
        solve();
}

原文地址:https://www.cnblogs.com/nonames/p/13024248.html