生成带权重的随机数

我会带着你远行 2021-12-03 12:41 376阅读 0赞

#### 2 ####

随机从10个数里面随机3个出来。
    每个数有一个权重值.
    1   概率:8000
    2   概率:1000
    3   概率:500
    4   概率:10
    5   概率:10
    6   概率:10
    7   概率:10
    8   概率:10
    9   概率:10
    10  概率:10
    假设已知随机数生成函数，均匀随机生成一个[min, max]区间的数
    int myRand(int min, int max);

生成具有权重的随机数。随机数ranIndex生成的区间为权重的总和，根据权重分割子区间。  
![在这里插入图片描述][20190718152444590.png]

//************************************
    // Method:    GetRandomNumWithWeight
    // Desc:	  生成具有权重的随机数。随机数ranIndex生成的区间为权重的总和，根据权重分割子区间。
    // Author:    ljm 
    // Date: 	  2019/07/18
    // FullName:  GetRandomNumWithWeight
    // Access:    public 
    // Returns:   void
    // Qualifier:
    // Parameter: vector<int> weight	:权重
    // Parameter: int number			:随机数数量
    //************************************
    void GetRandomNumWithWeight(vector<int> weight, int number)
    {
    	int size = weight.size();
    
    	//计算权重的总和
    	int accumulateVal = accumulate(weight.begin(), weight.end(), 0);
    	
    	vector<int> res(number, 0);
    	for (int i = 0; i < number; i++)
    	{
    		int tempSum = 0;
    		int ranIndex = myRand(0, accumulateVal);
    
    		//0 ~ weight[0]为1，weight[0]+1 ~ weight[1]为2，依次类推
    		for (int j = 0; j < size; j++)
    		{
    			tempSum += weight[j];
    			if (ranIndex <= tempSum + weight[j])
    			{
    				res[i] = j + 1;
    				break;
    			}
    		}
    	}
    
    	for (auto num : res)
    	{
    		cout << num << " ";
    	}
    
    }

测试

int main()
    {
    	vector<int> weight = { 8000, 1000, 500, 10, 10, 10, 10, 10, 10, 10};
    	GetRandomNumWithWeight(weight, 3);
    }

输出

1
    1
    1

[20190718152444590.png]: /images/20211203/909e92191456488493eacef71ce663a3.png