SSL2843 2017年11月8日提高组T2 拆网线（树形dp）

2017年11月8日提高组T2 拆网线

Description

企鹅国的网吧们之间由网线互相连接，形成一棵树的结构。现在由于冬天到了，供暖部门缺少燃料，于是他们决定去拆一些网线来做燃料。但是现在有K只企鹅要上网和别人联机游戏，所以他们需要把这K只企鹅安排到不同的机房（两只企鹅在同一个机房会吵架），然后拆掉一些网线，但是需要保证每只企鹅至少还能通过留下来的网线和至少另一只企鹅联机游戏。
所以他们想知道，最少需要保留多少根网线？

Input

第一行一个整数T，表示数据组数；
每组数据第一行两个整数N，K，表示总共的机房数目和企鹅数目。
第二行N-1个整数，第i个整数Ai表示机房i+1和机房Ai有一根网线连接(1≤Ai≤i)。

Output

每组数据输出一个整数表示最少保留的网线数目。

Sample Input

2
4 4
1 2 3
4 3
1 1 1
Sample Output

2
2
Hint

【数据规模和约定】
对于30%的数据：N≤15；
对于50%的数据：N≤300；
对于70%的数据：N≤2000；
对于100%的数据：2≤K≤N≤100000，T≤10。

分析：显然最优的情况是每两个企鹅共用一条网线，然后多出来的企鹅每只多用一条，设f[i][0]表示以i为根的子树不包括i最多有多少企鹅能够两两共用一条网线，f[i][1]表示包括i，相信聪明的读者看到这里已经推出方程。

代码

#include <cstdio>
#include <cstring>
#define N 200005
using namespace std;

struct arr
{
    int to,nxt;
}a[N];
int f[N][2],ls[N],n,m,l,ans;

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

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

void dfs(int x,int fa)
{
    for (int i=ls[x];i;i=a[i].nxt)
        if (a[i].to!=fa)
        {
            dfs(a[i].to,x);
            f[x][0]+=f[a[i].to][1];
        }
    for (int i=ls[x];i;i=a[i].nxt)
        if (a[i].to!=fa)
            f[x][1]=max(f[x][1],f[x][0]-f[a[i].to][1]+f[a[i].to][0]+2);
}

int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    int T;
    scanf("%d",&T);
    while (T--)
    {
        memset(f,0,sizeof(f));
        memset(ls,0,sizeof(ls));
        l=0;
        scanf("%d%d",&n,&m);
        for (int i=2;i<=n;i++)
        {
            int x;
            scanf("%d",&x);
            add(x,i);
            add(i,x);
        }
        dfs(1,0);
        ans=max(f[1][1],f[1][0]);
        if (ans>=m) ans=(m+1)/2;
            else ans=m-ans/2;
        printf("%d\n",ans);
    }
    fclose(stdin);
    fclose(stdout);
}