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

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;

}