hdu 4714 树+DFS

时间:2021-07-18 12:36:35

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4714

本来想直接求树的直径,再得出答案,后来发现是错的。

思路:任选一个点进行DFS,对于一棵以点u为根节点的子树来说,如果它的分支数大于1,那么我们把这颗子树从整棵树上剪下来(优先减去),同时把这颗子树的分支留下两个,其它多余的也剪掉,然后把剪下来的这些部分连接到根节点那里,从而形成一条直链,总代价就是我们减的次数+把剪下来的部分连接到根节点+把最后的直链连成环。在这里剪的次数=把剪下来的部分连接到根节点的次数,把最后的直链连成环只需要一步。

参考博客:https://blog.csdn.net/cc_again/article/details/11407157

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque>
using namespace std;
typedef long long LL;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 1000005
int n,m,k,t,cnt;
int ans;
struct node{
int v,next;
}edge[maxn*];
int head[maxn];
void add(int u,int v){
edge[++cnt].v=v;
edge[cnt].next=head[u];
head[u]=cnt;
}
int DFS(int u,int pre){
int num=;
for(int i=head[u];i!=-;i=edge[i].next){
int v=edge[i].v;
if(v==pre)
continue;
num+=DFS(v,u);
}
if(num>){//如果以点u为根节点的子树的分支数大于1
if(u==)
ans+=num-;
else
ans+=num-;
return ;//这颗子树剪断了,所以返回0
}
return ;//分支数只有一个
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d",&n);
int u,v;
cnt=;
memset(head,-,sizeof(head));
for(int i=;i<n;i++){
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
ans=;
DFS(,-);
printf("%d\n",ans*+);
}
return ;
}