LeetCode 1253. 重构 2 行二进制矩阵（贪心）

示例 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

class Solution {
public:
    vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum) {
        int i, n = colsum.size(), sum = 0;
        for(i = 0; i < n; ++i)
        	sum += colsum[i];
        if(upper+lower != sum || upper > n || lower > n)
        	return {};	//肯定可以排除的情况
        vector<vector<int>> ans(2, vector<int> (n,0));
        for(i = 0; i < n; ++i)
        {
        	if(colsum[i] == 0)
        		continue;
        	if(colsum[i] == 2)
        	{
        		ans[0][i] = ans[1][i] = 1;
        		upper--,lower--;
        	}
        	else// colsum[i] == 1，平均着分配（贪心）
        	{	//优先分配多的，后面万一有2，别一个多，一个少，少的不够了
        		if(lower >= upper)
        		{
        			ans[1][i] = 1;
        			lower--;
        		}
        		else
        		{
        			ans[0][i] = 1;
        			upper--;
        		}
        	}

        	if(lower < 0 || upper < 0)
    			return {};	//不够了，肯定错了
        }
        return ans;
    }
};