题目描述
给你一个 2 行 n 列的二进制数组：矩阵是一个二进制矩阵，这意味着矩阵中的每个元素不是 0 就是 1。第 0 行的元素之和为 upper。第 1 行的元素之和为 lower。第 i 列（从 0 开始编号）的元素之和为 colsum[i]，colsum 是一个长度为 n 的整数数组。你需要利用 upper，lower 和 colsum 来重构这个矩阵，并以二维整数数组的形式返回它。
如果有多个不同的答案，那么任意一个都可以通过本题。
如果不存在符合要求的答案，就请返回一个空的二维数组。
示例 1：
输入：upper = 2, lower = 1, colsum = [1,1,1]
输出：[[1,1,0],[0,0,1]]
解释：[[1,0,1],[0,1,0]] 和 [[0,1,1],[1,0,0]] 也是正确答案。示例 2：
输入：upper = 2, lower = 3, colsum = [2,2,1,1]
输出：[]示例 3：
输入：upper = 5, lower = 5, colsum = [2,1,2,0,1,0,1,2,0,1]
输出：[[1,1,1,0,1,0,0,1,0,0],[1,0,1,0,0,0,1,1,0,1]]提示：
1 <= colsum.length <= 10^5
0 <= upper, lower <= colsum.length
0 <= colsum[i] <= 2
解题思路[1]
使用贪心算法，大致思路为：
- 若 colsum[i]=2colsum[i]=2，则 一定上下均为 1；
- 若 colsum[i]=0colsum[i]=0，则 一定上下均为 0；
- 若 colsum[i]=1colsum[i]=1，则 上下一个 1 一个 0。
唯一需要讨论的只有 colsum[i]=1colsum[i]=1 的情形，我们可以规定：在把2分配完之后，先分配上为 1，再分配下为 1。
代码如下（代码有点长，但是思路很容易看懂）
class Solution {
public:vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum) {int length=colsum.size(),sum=0;vector <vector<int>> res;for(int i=0;i<length;i++){if(colsum[i]==2)//为2时，上下均需要分配1，此时lower和upper需要减去分配的1{lower--;upper--;}else if(colsum[i]==1)sum+=1;}//总数不相符合，或者上下某一行的数不符合则返回空vectorif(sum!=lower+upper || lower<0 || upper<0)return res;vector<int> a1,a2;for(int i=0;i<length;i++){if(colsum[i]==2){a1.push_back(1);a2.push_back(1);}else if(colsum[i]==0){a1.push_back(0);a2.push_back(0);}else//colsum[i]==1{if(upper>0)//上1，下0{a1.push_back(1);a2.push_back(0);upper--;}else if(lower>0)//上0，下1{a1.push_back(0);a2.push_back(1);lower--;}elsereturn res;}}res.push_back(a1);res.push_back(a2);return res;}
};
提交结果：
参考
- ^https://leetcode-cn.com/problems/reconstruct-a-2-row-binary-matrix/solution/java-tan-xin-suan-fa-xiang-jie-by-amanehayashi/
