力扣练习——21 有序矩阵中的第 k 个最小数组和

1.问题描述

给你一个 m * n 的矩阵 mat,以及一个整数 k ,矩阵中的每一行都以非递减的顺序排列。

你可以从每一行中选出 1 个元素形成一个数组。返回所有可能数组中的第 k 个 最小 数组和。

 

示例 1:

输入:mat = [[1,3,11],[2,4,6]], k = 5

输出:7

解释:从每一行中选出一个元素,前 k 个和最小的数组分别是:

[1,2], [1,4], [3,2], [3,4], [1,6]。其中第 5 个的和是 7 。  

 

示例 2:

输入:mat = [[1,3,11],[2,4,6]], k = 9

输出:17

 

示例 3:

输入:mat = [[1,10,10],[1,4,5],[2,3,6]], k = 7

输出:9

解释:从每一行中选出一个元素,前 k 个和最小的数组分别是:

[1,1,2], [1,1,3], [1,4,2], [1,4,3], [1,1,6], [1,5,2], [1,5,3]。其中第 7 个的和是 9 。 

 

示例 4:

输入:mat = [[1,1,10],[2,2,9]], k = 7

输出:12

 

说明:

m ,n为矩阵 mat的行列数

1 <= m, n <= 40

1 <= k <= min(200, n ^ m)

1 <= mat[i][j] <= 5000

mat[i] 是一个非递减数组

 

可参考以下main函数:

int main()

{

    int m, n,data,k;

    vector<vector<int> > mat;

    cin>>m>>n;

    for(int i=0; i<m; i++)

    {

        vector<int> row;

        for(int j=0; j<n; j++)

        {

            cin>>data;

            row.push_back(data);

        }

        mat.push_back(row);

    }

    cin>>k;

    int res=Solution().kthSmallest(mat,k);

    cout<<res<<endl;

    return 0;

}

 

2.输入说明

首先输入矩阵mat的行列数m和n

然后输入m行,每行n个非负整数,表示mat的元素值,以空格分隔。

最后输入k。

3.输出说明

输出一个整数,表示结果。

4.范例

输入

2 3
1 3 11
2 4 6
5

输出

7

5.代码

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
#include<unordered_map>
#include<set>
using namespace std;

int kthSmallest(vector<vector<int> >mat, int k)
{
    //暴力解法
    //参考网址https://leetcode.cn/problems/find-the-kth-smallest-sum-of-a-matrix-with-sorted-rows/solution/find-the-kth-smallest-sum-of-a-matrix-by-ikaruga/
    //1.计算行数和列数
    int m = mat.size();//行数
    int n = mat[0].size();//列数
    
    vector<int>ans(mat[0]);//ans中保存第一行各数

    for (int i = 1; i < m; i++)
    {
        multiset<int>st;//multiset能时刻保证序列中的数是有序的,而且序列中可以存在重复的数。
        for (int x : ans)//在i=1时,取出第一行的数 ;i=2时,取出第一行和第二行相加后得到的前K个组合数
        {
            for (int y : mat[i]) //i=1时,第二行中取数  etc
            {
                st.insert(x + y);//i=1,将第一行和第二行的数组合相加
            }
        }
        ans.assign(st.begin(), st.end());//将相加得到的数赋值给ans,实现更新操作
        ans.resize(min(k, (int)ans.size()));//只取有序列表中前K个值
    }
    return ans.back();//最后的值就是所求的第K个最小数组和
}

 
int main()

{

    int m, n, data, k;

    vector<vector<int> > mat;

    cin >> m >> n;

    for (int i = 0; i < m; i++)

    {

        vector<int> row;

        for (int j = 0; j < n; j++)

        {

            cin >> data;

            row.push_back(data);

        }

        mat.push_back(row);

    }

    cin >> k;

    int res = kthSmallest(mat, k);

    cout << res << endl;

    return 0;

}

 

posted @ 2022-07-17 15:40  努力奋斗的小企鹅  阅读(104)  评论(0)    收藏  举报