bzoj 1131（树形dp）

最新推荐文章于 2020-09-15 13:30:16 发布

嘉伟森的猫

最新推荐文章于 2020-09-15 13:30:16 发布

阅读量241

点赞数

分类专栏： OI-dp 文章标签：树形dp

本文链接：https://blog.csdn.net/KGV093/article/details/77827406

版权

OI-dp 专栏收录该内容

83 篇文章 0 订阅

订阅专栏

传送门
虽然是道水题，但是这回真的是自己搞出来的╰(°▽°)╯
先求出1号结点的dep，ans[1]=dep[1]+dep[2]+…+dep[n]，
往下搜索，对于p的儿子v，子树外所有结点的dep增加1，子树内减少1：
ans[v]=ans[p]-siz[v]*1+(n-siz[v])*1
最后统计最大答案时要开longlong，要不是一开始这儿出了小问题，就能1A了。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=1e6+2;
int n,siz[MAXN],dep[MAXN];
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt;
}e[MAXN<<1];
ll ans[MAXN];
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,head[v]=edge++;
}
void pre_dfs(int p,int fa) {
    siz[p]=1,dep[p]=dep[fa]+1;
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) {
            pre_dfs(v,p);
            siz[p]+=siz[v];
        }
    }
}
void cal(int p,int fa) {
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) {
            ans[v]=ans[p]-siz[v]+n-siz[v];
            cal(v,p);
        }
    }
}
int main() {
//  freopen("bzoj 1131.in","r",stdin);
    memset(siz,0,sizeof(siz));
    memset(head,-1,sizeof(head));
    n=read();
    for (register int i=1;i<n;++i) {
        int u=read(),v=read();
        adde(u,v);
    }
    dep[0]=0;
    pre_dfs(1,0);
    for (register int i=1;i<=n;++i) ans[1]+=dep[i];
    cal(1,0);
    ll res=0,pos;
    for (register int i=1;i<=n;++i)
        if (res<ans[i])
            res=ans[i],pos=i;
    printf("%d\n",pos);
    return 0;
}