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