
比如说按照上图的理解就是,总共现在有3列列车,比如这时候我们求车厢个数是4个的火车有多少那就应该是1,如果5节的话就是没有。
下面说一下我的思路:
一开始先设定一个10000cycle的loop然后得到一个10000个数的随机数array,然后用这个array里面每一个数去跟设定的概率p比较,如果小于p就说明这个slot有车厢,如果大于就没有,然后把每个小于p的概率的位置换成1,大于p概率的array位置换成0,这样就有了一个0,1 array去表示10000个slot里面的车厢分布情况,如下图:

然后我就想,既然我已经有这样的array了,我想直接计算出连续有i个1的数据的个数,因为连续的1就表示一个列车了,然后就能直接算出答案。可是我不是很明白怎么写那种方程就是,求出一个array里面连续i个数据的个数是多少。
所以我的code里面就用了forloop来求。
下面是我自己写的code。





