题意:
给你一颗树,设每两个节点间距离为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,所以是0return 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;
}