P1395 会议（树上找一点到所有点距离和最小）

题目传送门

题意： 一颗n个点的树，n-1条边，每条边的权值为1，找到一个点，使得所有点到这个点的距离之和最小。

思路： 我们假设点1为根时，总花费为f[1]，假设点2是点1的子节点，那么f[2]=f[1]-点2的子节点个数+（n-点2的子节点个数），因为我们把根从1换到2的时候，2的子树中节点贡献全部-1，其他节点贡献全部+1。所以我们以1位根节点dfs一遍，找出每个节点的子树节点数量，再dfs算出以1为根时的代价花费，最后dfs算出每个点的花费，最后输出最小花费点和最小花费就行了。主要是式子不好想。

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
const int inf=0x7fffffff;
const int mod=1e9+7;
const int eps=1e-6;
typedef long long ll;
typedef unsigned long long ull;
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl '\n'
#define null NULL
int head[N],nxt[N],to[N],tot=0;
void add(int u,int v)
{
    nxt[++tot]=head[u];
    to[tot]=v;
    head[u]=tot;
}
int s[N],f[N];
int dfs(int x,int y)
{
    for(int i=head[x];i;i=nxt[i])
    {
        if(to[i]!=y)
            s[x]+=1+dfs(to[i],x);
    }
    return s[x];
}
void dfs2(int x,int y,int z)
{
    f[1]+=z;
    for(int i=head[x];i;i=nxt[i])
    {
        if(to[i]!=y)
            dfs2(to[i],x,z+1);
    }
}
int res,n,pos;
void dfs3(int x,int y)
{
    f[x]=f[y]-(s[x]+1)+n-s[x]-1;
    for(int i=head[x];i;i=nxt[i])
        if(to[i]!=y)
            dfs3(to[i],x);
}
signed main()
{
    IOS;
    cin>>n;
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        cin>>u>>v;
        add(u,v);add(v,u);
    }
    dfs(1,0);
    for(int i=head[1];i;i=nxt[i])
        dfs2(to[i],1,1);
    for(int i=head[1];i;i=nxt[i])
        dfs3(to[i],1);
    res=f[1];pos=1;
    for(int i=1;i<=n;i++)
    {
        if(f[i]<res)
        {
            res=f[i];
            pos=i;
        }
    }
    cout<<pos<<' '<<res<<endl;
}