Java代码蒙特卡罗方法求盒子取球概率问题

最新推荐文章于 2022-06-05 16:24:46 发布

唐力伟

最新推荐文章于 2022-06-05 16:24:46 发布

阅读量1k

点赞数

分类专栏：统计方法文章标签： Java 统计

本文链接：https://blog.csdn.net/killingbow/article/details/90510964

版权

统计方法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

蒙特卡罗方法求盒子取球概率问题

题目

三个盒子, 分别装有白球、白球，白球、黑球，黑球、黑球。
问: 当取出一个球是黑色时，再取出另外一球也是黑色的概率。

Java代码实现

public class Main {
    public static void main(String[] args) {
        Random random = new Random();
        //模拟三个盒子 0是白色， 1 是黑色
        Box[] boxes = new Box[]{
                new Box(0, 0),
                new Box(0, 1),
                new Box(1, 1)
                //当拿到一个黑色时，再取出一个的概率是多少?

        };
        //取第一个球，是黑色时的次数
        double countBlackOneCount = 0;
        //另一个个球，仍然是黑色的次数
        double countBlackTwoCount = 0;
        //总共取球次数
        int count = 1000000;
        for(int i =0;i<count;i++) {
            int selectedBox = random.nextInt(3);
            int selectedBall = random.nextInt(2);
            if(boxes[selectedBox].getBall(selectedBall) == 0){

                //取得一个黑球
            }else if(boxes[selectedBox].getBall(selectedBall) == 1){
                countBlackOneCount++;
                int otherBall;
                if(selectedBall == 0){
                    otherBall = 1;
                }else{
                    otherBall = 0;
                }
                if(boxes[selectedBox].getBall(otherBall) == 1){
                    countBlackTwoCount++;
                }
            }
        }
        System.out.println("总取球次数: " + count);
        System.out.println("取出第一个球是黑球的次数: " + countBlackOneCount);
        System.out.println("取出黑球后取另一个球仍然是黑球的次数: " + countBlackTwoCount);
        System.out.println("取出第二个球是黑球的次数/取出第一个球是黑球的次数: " + (countBlackTwoCount / countBlackOneCount) + "%");
    }

    static class Box{
        int ball0;
        int ball1;
        public Box(int ball0, int ball1){
            this.ball0 = ball0;
            this.ball1 = ball1;
        }
        public int getBall(int id){
            if(id == 0){
                return ball0;
            }else if(id == 1){
                return ball1;
            }else{
                throw new RuntimeException("Unsupported ball id: " + id);
            }
        }
    }
}