UVA.11181 Probability|Given (概率)

本文解析了UVA.11181概率问题,针对n个人中r人购买商品的情况,通过枚举所有可能性来计算每个人实际购买商品的概率,并提供了完整的C++实现代码。

UVA.11181 Probability|Given (概率)

题意分析

n个人去逛超市,r人买了东西(rn),并且知道每个人买东西的概率,求解每个人实际买东西的可能性是多少。

要在n个人中枚举所有的r人买东西的情况,举个例子。n=3,r=2,所有的可能情况有110,101,011。共三种。对应每种的可能性为P1P2(1P3),P1(1P2)P3,(1P1)P2P3,对三项求和得到P(E).

对应每个人,当其位置为1的时候,求和得到P(Ei)

对应每个人的概率即为

Pi=P(Ei)PE

代码总览

#include <bits/stdc++.h>
#define nmax 25
using namespace std;
double p[nmax];
bool choose[nmax];
double ans[nmax];
double tot = 0;
int n,r;
int times = 0;
void dfs(int depth,int remain){
    if(depth > n) return;
    if(n - depth < remain) return;
    if(remain == 0){
        for(int i = depth;i<n;++i) choose[i] = 0;
        double temp = 1;
        for(int i = 0;i<n;++i){
            if(choose[i]) temp *= p[i];
            else temp *= 1.0*(1-p[i]);
        }
        tot+=temp;
        for(int i = 0;i<n;++i){
            if(choose[i]) ans[i]+=temp;
        }
        return;
    }
    choose[depth] = 1;
    dfs(depth+1,remain-1);
    choose[depth] = 0;
    dfs(depth+1,remain);
}
int main()
{
//    freopen("in.txt","r",stdin);
    int kase  = 1;
    while(scanf("%d %d",&n,&r) != EOF){
        memset(ans,0,sizeof ans);
        tot = 0;
        if(n == 0 && r == 0) break;
        for(int i = 0;i<n;++i) scanf("%lf",&p[i]);
        dfs(0,r);
        printf("Case %d:\n",kase++);
        for(int i = 0;i<n;++i){
            printf("%f\n",ans[i]/tot);
        }
    }
    return 0;
}
# 安全计算二项分布概率 def binomial_probability(n, k, p) log_comb = Math.lgamma(n + 1)[0] - Math.lgamma(k + 1)[0] - Math.lgamma(n - k + 1)[0] log_prob = log_comb + k * Math.log(p) + (n - k) * Math.log(1 - p) Math.exp(log_prob) end # 定义参数 n_values = (11..27).to_a.sort p_value = 1/16.0 max_k = n_values.max # 创建数据容器 results = {} # 计算所有概率 puts "开始计算二项分布概率 (p=#{p_value})..." n_values.each do |n| results[n] = {} (0..n).each do |k| results[n][k] = binomial_probability(n, k, p_value) end end # ===== 矩阵形式打印全部数据 ===== puts "\n二项分布概率矩阵 (p=#{p_value}, n从#{n_values.min}到#{n_values.max})" puts "=" * (10 * (max_k + 2) + 10) # 动态宽度分隔线 # 打印表头 (k值) header_row = "n\\k |" (0..max_k).each do |k| header_row += sprintf " %8d", k end puts header_row puts "-" * (10 * (max_k + 2) + 10) # 分隔线 # 打印数据行 n_values.each do |n| # 打印n值作为行首 row_str = sprintf "%4d |", n # 添加每个k值的概率 (0..max_k).each do |k| if k <= n # 显示实际计算值 row_str += sprintf " %8.4f", results[n][k] else # k > n 时显示0(填充NA) row_str += " 0.000" end end puts row_str end puts "=" * (10 * (max_k + 2) + 10) # 结束分隔线 # ===== 保存到CSV文件 ===== require 'csv' filename = "binomial_matrix.csv" CSV.open(filename, "w") do |csv| # 写入表头 header = ["n\\k"] + (0..max_k).to_a csv << header # 写入数据行 n_values.each do |n| row = [n] (0..max_k).each do |k| if k <= n row << results[n][k] else row << 0.0 # 使用0填充NA end end csv << row end end # 输出完成信息 puts "\n数据已保存到 #{filename} (使用0填充NA数据)" puts "矩阵维度: #{n_values.size}行 x #{max_k + 1}列" 根据以上代码,计算这个二项分布概率的数学期望,完成输出,有打印,有csv文件保存。
12-05
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值