uva 11997(priority_queue)

题意:有k个组,每组k个数,要从每组挑出一个数,k个组相加,一共有k^k种和,要求升序输出k个最小的和。

题解:书上的题解的方法是用优先队列和两两归并,先看两个排好序的数组,(s, b)代表一个元素和,让s = Aa+Bb,那么下一个可能元素和就是(s', b + 1),s' = s - Bb + Bb+1,然后优先队列每次取出队首就是此时队伍中和最小的情况,然后得到s'入队,因为是排好序就不会重复,循环k - 1次得到k个数组中最小的k个和。

#include <stdio.h>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 755;
int n, arr[N][N];
struct Item {
	int s, b;
	Item(int s, int b): s(s), b(b) {}
	bool operator < (const Item& temp) const {
		return s > temp.s;
	}
};

void merge(int* A, int* B, int* C) {
	priority_queue<Item> q;
	for (int i = 0; i < n; i++)
		q.push(Item(A[i] + B[0], 0));
	for (int i = 0; i < n; i++) {
		int bb = q.top().b;
		C[i] = q.top().s;
		q.pop();
		if (bb + 1 < n)
			q.push(Item(C[i] - B[bb] + B[bb + 1], bb + 1));
	}
}

int main() {
	while (scanf("%d", &n) == 1) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++)
				scanf("%d", &arr[i][j]);
			sort(arr[i], arr[i] + n);
		}
		for (int i = 1; i < n; i++)
			merge(arr[0], arr[i], arr[0]);
		printf("%d", arr[0][0]);
		for (int i = 1; i < n; i++)
			printf(" %d", arr[0][i]);
		printf("\n");
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值