【优先队列之多路合并】UVA - 11997 K Smallest Sums

最新推荐文章于 2019-07-03 14:27:15 发布

Hacker_vision

最新推荐文章于 2019-07-03 14:27:15 发布

阅读量992

点赞数 1

分类专栏：数据结构文章标签：优先队列多路合并

本文链接：https://blog.csdn.net/u012717411/article/details/48053293

版权

数据结构专栏收录该内容

28 篇文章 0 订阅

订阅专栏

Source : UVA - 11997 K Smallest Sums

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=18702

题意

有k个整数数组，各包含k个元素，从每个数组中选取一个元素加起来，可以得到k^k个和，求这些和中最小的k个值。

示例

Sample Input
3
1 8 5
9 2 5
10 7 6
2
1 1
1 2

Sample Output
9 10 12
2 2

思路

二路归并构建二维平面表：

表 1 ： A 1 + B 1 < = A 1 + B 2 < = A 1 + B 3 \dots \dots 表 2 ： A 2 + B 1 < = A 2 + B 2 < = A 2 + B 3 \dots \dots \dots \dots 表 n ： A n + B 1 < = A n + B 2 < = A n + B 3 \dots \dots

$\begin{align}\notag 表1：A_1+B_1<=A_1+B_2<=A_1+B_3……\\ 表2：A_2+B_1<=A_2+B_2<=A_2+B_3……\\ ……\\ 表n：A_n+B_1<=A_n+B_2<=A_n+B_3……\\ \end{align}$

首先将第一列入队，第一列的最小值即为全局的最小值；接下来定位到最小值的那一行，跳过这一个元素，把下一个元素入队，始终保持priority_queue中有n个元素，队首元素即为全局下一小的元素和，记录即可。

多路归并就要构建三位立体表：想象下

k \times k \times k 的 立 方 体 表 格

$\begin{align}\notag k×k×k的立方体表格 \end{align}$

逐层二路合并即可。

★ 参考刘汝佳《算法竞赛入门经典训练指南》page 190

参考代码

#include<bits/stdc++.h>
using namespace std;

const int _max = 750 + 10;
int n,A[_max],B[_max];
struct Item{
  int s,b;//(s,b) s = Aa + Bb
  Item (){}
  Item(int s,int b):s(s) , b(b){}//构造函数
  bool operator < (const Item& a) const{//定义优先级
    return s > a.s;
  }
};
priority_queue<Item>pq;
Item item;

void merge(int A[],int B[],int C[]){//二路归并，前n个最小和存于C[]
  while(!pq.empty()) pq.pop();
  for(int i = 0; i < n; ++ i){
    item = Item(A[i] + B[0],0);
    pq.push(item); //第一列元素入队
  }
  for(int i = 0; i < n; ++ i){//O(nlogn)
    item = pq.top();pq.pop();
    C[i] = item.s;//第一列最小的一定是全局最小，跳过它读取同一行的下一个元素入队，循环n次
    int b = item.b;
    if(b + 1 < n) pq.push(Item(item.s - B[b] + B[b+1],b + 1));
  }
}

int main(){
 #ifndef ONLINE_JUDGE
 freopen("input.txt","r",stdin);
 #endif // ONLINE_JUDGE
 while(scanf("%d",&n) == 1){
    for(int i = 0; i < n; ++ i) //先读一路数据
       scanf("%d",A + i);
    sort(A , A + n);
    for(int i = 1; i < n; ++ i){ //以二路为基础做多路归并
        for(int j = 0; j < n; ++ j) scanf("%d",B+j);
        sort(B,B+n);
        merge(A,B,A);//A[]元素只用做初始化，所以在合并过程可以直接把结果存储到A中
    }
    printf("%d",A[0]);//输出格式
    for(int i = 1; i < n; ++ i) printf(" %d",A[i]);
    printf("\n");
 }
 return 0;
}