總結
自己當時賽中的問題,沒想太清楚就是開始碼了,自己的寫法,雖然也是貪心,但是兩次DFS,代碼大,先排序從小到大,搜未搜索的子樹,然後把多餘的點往上移動。
第一次DFS:確定爸爸
第二次DFS:更新子樹的最優值
當看了別人的代碼後,就知道自己有些多餘了,每個結點的如果需要向目標值改變,需要的是祖祖輩輩最小的點權,DFS更新下去就可以了,我還傻乎乎的不斷貪心求子樹。
算法核心
樹形DP:
1:DFS每次得知道爸爸
2:遞歸:祖輩(root->父節點的簡單路徑)的關係
3:回溯:子孫(子樹)的關係
題目鏈接
//#pragma GCC optimize(2)
//#pragma GCC target ("sse4")
#include<bits/stdc++.h>
//typedef long long ll;
#define ull unsigned long long
#define int long long
#define F first
#define S second
#define endl "\n"//<<flush
#define eps 1e-6
#define base 131
#define lowbit(x) (x&(-x))
#define PI acos(-1.0)
#define inf 0x3f3f3f3f
#define MAXN 0x7fffffff
#define INF 0x3f3f3f3f3f3f3f3f
#define ferma(a,b) pow(a,b-2)
#define mod(x) (x%mod+mod)%mod
#define pb push_back
#define decimal(x) cout << fixed << setprecision(x);
#define all(x) x.begin(),x.end()
#define memset(a,b) memset(a,b,sizeof(a));
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
void file()
{
#ifdef ONLINE_JUDGE
#else
freopen("D:/LSNU/codeforces/duipai/data.txt","r",stdin);
// freopen("D:/LSNU/codeforces/duipai/WA.txt","w",stdout);
#endif
}
const int N=2e5+5;
int dp[N][2],a[N],ans=0;
vector<int>G[N];
struct node
{
int a,b,c;
}s[N];
void dfs(int u,int fat)
{
s[u].a=min(s[u].a,s[fat].a);
if(s[u].b!=s[u].c)
dp[u][s[u].b]++;
for(auto v:G[u])
{
if(v==fat)
continue;
dfs(v,u);
dp[u][0]+=dp[v][0];
dp[u][1]+=dp[v][1];
}
int sum=min(dp[u][0],dp[u][1]);
dp[u][1]-=sum;
dp[u][0]-=sum;
ans+=s[u].a*sum*2;
}
signed main()
{
IOS;
file();
int n;
cin>>n;
int cnt=0;
for(int i=1;i<=n;i++)
{
cin>>s[i].a>>s[i].b>>s[i].c;
if(s[i].b!=s[i].c)
cnt+=(s[i].b==0?1:-1);
}
s[0].a=inf;
for(int i=1;i<n;i++)
{
int u,v;
cin>>u>>v;
G[u].pb(v);
G[v].pb(u);
}
dfs(1,0);
cout<<(cnt?-1:ans)<<endl;
return 0;
}