codeforces 1153 D. Serval and Rooted Tree(树形DP)

http://codeforces.com/contest/1153/problem/D

题意:给出一棵树,1位根节点,有k个叶子节点,叶子节点标号1-k(不重复)
,每个节点有两种操作 1 取子节点最大值,0 取子节点最小值,问根节点最大为多少。

自己做的时候想到dp这个方向,但是具体还是不知道怎么操作,看了题解还是自闭了好久,不知道自己是英语不好还是语文不好…

我的思考:题目要求最大的根结点的值,那么可以换个角度,维护一个dp,以该节点为根节点,子节点叶子数的最小值的最大情况,这样才能保证后面答案合法,相当于一个贪心的思想(这里随便画几个例子验证一下),那么对于每颗子树来说,答案=叶子数+1-dp。

MIN操作:最小值的最大情况为叶子数的和
MAX操作:最小值直接为叶子数目的最小。

对于1号根结点答案为 k+1-dp;
要求出dp[1] ,那么就要先求出他的子节点,那么这道题就变成了一个树形dp问题。

#include<bits/stdc++.h>
#define fi first
#define se second
#define log2(a) log(n)/log(2)
#define show(a) cout<<a<<endl;
#define show2(a,b) cout<<a<<" "<<b<<endl;
#define show3(a,b,c) cout<<a<<" "<<b<<" "<<c<<endl;
using namespace std;

typedef long long ll;
typedef pair<int, int> P;
typedef pair<P, int> LP;
const ll inf = 1e17 + 10;
const int N = 1e6 + 10;
const ll mod = 1e9 + 7;
const int base = 131;
const double pi = acos ( -1 );
map<string, int>ml;


map<char, ll> mp;
map<int, int> vi;


ll b[N], a[N], vis[N], num[N], n, m,  k, x, y, h;
ll ex, ey, t, cnt, cnt2, ans, sum, flag, dp[N];


vector<int> v[N], A, B, C, D;

void dfs ( int x )
{
	if ( x > 1 && num[x] == 0 )
	{
		//show2("x",x)
		cnt++;
		dp[x] = 1;
		return ;
	}
	if ( a[x] ) dp[x] = inf;
	else dp[x] = 0;
	for ( int to : v[x] )
	{
		dfs ( to );
		if ( a[x] )
		{
			dp[x] = min ( dp[x], dp[to] );
		}
		else dp[x] += dp[to];
	}
}

int main( )
{
	ios::sync_with_stdio ( false );
	cin.tie ( 0 );

	cin >> n;
	for ( int i = 1; i <= n; i++ )
	{
		cin >> a[i];
	}
	for ( int i = 2; i <= n; i++ )
	{
		cin >> x;
		v[x].push_back ( i );
		num[x]++;
	}
	dfs ( 1 );
	cout << cnt - dp[1] + 1 << endl;

}




  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值