2019ICPC徐州M Kill the tree

https://nanti.jisuanke.com/t/42552

这场终于有道会做的题了。。。我还是只会dp，规律题数学题蛤不清be

这个感觉就是求出每个子树的重心啊，然后因为没看到两个重心都要算出来还wa了以后对拍，结果发现经典题目看漏了，然后样例又没有2个的情况。。。

子树的重心要么是1个要么是连续的两个

那么树形DP就行了，dpf[u]表示当前子树其他所有点到u的距离综合，ans[u][0]表示u的重心中较深的那个,dp[u]表示重心的那个所有点距离之和，son[u]表示子树节点个数，

然后我们考虑怎么由u的所有儿子v得到u的重心

首先考虑重心是在u还是在哪棵子树v上，由于这两个重心向两边一定是答案变大的，所以如果从c(u)>=c(v)的话，那么重心们一定在v子树或者u上

具体的计算就代码见

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;

int n,m,k,cnt,tot,cas;
int a[maxl],son[maxl],f[maxl],dep[maxl];
ll dp[maxl],dpf[maxl];
bool vis[maxl];
char s[maxl];
vector<int> e[maxl],ans[maxl];

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
	}
}

inline void dfs(int u,int fa)
{
	son[u]=1;dpf[u]=0;
	for(int v:e[u])
	if(v!=fa)
	{
		f[v]=u;dep[v]=dep[u]+1;
		dfs(v,u),son[u]+=son[v];
		dpf[u]+=dpf[v]+son[v];
	}
	if(son[u]==1)
	{
		dp[u]=0;ans[u].push_back(u);
		return;
	}
	ll nowd=u,now=dpf[u];
	for(int v:e[u])
	if(v!=fa)
	{
		ll nsum=dpf[v]+dpf[u]-dpf[v]-son[v]+son[u]-son[v];
		if(nsum<=dpf[u])
		{
			nowd=ans[v][0];
			now=dp[v]+1ll*(dep[nowd]-dep[v]+1)*(son[u]-son[v]);
		}
	}
	dp[u]=now;ans[u].push_back(nowd);
	while(nowd!=u)
	{
		ll nsum=now-(son[u]-son[nowd])+son[nowd];
		if(nsum>now)
			break;
		if(nsum<now)
			ans[u].clear(),dp[u]=nsum;
		ans[u].push_back(f[nowd]);
		now=nsum;nowd=f[nowd];
	}
}

inline void mainwork()
{
	dfs(1,0);
}

inline void print()
{
	for(int i=1;i<=n;i++)
	{
		sort(ans[i].begin(),ans[i].end());
		int l=ans[i].size();
		for(int j=0;j<l;j++)
 			printf("%d%c",ans[i][j]," \n"[j==l-1]);
	}
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}