您的位置:首页 > 其它

hdu4714(树形dp)

2016-06-13 14:40 465 查看
链接:点击打开链接

题意:给出一棵树,删除一条边或者加入一条边的代价都是1,蚊最后使n个节点连成环所需的最小代价

代码:

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
vector<int> G[1000005];
int dp[1000005][3],vis[1000005];                //dp[i][j]表示以i为根节点的子树,可用度
void dfs(int s){                                //为j时树的分支数,也就是还可以连j个节点
int i,tmp;                                  //时树的分支数
vis[s]=1;
dp[s][0]=dp[s][1]=dp[s][2]=1;
for(i=0;i<G[s].size();i++){
tmp=G[s][i];
if(!vis[tmp]){
dfs(tmp);                               //两个分支合并则减1
dp[s][0]=min(dp[s][0]+dp[tmp][0],dp[s][1]+dp[tmp][1]-1);
dp[s][1]=min(dp[s][1]+dp[tmp][0],dp[s][2]+dp[tmp][1]-1);
dp[s][2]=dp[s][2]+dp[tmp][0];           //保证s节点为根的子树可用度为2,则不能在向上添加节点
}
}
}
int main(){
int i,j,t,u,v,n,ans;
scanf("%d",&t);
while(t--){
ans=0;
memset(dp,0,sizeof(dp));
memset(vis,0,sizeof(vis));
scanf("%d",&n);
for(i=1;i<=n;i++)
G[i].clear();
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1);
ans=min(min(dp[1][0],dp[1][1]),dp[1][2]);
printf("%d\n",2*ans-1);                 //知道分支数,则答案就是分支数-1+分支数,也就是
}                                           //先分离在合并
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: