题目描述
六一到了，为了庆祝这个节日，好多商家都推出了很多好玩的小游戏。Tongtong看到了一个猜球球的游戏，有n种除了颜色之外完全相同的球，商家从中拿出来一个球球放到了箱子里，已知第i种颜色的球出现在箱子里的概率为ai。Tongtong可以用下面这种方法来确定箱子中球的颜色：向商家提出猜测：“是第x种颜色的球球或第y种颜色的球球或...........中的一个”，商家会回答你的猜测是正确还是错误的，直到你有百分百的把握确定箱子里的球球，猜测的次数越少，Tongtong能够得到的礼物就更好。为了让Tongtong过一个开开心心的六一，请你找出一种最优的策略，尽可能少的向店主提出猜测来确定球的颜色，输出猜测次数的期望值。
策略“最优”是指：猜测次数的期望最小。当你有百分百的把握确定箱子里的球球颜色种类时，则不需要继续猜测。例如，如果有两种颜色的球球，箱子里放的是第二种颜色的球球，你可以猜测“是第一种颜色的球球”。商家会告诉你“错误”，所以你可以推测“箱子里的球球是第二种颜色的”，并且有百分百的把握，所以你就可以结束猜测而不需要额外的一次猜测。这种询问方式猜测次数为一次（不管你这一次有没有猜对）。
输入
第一行输入一个n，表示有n种颜色的球。 n<=2 000
第二行输入n个非负小数a1~an，表示是第i种颜色球球的概率，保证加和起来为1。
输出
输出最小期望，保留7位小数。
样例输入
复制样例数据
3 0.5000000000 0.2500000000 0.2500000000
样例输出
1.5000000
提示
最佳策略下：第一次询问“是不是第二种颜色的球球或第三种颜色的球球中的一种”，如果回答“否”则可以知道是第一种颜色的球球，结束询问；如果回答是“是”则询问第二次“是不是第二种颜色的球球”。
思路：询问决策可构成一个二叉决策树；这种询问决策下的询问次数的期望等于∑(每个叶子结点的深度*出现概率)；要使其最小，则构建哈夫曼树。注意概率为0的球不需考虑。
AC代码：
#include<bits/stdc++.h>
using namespace std;
double p[2005];
struct Node{
double p;
bool operator <(const Node &b)const{
return p>b.p;
}
};
priority_queue<Node> q;
int main()
{
int n;scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%lf",&p[i]);
if(p[i]!=0) q.push(Node{p[i]});
}
double ans=0;
while(q.size()>=2){
double p1=q.top().p;q.pop();
double p2=q.top().p;q.pop();
ans+=(p1+p2);
q.push(Node{p1+p2});
}
printf("%.7f\n",ans);
return 0;
}
原文地址：https://www.cnblogs.com/lllxq/p/11143381.html