SSL2811 2017年10月30日提高组T2 摘Galo（树形dp）

2017年10月30日提高组T2 摘Galo

Description

0v0在野外看到了一棵Galo树，看到食物的0v0瞪大了眼睛，变成了OvO。
这棵Galo树可以看做是一棵以1号点为根的n个点的有根数，除了根节点以外，每个节点i都有一个Galo，美味度为w[i]。
OvO发现，如果她摘下了i号Galo，那么i的子树中的Galo以及i到根的路径上的其他Galo都会死掉。
OvO的袋子只能装k个Galo，她的嘴巴里还能叼1个，请问她所摘Galo的美味度之和的最大值是多少？

Input

第一行两个正整数n，k。
第二行到第n行，第i行两个正整数f[i]，w[i]，表示i号点的父亲为f[i] （保证x[i]

Output

一行一个非负整数，为最大美味值。

Sample Input

4 1
1 10
2 3
2 6
Sample Output

10
Hint

30% n,k<=200
30% n*k*k<=10^7
40% n*k<=10^7
对于所有数据，n,k,w[i]<=10^5

分析：题目对取 Galo 的要求，即不能同时取一对祖先和子孙。我们可 以在 DFS 序上 DP。对于选择取 Galo[i]的决策，即选择了 DFS 序上 i 号节点的子树所对应的区间。只要选择的区间不相交、不重复，则 是一个合法的取法。这样，题目就成了一个区间覆盖 DP 问题，复 杂度 O(n*k)。

代码

#include <cstdio>
#define N 100005
using namespace std;

struct arr
{
    int to,nxt;
}p[N];

int f[N][105],a[N],ls[N],dfn[N],w[N],size[N];
int n,k,l;
bool v[N];

void add(int x,int y)
{
    p[++l].to=y;
    p[l].nxt=ls[x];
    ls[x]=l;
}

int max(int x,int y)
{
    return x>y?x:y;
}

void dfs(int x)
{
    size[x]=1;
    dfn[++l]=x;
    for (int i=ls[x];i;i=p[i].nxt)
        if (!v[p[i].to])
        {
            v[p[i].to]=true;
            dfs(p[i].to);
            size[x]+=size[p[i].to];
        }
}

int main()
{
//  freopen("galo.in","r",stdin);
//  freopen("galo.out","w",stdout);
    scanf("%d%d",&n,&k);
    k++;
    for (int i=2;i<=n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&w[i]);
        add(x,i);
    }
    l=0;
    dfs(1);
    for (int i=n;i>=1;i--)
        for (int j=1;j<=k;j++)
            f[dfn[i]][j]=max(f[dfn[i+size[dfn[i]]]][j-1]+w[dfn[i]],f[dfn[i+1]][j]);
    int ans=0;
    for (int i=0;i<=k;i++)
        ans=max(ans,f[1][i]);
    printf("%d",ans);
}