2017.11.2 树上期望DP 解题报告

题目描述

给你一棵包含个n点的有根树，点的标号是1…n，在t=1时（ t表示时间），你在1号点，接下来，你会随机跑到当前点相邻的点，然后继续这个过程，直到访问了所有的点，已从一个点到另一个点需要的时间是1秒，那么问题来了，请问在这个随机过程中，对于每个节点，冬雪第一次访问的期望时间是多少？

输入数据：

3
1 2
2 3

输出数据：

1.000
2.000
5.000

【解题报告】

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
#define LL long long

int n,cnt=-1;
struct Edge{int to,nxt;}e[N<<1];
int head[N],size[N];
LL dp[N];

void adde(int u,int v)
{
    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].nxt=head[v];head[v]=cnt;
}
void dfs(int u,int fa)
{
    size[u]=1;
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
        size[u]+=size[v];
    }
    if(u!=1) dp[u]=dp[fa]+2*(n-size[u])-1;
}
void dfs2(int u,int fa) 
{
    for(int i=head[u];~i;i=e[i].nxt) 
    {
        int v=e[i].to;
        if(v==fa) continue;
        dp[v]=dp[u]+1+2*(n-1-size[v]);
        dfs2(v,u);
    }
}
int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;++i)
    {
        int u,v;scanf("%d%d",&u,&v);
        adde(u,v);
    }
    dp[1]=1;
    dfs(1,1);dfs2(1,1);
//  for(int i=1;i<=n;++i) printf("%d ",size[i]);puts("");
    for(int i=1;i<=n;++i) printf("%lld.000\n",dp[i]);
    return 0;
}