LeetCode 1605. 给定行和列的和求可行矩阵（贪心、矩阵）

岁忧

已于 2024-05-15 20:46:35 修改

阅读量124

点赞数 7

分类专栏： LeetCode 文章标签： leetcode 矩阵算法 c++ 贪心贪心算法

于 2024-05-15 20:44:54 首次发布

本文链接：https://blog.csdn.net/weixin_46028214/article/details/138922673

版权

LeetCode 专栏收录该内容

235 篇文章 0 订阅

订阅专栏

1605. 给定行和列的和求可行矩阵

在这里插入图片描述

思路：大佬题解，方法一，每次到达一个位置[i,j]，我们选最小的min(rowSum[i],colSum[j])来赋值，这样就不会超过某一行或某一列的和。赋值完要记得进行更新操作。

class Solution {
public:
    vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
        int n=rowSum.size(),m=colSum.size();
        vector<vector<int>> v(n,vector<int>(m,0));
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(rowSum[i]<colSum[j]){
                    v[i][j]=rowSum[i];
                    colSum[j]-=rowSum[i];
                    rowSum[i]=0;
                }else{
                    v[i][j]=colSum[j];
                    rowSum[i]-=colSum[j];
                    colSum[j]=0;
                }
            }
        }
        return v;
    }
};

思路：方法二，每次相当于只能向下和向右走。

class Solution {
public:
    vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
        int n=rowSum.size(),m=colSum.size();
        vector<vector<int>> v(n,vector<int>(m,0));
        for(int i=0,j=0;i<n&&j<m;){
            if(rowSum[i]<colSum[j]){
                v[i][j]=rowSum[i];
                colSum[j]-=rowSum[i];
                i++;
            }else{
                v[i][j]=colSum[j];
                rowSum[i]-=colSum[j];
                j++;
            }
        }
        return v;
    }
};