CodeForces - 839C Journey dfs+期望

http://codeforces.com/problemset/problem/839/C

题意：有一棵树，根节点为1。从根节点出发，每次不能经过已经到达过的结点。求最长能走的长度的数学期望。

题解：期望没学好，昨天4wa。从根结点走，最长的长度就是根节点到叶子结点的长度。所以求出根节点到每个叶子结点的概率*长度就是期望了。

代码：

#include<bits/stdc++.h>
#define debug cout<<"aaa"<<endl
#define d(a) cout<<a<<endl
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define MIN_INT (-2147483647-1)
#define MAX_INT 2147483647
#define MAX_LL 9223372036854775807i64
#define MIN_LL (-9223372036854775807i64-1)
using namespace std;

const int N = 100000 + 5;
const int mod = 1000000000 + 7;
const double eps = 1e-8;
double ans;
vector<int> son[N];
bool vis[N];

void dfs(int u,int len,double num){
	vis[u]=1;
	int sum=0;
	for(int i=0;i<son[u].size();i++){//计算儿子个数 
		int v=son[u][i];
		if(!vis[v]){
			sum++;
		}
	}
	for(int i=0;i<son[u].size();i++){
		int v=son[u][i]; 
		if(!vis[v]){//num/sum就是根节点到u的下一个结点的概率
			dfs(v,len+1,num/sum);
		}
	}
	if(len!=0&&son[u].size()==1){
		ans+=num*len;//期望
	}
}

int main(){
	mem(vis,0),ans=0;
	int n,u,v;
	cin>>n;
	if(n==1){
		puts("0.0000000");
		return 0;
	}
	for(int i=1;i<n;i++){
		cin>>u>>v;
		son[u].push_back(v);
		son[v].push_back(u);
	}
	dfs(1,0,1.0);
	printf("%.7f\n",ans);
	return 0;
}