在C ++中重建2行二进制矩阵

假设我们具有n列2行的矩阵的以下详细信息-

矩阵元素将为0或1

第0（上）行的元素总和作为上限给出。

第1个（较低）行的元素总和较小。

第i列（索引为0）中的元素之和为colsum[i]，其中colsum作为长度为n的整数数组给出。

任务是用上，下和求和重建矩阵。我们必须找到它作为2D整数数组。如果有多个有效解决方案，则将接受其中任何一个。如果没有有效的解决方案，则返回一个空的2D数组。因此，如果输入像上=2，下=3并且colsum是[1,1,1]，那么输出将是[[1,1,0]，[0,0,1]]

为了解决这个问题，我们将遵循以下步骤-

设置标志：=true，n：=c的大小，使一个数组ans的顺序为2*n

当我在0到n的范围内

如果c[i]>0，则设置标志：=false

如果u>l，则将u减1，ans[0，i]：=1

否则，当u<l时，将l减小1ans[1，i]：=1

否则当c[i]=1时

如果u>0，则将u减1，然后ans[0，i]：=1

否则l>0，然后将l减1，然后ans[1，i]：=1

否则设置标志：=false

将u和l减1

如果u<0或l<0，则标志：=false

将ans[0，i]=1和ans[1，i]=1

如果c[i]=2，则

否则，如果c[i]=1，则

否则c[i]=0

否则设置标志：=false

如果flag为false或u不为0或l不为0，则返回空

返回ans

让我们看下面的实现以更好地理解-

示例

#include <bits/stdc++.h> using namespace std; void print_vector(vector<vector<auto> > v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << "["; for(int j = 0; j <v[i].size(); j++){ cout << v[i][j] << ", "; } cout << "],"; } cout << "]"<<endl; } class Solution { public: vector<vector<int>> reconstructMatrix(int u, int l, vector<int>& c) { bool flag = true; int n = c.size(); vector < vector <int> > ans(2, vector <int> (n)); for(int i = 0; i < n; i++){ if(c[i] == 2){ u--; l--; if(u<0 || l<0)flag = false; ans[0][i] = 1; ans[1][i] = 1; }else if(c[i] == 1){ if(u>l){ u--; ans[0][i] = 1; }else if(u<l){ l--; ans[1][i] = 1; }else{ if(u>0){ u--; ans[0][i] = 1; }else if(l > 0){ l--; ans[1][i] = 1; }else flag = false; } }else if(c[i] == 0){ if(c[i]>0)flag = false; }else{ flag = false; } } if(!flag || u!=0 ||l!=0 )return {}; return ans; } }; main(){ vector<int> v = {1,1,1}; Solution ob; print_vector(ob.reconstructMatrix(2,1,v)); }

输入值

2 1 [1,1,1]

输出结果

[[1, 1, 0, ],[0, 0, 1, ],]