hdu 4714 Tree2cycle dp

时间:2023-11-11 14:26:32

用树形dp做的,dp[t][i]表示t及其孩子入度都已经小于等于2并且t这个节点的入度等于i的最优解。

那么转移什么的自己想想就能明白了。

关键在于这个题目会暴栈,所以我用了一次bfs搜索出节点的顺序,然后再计算。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e6+9;
struct
{
int to,next;
}e[maxn*2];
int head[maxn],lon;
bool text[maxn];
void edgeini()
{
memset(head,-1,sizeof(head));
lon=-1;
}
void edgemake(int from,int to)
{
e[++lon].to=to;
e[lon].next=head[from];
head[from]=lon;
}
int dp[maxn][3];
int from[maxn];
void dfs(int t)
{
int k;
int a1=10,a2=10,ret=0,u;
for(k=head[t];k!=-1;k=e[k].next)
{
u=e[k].to;
if(u==from[t]) continue;
ret++;
dp[t][0]+=dp[u][2]+2;
if(dp[u][1]-dp[u][2]<a1)
{
a2=a1;
a1=dp[u][1]-dp[u][2];
}
else if(dp[u][1]-dp[u][2]<a2)
a2=dp[u][1]-dp[u][2];
}
if(ret>=1)
dp[t][1]=max(0,dp[t][0]+a1-2);
else
dp[t][1]=dp[t][0];
if(ret>=2)
dp[t][2]=max(0,dp[t][0]+a1+a2-4);
else
dp[t][2]=dp[t][1];
// cout<<0<<endl;
} int que[maxn];
void bfs()
{
int front=1,end=0;
que[++end]=1;
text[1]=1;
while(front<=end)
{
int t=que[front++];
for(int k=head[t];k!=-1;k=e[k].next)
{
int u=e[k].to;
if(!text[u])
{
from[u]=t;
text[u]=1;
que[++end]=u;
}
}
}
for(int i=end;i>=1;i--)
dfs(que[i]);
} int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(dp,0,sizeof(dp));
edgeini();
int n;
scanf("%d",&n);
for(int i=1,from,to;i<n;i++)
{
scanf("%d %d",&from,&to);
edgemake(from,to);
edgemake(to,from);
}
memset(text,0,sizeof(text));
bfs();
printf("%d\n",dp[1][2]+1);
}
return 0;
}