UVa 11997 K Smallest Sums (优先队列)

本文介绍了一种求k个整数数组中最小k个和的算法实现,通过使用优先队列维护每一步的最小值,实现了高效求解。算法采用逐步合并的方式,每次将当前最小的k个值与下一个数组进行合并。

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

题意:有 k 个整数数组,各包含 k 个元素。在每个数组中取一个元素加起来,可以得到哦 k^k 个和。求这些和中最小的 k 个值(重复的值算多次)

思路:在《算法竞赛入门经典训练指南》中看到这道题,再自己写一写思路,加深理解。

           从 k 个数组中各取一个元素之和中前 k 个最小的值为 k 个组合,则这 k 个组合一定来自前 k - 1 个数组中各取一个元素之和中前 k 个最小的(值)组合和第 k 个数组中的一个元素的组合。因此先求出从第一个和第二个数组中各取一个元素之和中的前 k 个最小值(不用知道他们是由哪些元素组合的和),再将这 k 个值作为一个数组和第三个数组同样求出前 k 个最小的值,直到第 k 个数组,求出的最终的 k 个值即为所要答案。

           求前 k 个最小的值可用priority_queue维护,复杂度为 k log k ,若直接暴力则为 k * k 。


#include <cstdio>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;

int a[760];
int c[760];
struct Item
{
	int s, b;
	Item(int s, int b):s(s), b(b){ }
	bool operator < (const Item& rhs) const
	{
		return s > rhs.s;
	}	
};
void merge(int* C, int* B, int k)
{
	priority_queue<Item> q;
	for(int i = 0; i < k; i++)
	{
		q.push(Item(C[i] + B[0], 0));
	}
	for(int i = 0; i < k; i++)
	{
		Item item = q.top();
		q.pop();
		C[i] = item.s;
		int b = item.b;
		q.push(Item(item.s - B[b] + B[b + 1], b+ 1));
	}
}
int main()
{
	int k;
	while(~scanf("%d", &k))
	{
		memset(a, 0, sizeof(a));
		memset(c, 0, sizeof(c));
		for(int i = 0; i < k; i++)
		{
			for(int j = 0; j < k; j++)
			{
				scanf("%d", &a[j]);
			}
			sort(a, a + k);
			if(i == 0)
			{
				memcpy(c, a, sizeof(a));
			}
			else
			{
				merge(c, a, k);
			}
		}
		printf("%d", c[0]);
		for(int i = 1; i < k; i++)
		{
			printf(" %d", c[i]);
		}
		printf("\n");
	}
}





评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值