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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值