对于多项分布进行采样的函数

function  x = gen_Multinom(p)
       % 生成多项式分布
     
       p = cumsum(p);
       pmax = max(max(p))+1;
       u = repmat(rand(1,size(p,2)).*p(end,:), size(p,1), 1);   
       m = (u < p) .* (pmax-p);  
       [~,x] = max(m);   % x返回的是最大值的索引，在第几行

end

从多项分布中采样的函数