请写一个程序，随机打印“0”、“1”、“2”，但保证“0”、“1”、“2”出现的概率是可配置的，例如2：1：3。

请写一个程序，随机打印“0”、“1”、“2”，但保证“0”、“1”、“2”出现的概率是可配置的，例如2：1：3。

 

#include <iostream>

using namespace std;

 

#include <time.h>

 

void Rand012(itn __0, int __1, int __2, int __count) {

int _0 = __0;

int _1 = __1;

int _2 = __2;

int count_0 = count_1 = count_2 = 0;

float real_0 = real_1 = real_2 = 0.0;

 

real_0 = 1.0 * __count * _0 / (_0 + _1 + _2);

real_1 = 1.0 * __count * _1 / (_0 + _1 + _2);

real_2 = 1.0 * __count * _2 / (_0 + _1 + _2);

 

srand(time(0));

 

int sum = __count;

for(int i = 0; i < sum; i++) {

int num = rand() % __count;

if(num <= real_0 && count_0 < real_0) {

cout << 0 << '  ';

count_0++;

} else if(num > real_0 && num <= real_0 + real_1 && count_1 < real_1) {

cout << 1 << '  ';

count_1++;

} else if(num > real_0 + real_1 && num <= real_0 + real_1 + real_2 && count_2 < real_2) {

cout << 2 << '  ';

} else {

sum++;

}

}

 

cout << endl;

 

cout << "-------------------------------request----------------------" << endl;

cout << "_0: " << real_0 << endl;

cout << "_1: " << real_1 << endl;

cout << "_2: " << real_2 << endl;

 

cout <<"-------------------------------test-------------------------" << endl;

cout << "count_0: " << count_0 << endl;

cout << "count_1: " << count_1 << endl;

cout << "count_2: " << count_2 << endl;

}