C ++中的随机翻转矩阵
假设我们有一个二进制矩阵，行数为n_rows，列数为n_cols。这里所有的值最初都是0。我们必须定义一个函数flip()，该函数随机地均匀选择0值，将其更改为1，然后返回该值的位置[row.id，col.id]。同样，我们必须编写另一个函数reset()，将所有值都设置为0。我们必须尝试最小化对系统Math.random()的调用次数，并优化时间和空间复杂度。
如果我们具有2x3阶矩阵，并且调用四次翻转，则结果将为[0,1]，[1,2]，[1,0]，[1,1]。
让我们看下面的实现以更好地理解-示例
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int> v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << v[i] << ", ";
}
cout << "]"<<endl;
}
class Solution {
public:
unordered_map <int, int> holes;
int n;
int m;
int size;
Solution(int n_rows, int n_cols) {
srand(time(NULL));
n = n_rows;
m = n_cols;
size = n * m;
}
vector<int> flip() {
int id = rand() % size;
size--;
int rid = id;
if(holes.count(id)){
id = holes[id];
}
holes[rid] = holes.count(size) ? holes[size] : size;
return {id / m, id % m};
}
void reset() {
size = n * m;
holes.clear();
}
};
main(){
Solution ob(2,2);
print_vector(ob.flip());
print_vector(ob.flip());
print_vector(ob.flip());
print_vector(ob.flip());
}输入值
初始化构造函数使用 2,2 以及调用 flip4次
输出结果
[1, 1, ] [0, 0, ] [1, 0, ] [0, 1, ]