牛客OI周赛8-提高组 B 死宅选点

由于我们只要选取总权值最小的点，而当时，

我们可以将该路径的权值，等价于进行考虑，对结果没有影响。即长度为m的路径，权值为。

很容易得到，条与点距离为的路径，等价于条与点距离为的路径。在处理时，可以采取如此的合并拆分操作。

设表示在点的子树中，与点距离为的路径数量。

设表示在整棵树中，与点距离为的路径数量。

可以通过以下转移式得到：

可以通过以下转移式得到：

，其中为的父节点。

最后，我们只需要比较数组即可。

#include<bits/stdc++.h>
using namespace std;
 
const int MAXN=20020;
const int MAXL=550;
int dp[MAXN][MAXL+5],ans[MAXN][MAXL+5],tmp[MAXL+5];
int k;
vector<int> vec[MAXN];
 
void dfs1(int x,int fa)
{
    int i,j,son;
    for(i=0;i<vec[x].size();i++)
    {
        son=vec[x][i];
        if(son==fa)
            continue;
        dfs1(son,x);
    }
    for(j=1;j<MAXL;j++)
    {
        for(i=0;i<vec[x].size();i++)
        {
            son=vec[x][i];
            if(son==fa)
                continue;
            dp[x][j]+=dp[son][j-1];
            if(k>1&&dp[x][j]>=k)
            {
                dp[x][j+1]++;
                dp[x][j]-=k;
            }
        }
    }
}
 
void dfs2(int x,int fa)
{
    int i,j,son,lazy;
    for(int i=0;i<vec[x].size();i++)
    {
        son=vec[x][i];
        if(son==fa)
            continue;
        lazy=0;
        tmp[0]=1;
        for(j=1;j<MAXL;j++)
        {
            tmp[j]=ans[x][j]-dp[son][j-1]-lazy;
            if(k>1&&tmp[j]<0)
            {
                tmp[j]+=k;
                lazy=1;
            }
            else
                lazy=0;
        }
        for(j=1;j<MAXL;j++)
        {
            ans[son][j]+=tmp[j-1]+dp[son][j];
            if(k>1&&ans[son][j]>=k)
            {
                ans[son][j]-=k;
                ans[son][j+1]++;
            }
        }
        dfs2(son,x);
    }
}
 
bool cmp(int x,int y)
{
    int sum1,sum2,i;
    if(k==1)
    {
        sum1=sum2=0;
        for(i=1;i<MAXL;i++)
        {
            sum1+=ans[x][i]*i;
            sum2+=ans[y][i]*i;
        }
        return sum1<sum2;
    }
    else
    {
        for(i=MAXL-1;i>=1;i--)
        {
            if(ans[x][i]!=ans[y][i])
                return ans[x][i]<ans[y][i];
        }
        return 0;
    }
}
 
int main()
{
    int n,x,y,i;
    scanf("%d%d",&n,&k);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        vec[x].push_back(y);
        vec[y].push_back(x);
    }
    for(i=1;i<=n;i++)
        dp[i][0]=1;
    dfs1(1,-1);
    for(i=0;i<MAXL;i++)
        ans[1][i]=dp[1][i];
    dfs2(1,-1);
    x=1;
    for(i=2;i<=n;i++)
    {
        if(cmp(i,x))
            x=i;
    }
    printf("%d\n",x);
}