题意:
给你一颗树,设每两个节点间距离为1,从1开始,让你求这棵树的路程的期望
思路:
一开始我想用bfs求出每个叶子节点的长度和个数,用长度的和除以个数的做法在第五组就过不了。。。这里应该是涉及一堆数学推导这样求的结果不等同于期望吧,然后看了题解才知道这里的期望要一个个节点递归相加来求解。。具体原理还是不太懂,哎只会图论和数据结构的弱渣
代码:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<stdio.h>
#include<math.h>
#include<string>
#include<stdio.h>
#include<queue>
#include<stack>
#include<map>
#include<deque>
#define M (t[k].l+t[k].r)/2
#define lson k*2
#define rson k*2+1
#define ll long long
#define INF 100861111;
using namespace std;
vector<int>p[100005];//放邻接表
int vis[100005];//遍历数组
double dfs(int index)//index表示当前节点下标
{
vis[index]=1;
int i;
double res=0,ans=0;
for(i=0;i<p[index].size();i++)
{
if(!vis[p[index][i]])//没有遍历
{
res+=dfs(p[index][i]);
ans++;
}
}
if(ans==0)//当前为叶子节点,由于已经不能走了,路径条数为0,所以是0
return 0;
res=1+res/ans;//不是叶子节点的话用自身的期望1(这里可以稍微画图推一下)加上叶子节点的期望
return res;
}
int main()
{
int i,j,k,n,x,y;
double s=0,ans=0;
memset(vis,0,sizeof(vis));
scanf("%d",&n);
for(i=0;i<n-1;i++)
{
scanf("%d%d",&x,&y);
p[x].push_back(y);//注意是双向
p[y].push_back(x);
}
printf("%.15lf\n",dfs(1));
return 0;
}