数据结构与算法实验题 6.1 s_sin’s bonus

数据结构与算法实验题 6.1 s_sin’s bonus
★实验任务
正如你所知道的 s_sin 是一个非常贪玩的人 QAQ(如果你非常讨厌他请直接从第二段开
始看)，并且令人感到非常遗憾的是，他是一只非常非常穷的 ds，为了改变自己的经济情况
他决定外出打工，但是因为 s_sin 是一只死宅，力量魔法耐力速度运气五围都差到爆表 QAQ。
正当他对人生无比绝望的时候，一个游戏店的老板找到他请他帮助设计一个游戏，并答应给
他一定的报酬。游戏的内容如下：
玩家从 n 个点 n-1 条边的图，从节点 1 丢下一个小球，小球将由于重力作用向下落，而
从小球所在点延伸出的每一条边有一个值 pi 为小球通过该条边的概率（注意从同一个节点
向下延伸的所有边的 pi 的和可以小于 1，也可以大于 1，并且保证对于单独的一条边不会出
现 pi>1 的情况），而对于所有处于最下方的节点（如图红点所示）都可以有一个值 vi，代
表玩家可以获得的奖励。现在老板给你这样一张图，之后给你 n 个 vi 的值，老板希望玩家
可以获得的奖励的期望值最小。（对题目不理解可以参见样例）
Ps：小球不会逆着重力往回滚 QAQ。保证所给出的图无重边。
★数据输入
输入第一行为一个正整数 N (2 < N < 10000)， 表示有 n 个节点，编号为 1 到 N。
接下来 N-1 行，每行三个整数 a b pi ，表示从 a，b 之间有一条路径，经过这条路径的
可能性为 pi。
接下来一行为有 n 个整数，表示 n 个 vi 的值（10000>=vi>0）。 ★数据输出
对于每个询问，输出一行一个数精度要求为.10lf，表示最小的奖励期望值。
输入示例 输出示例
7
1 2 0.8
1 3 0.2
2 4 1.0
4 7 1.0
3 5 0.7
3 6 0.3
1 2 3 4 5 6 7
1.2600000000
★HINT
分数值最小期望为 0.8*1+0.14*2+0.06*3=1.26
图如左图所示

期望值及其公式：http://zh.wikipedia.org/wiki/%E6%9C%9F%E6%9C%9B%E5%80%BC

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f
struct node
{
	int a;
	double pi;
}p[10005];
bool cmp(double a,double b)
{
	return a<b;
}
bool bmp(int a,int b)
{
	return a<b;
}
int a[10005];
double b[10005];
int main()
{
	memset(a,INF,sizeof(a));
	memset(b,INF,sizeof(b));
	int n,i,j,m,z;
	double f,sum=0.0;
	scanf("%d",&n);
	for(i=0;i<n;i++)
		p[i].a=-1;
	for(i=0;i<n-1;i++)
	{
	
		scanf("%d %d %lf",&m,&z,&f);
		p[m].a=1;
		p[z].a=0;
		if(i==0)
		{
				p[m].pi=1.0;

		}
			
			p[z].pi=f*p[m].pi;
	
	}

	j=0;
	for(i=0;i<=n;i++)
	{
		if(p[i].a==0)
		{
			b[j++]=p[i].pi;
		
		}
	}
	sort(b,b+j,cmp);
	for(i=0;i<n;i++)
		scanf("%d",&a[i]);
	sort(a,a+n,bmp);
	i=0;
	j--;
	
	while(j>=0)
	{
		sum+=a[i]*b[j];
		j--;
		i++;
	}
	printf("%.10lf\n",sum);
}