随机洗牌

#include<iostream>
#include<vector>
#include<cmath>
#include<random>
#include<cstdlib>
using namespace std;

/*

假设一维矩阵,200个元素,下标为0-199,雷的个数为20个.雷所在位置为1，非雷的位置为0,将20个雷随机分配到矩阵的各个位置。
初始化最后20个元素为1(即当前雷所在位置为1，非雷的位置为0)
先选第199个数的位置，然后从0-199中随机一个数（随机数对200取余）,交换他们的位置
swap(a[random()%200], a[199])
然后选第198个位置
swap(a[random()%200], a[198])
...
...
...
这些都是等概率的交换，如交换倒数第二个位置a[198]和a[random()]
第一次选择199的位置的概率为1/200
第二次选择198的位置的概率为:第一次没选择198的概率为199/200 * 第二次选择198的概率为1/199 = 1/200
即第一次和第二次的概率相等
...
...
...
*/
int main()
{
    vector<vector<int> > a(10, vector<int>(20, 0));
    int m = 20; //随机数的个数
    int i = a.size()-1; //行
    int j = a[0].size()-1; //列
    int m_cp = m;
    while(m_cp--) //初始化
    {
        a[i][j--] = 1;
        if(j < 0)
        {
            i--;
            j=a[0].size()-1;
        }
    }

    i = a.size(); //行
    j = a[0].size(); //列
    int cur_val = i * j;
    while(m--) //随机洗牌,交换m次
    {
        srand(time(NULL));
        int rd = rand() % (cur_val--);
        int row = rd / j;    //对列取整得到行数
        int colomn = rd % j; //对列取余得到列数
        swap(a[row][colomn], a[cur_val/j][cur_val%j]);
    }

    for(int i = 0; i < a.size(); i++)
    {
        for(int j = 0; j < a[i].size(); j++)
            cout << a[i][j] << " ";
        cout << endl;
    }
    return 0;
}