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;
}