- 题目大意
给你一颗树,让你选出k个节点和尽量少的边使得这k个节点的每个点都至少通过选出的边与一个节点相邻,文这个最少的边数是多少 - 分析
比较明显的DP问题
通过观察不难发现我们应该尽量将点分成能够两两配对的点对
所以我们希望知道以某个节点为根节点的子树中最大能匹配的点数
为了方便状态的转移,我的做法是设
dp[i][0] 表示以i为根的子树中能够两两配对的最大点数,不包含节点i
dp[i][0] 表示以i为根的子树中能够两两配对的最大点数,包含节点i
这样就有转移方程:
dp[u][0]=∑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即可 总结
- 这道题输入需要fread,scanf会超时
- 我看别人的做法是直接用一个状态表示某个节点不用分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
*/