bzoj-1131 Sta

来源:互联网 发布:js div隐藏与显示出来 编辑:程序博客网 时间:2024/06/05 04:17

题意:

给出一个n个点的树,找出一个点来,使以这个点为根的树所有点的深度之和最大;

n<=1000000;


题解:

其实我做这道题的时候总有一种莫名其妙的即视感怎么回事。。。

算了说不定这道题我真的做过。。。

比较暴力的是将所有点枚举,然后深搜累加所有深度;

但是显然所有点等于父树的点+子树的点;

那么只要求出这两者累加就好了;

子树的总深度简直好求,就是将儿子的总深度+size就好了;

父树的稍微有点麻烦,因为需要一些预处理所以在第二次深搜解决;

对于一个点来说,这个点的父树就是【父亲+父亲的父树+兄弟的子树】;

那么分别累加即可:
注意一下推公式就好; (加减同一项不要消下去。。不然没法调。。)

时间复杂度O(n)

所以就这么一道树形DP傻逼题我还是犯了一堆傻逼错误;

一百万看成十万,公式写错,int输出%lld,更新没判相等。。

对了这题还要开long long,这估计是我唯一没掉进去的坑了= =;


代码:


#include<stdio.h>#include<string.h>#include<algorithm>#define N 1100000using namespace std;typedef long long ll;int n,no,to[N<<1],next[N<<1],head[N],ce;ll sum[N],size[N],fas[N],ans;void add(int x,int y){to[++ce]=y;next[ce]=head[x];head[x]=ce;}void dfs1(int x,int pre){size[x]=1;int i,y;for(i=head[x];i;i=next[i]){if((y=to[i])!=pre){dfs1(y,x);size[x]+=size[y];sum[x]+=sum[y]+size[y];}}}void dfs2(int x,int pre){int i,y;if(x!=1)fas[x]=fas[pre]+sum[pre]-sum[x]-size[x]+size[pre]-size[x]-1+n-size[pre]+1;if(ans<fas[x]+sum[x]){ans=fas[x]+sum[x];no=x;}else if(ans==fas[x]+sum[x])no=min(no,x);for(i=head[x];i;i=next[i]){if((y=to[i])!=pre){dfs2(y,x);}}}int main(){int m,i,j,k,x,y;scanf("%d",&n);for(i=1;i<n;i++){scanf("%d%d",&x,&y);add(x,y),add(y,x);}dfs1(1,0);dfs2(1,0);printf("%d\n",no);return 0;}



0 0
原创粉丝点击