【每日一题Day250】LC1253重构 2 行二进制矩阵 | 贪心模拟

给定一个二行n列的二进制矩阵,其中第0行和第1行的元素之和分别为upper和lower,每列的和为colsum数组中的值。首先,将colsum中为2的位置设为1,然后检查剩余的1是否能均匀分配给upper和lower,若不能则无解。最后,将剩余的1按顺序分配给upper和lower,返回重构后的矩阵。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

重构 2 行二进制矩阵【LC1253】

给你一个 2n 列的二进制数组:

  • 矩阵是一个二进制矩阵,这意味着矩阵中的每个元素不是 0 就是 1
  • 0 行的元素之和为 upper
  • 1 行的元素之和为 lower
  • i 列(从 0 开始编号)的元素之和为 colsum[i]colsum 是一个长度为 n 的整数数组。

你需要利用 upperlowercolsum 来重构这个矩阵,并以二维整数数组的形式返回它。

如果有多个不同的答案,那么任意一个都可以通过本题。

如果不存在符合要求的答案,就请返回一个空的二维数组。

  • 思路

    如果colsum中某个位置为2,那么两行中该位置都为1;如果某个位置为1,那么设置任意一行为1,另一行为0。因此可以先将一定是1的位置赋值为1,并计算需要赋值为1的位置和 upperlower中剩余的数目是否相等,如果不相等,那么表示不可构造。否则,我们可以构造出该数组,由于只需要返回一个可能的答案,那么先将upper赋值为1,再将lower赋值为1

  • 实现

    class Solution {
        public List<List<Integer>> reconstructMatrix(int upper, int lower, int[] colsum) {
            List<List<Integer>> res = new ArrayList<>();
            int n = colsum.length;
            int[][] cur = new int[n][2];
            int count1 = 0;
            for (int i = 0; i < n; i++){
                if (colsum[i] == 2){
                    cur[i][0] = 1;
                    cur[i][1] = 1;
                    upper -= 1;
                    lower -= 1;
                    colsum[i] = 0;
                }else if (colsum[i] == 1){
                    count1++;
                }
            }
            if (upper < 0 || lower < 0 || upper + lower != count1){
                return res;
            }
            for (int i = 0; i < n; i++){
                if (colsum[i] == 1){
                    if (upper > 0){
                        cur[i][0] = 1;
                        upper--;
                    }else{
                        cur[i][1] = 1;
                        lower--;
                    }
                }
            }
            List<Integer> u = new ArrayList<>();
            List<Integer> l = new ArrayList<>();
            for (int i = 0; i < n; i++){
                u.add(cur[i][0]);
                l.add(cur[i][1]);
            }
            res.add(u);
            res.add(l);
            return res;
        }
    }
    
    • 复杂度
      • 时间复杂度:O(nlogn)\mathcal{O}(nlogn)O(nlogn)
      • 空间复杂度:O(n)\mathcal{O}(n)O(n)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值