uva11997(优先队列,归并)

题意:

k个数组,每个数字k个值;

如果每个数组取一个值相加,那么总共有k^k种结果,取前k小的值,输出;


思路:

首先我们两个数组,两个数组找出前k小的和了,在加入第三个,这样一直两两算;

因为两个数组取出前k小的和了;那么加入第三个数组后,那么新的前k小肯定是第三个数组的值和那之前前k的值的和;

而求两个数组,和的前k小,我们可以用优先队列,加上一个动态规划;

只有AB两个数组.

那么S = A[i] + B[j] 就可以推出A[i] +B[j + 1];

即S - B[j] +B[j +1];

具体可以看刘汝佳书里的解析;


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

struct point {
	int s, b;
	point(int a, int c) {
		s = a;
		b = c;
	}
	bool operator <(point a)const {
		return this->s > a.s;
	}
};

const int N = 1000;
int A[N][N];
int n;

void merge(int* A, int* B, int* C, int n) {
	priority_queue<point> q;
	for(int i = 0; i < n; i++)
		q.push(point(A[i] + B[0], 0));
	for(int i = 0; i < n; i++) {
		point p = q.top();
		q.pop();
		C[i] = p.s;
		int b = p.b;
		if(b + 1 < n) 
			q.push(point(p.s - B[b] + B[b + 1], b + 1));
	}
}
int main() {
	while(scanf("%d",&n) == 1) {
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				scanf("%d",&A[i][j]);
			}
			sort(A[i], A[i] + n);
		}
		for(int i = 1; i < n; i++)
			merge(A[0], A[i], A[0], n);
		printf("%d",A[0][0]);
		for(int i = 1; i < n; i++) {
			printf(" %d",A[0][i]);
		}
		printf("\n");
	}
	return 0;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值