HDU 6178 Monkeys(树上DP 17多校第十场)

• 题目大意
  给你一颗树，让你选出k个节点和尽量少的边使得这k个节点的每个点都至少通过选出的边与一个节点相邻，文这个最少的边数是多少
• 分析
  比较明显的DP问题
  通过观察不难发现我们应该尽量将点分成能够两两配对的点对
  所以我们希望知道以某个节点为根节点的子树中最大能匹配的点数
  为了方便状态的转移，我的做法是设
  $dp[i][0]$表示以i为根的子树中能够两两配对的最大点数，不包含节点i
  $dp[i][0]$表示以i为根的子树中能够两两配对的最大点数，包含节点i
  这样就有转移方程：
  $dp[u][0]=\sum_{v是u的儿子} dp[v][1];$
  $dp[u][1]=max(dp[u][1],dp[u][0]-dp[v][1]+dp[v][0]+2);$
  最后再讨论以下$max(dp[1][0],dp[1][1])$是否大于k即可

• 总结

  1. 这道题输入需要fread，scanf会超时
  2. 我看别人的做法是直接用一个状态表示某个节点不用分0和1,更具的启发式信息是:
     看他有几个儿子和儿子中已经成对的个数，若两者之差>=1，则至少存在一个儿子能与它成对

• 代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
#define LL long long int
const int MAXN=200055;
int T;
int n,k;
int  sum_node[MAXN];//保存以某个节点为根节点的子树的节点数目
int dp[MAXN][2];//dp[i]表示以某个点为根的子树能匹配的点数的最大值，dp[i][0]表示这个点不在匹配点中,dp[i][1]表示这个点可在可不在匹配点中
struct Edge
{
    int v;
    int next;
}edge[MAXN];
int edgecount;
int head[MAXN];
struct FastIO
{
    static const int S = 2*100;
    int wpos;
    char wbuf[S];
    FastIO() : wpos(0) {}
    inline int xchar()
    {
        static char buf[S];
        static int len = 0, pos = 0;
        if (pos==len)
            pos = 0, len = fread(buf, 1, S, stdin);
        if (pos==len) exit(0);
        return buf[pos ++];
    }
    inline int xint()
    {
        int s = 1, c = xchar(), x = 0;
        while(c<=32) c = xchar();
        if(c=='-') s = -1, c = xchar();
        for(;'0'<=c && c<='9';c=xchar()) x = x*10+c-'0';
        return x * s;
    }
    ~FastIO()
    {
        if(wpos) fwrite(wbuf, 1, wpos, stdout), wpos = 0;
    }
}io;
void Init()
{
    edgecount=0;
    memset(head,-1,sizeof(head));
    memset(sum_node,0,sizeof(sum_node));
    memset(dp,0,sizeof(dp));
}
void Add_edge(int u,int v)
{
    edge[++edgecount].v=v;
    edge[edgecount].next=head[u];
    head[u]=edgecount;
}
void In()
{
    int a;
   n=io.xint();
    k=io.xint();
    // scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++)
    {
        //scanf("%d",&a);
        a=io.xint();
        Add_edge(a,i+1);
    }
}
void Dp(int u)//更新以u为根节点的sum_node和dp
{
    if(head[u]==-1){sum_node[u]=1;dp[u][0]=dp[u][1]=0;return ;}
    for(int k=head[u];k!=-1;k=edge[k].next)
    {
        int v=edge[k].v;
        Dp(v);
        sum_node[u]+=sum_node[v];
        dp[u][0]+=dp[v][1];
    }
    sum_node[u]++;
    for(int k=head[u];k!=-1;k=edge[k].next)
    {
        int v=edge[k].v;
        dp[u][1]=max(dp[u][1],dp[u][0]-dp[v][1]+dp[v][0]+2);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    // freopen("out1.txt","w",stdout);
    scanf("%d",&T);
    while(T--)
    {
        Init();
        In();
        Dp(1);
        int t=max(dp[1][0],dp[1][1]);
        if(t<k)printf("%d\n",t/2+k-t);
        else printf("%d\n",(k+1)/2);
        //cout<<t/2+k-t<<endl;
        //cout<<dp[1][0]<<" "<<dp[1][1]<<endl;
    }
    return 0;
}
/*
2
5 2
1 1 2 2
*/