POJ - 3107 Godfather 解题报告（树形dp）（链式向前星）

题目描述

链接：https://vjudge.net/problem/POJ-3107
其实还是找树的重心。这道题和上一道题思路是一样的，但是写法完全不一样。为什么？接着往下看。

思路分析

这道题还是那个想法：求出每个点作为根节点的时候的最大子树节点数，通过dfs，分别计算出每一层的子树节点数，然后dp回去求上一层的子树节点数。但是这道题卡vector，所以需要用链式向前星模拟vector进行dfs。。。还得学个链式向前星。主要要关注的也是这个了。
ps：这道题为什么还卡输入啊。。强制用scanf，切断同步还是不行。。。

AC代码

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=5e4+5;
struct Edge{
	int next;
	int to;
};
int n;
Edge e[maxn<<1];
int head[maxn];
int dis[maxn];
int du[maxn];
int ans[maxn]={0};
int cnt=0;
int cnt1=0;
ll mi=INT_MAX;
void add(int u,int v){
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int father,int now){
	int tmp=0;
	for(int i=head[now];i!=-1;i=e[i].next){
		int v=e[i].to;
		if(v==father) continue;
		dfs(now,v);
		dis[now]+=dis[v];
		tmp=max(tmp,dis[v]);
	}
	tmp=max(tmp,n-dis[now]-1);
	if(tmp<mi){
		mi=tmp;
		cnt1=0;
		ans[cnt1++]=now;
	}
	else if(tmp==mi) ans[cnt1++]=now;
	dis[now]+=1;
}

int main()
{
	cin>>n;
	for(int i=0;i<=n;i++) head[i]=-1;
	for(int i=1;i<n;i++){
		int a,b;
		scanf("%d %d",&a,&b);
		add(a,b);
		add(b,a);
		du[a]++;
		du[b]++;
	}
	for(int i=1;i<=n;i++){
		if(du[i]==1){
			dfs(-1,i);
			break;
		}
	}
	sort(ans,ans+cnt1);
	for(int i=0;i<cnt1;i++){
		if(i) printf(" ");
		printf("%d",ans[i]);
	}
	return 0; 
}