1253. 重构 2 行二进制矩阵 (opens new window)

1.模拟

根据colsum列之和存在以下几种情况：

• 列和为2：两行矩阵对应位置元素值只能填1
• 列和为0：两行矩阵对应位置元素值只能填0
• 列和为1：先填入第一行，直到第一行的总和len1达到upper后，再填入第二行。当第二行的总和len2超过lower说明不存在符合要求答案。

class Solution {
    public List<List<Integer>> reconstructMatrix(int upper, int lower, int[] colsum) {
        Integer[] list1=new Integer[colsum.length];
        Integer[] list2=new Integer[colsum.length];
        List<List<Integer>> res=new ArrayList<>();
        int len1=0,len2=0;
        for(int i=0;i<colsum.length;i++){
            if(colsum[i]==2){
                list1[i]=1;
                list2[i]=1;
                len1++;
                len2++;
            }
        }
        for (int i = 0; i <colsum.length; i++) {
            if(colsum[i]==1){
                //策略是先填第一行
                if(len1<upper){
                    list1[i]=1;
                    list2[i]=0;
                    len1++;
                }
                //第一行总和达到upper再填第二行
                else{
                    list2[i]=1;
                    list1[i]=0;
                    len2++;
                }
            }
            else if(colsum[i]==0){
                list1[i]=0;
                list2[i]=0;
            }
			//第二行也满了直接返回空
            if(len2>lower) return res;
        }
        if(len1==upper&&len2==lower){
            res.add(Arrays.asList(list1));
            res.add(Arrays.asList(list2));
        }
        return res;
    }
}