K Smallest Sums(优先队列+思维推广)
来源:互联网 发布:产品外观结构设计软件 编辑:程序博客网 时间:2024/06/06 04:34
You’re given k arrays, each array has k integers. There are k
k ways to pick exactly one element in each
array and calculate the sum of the integers. Your task is to find the k smallest sums among them.
Input
There will be several test cases. The first line of each case contains an integer k (2 ≤ k ≤ 750). Each of
the following k lines contains k positive integers in each array. Each of these integers does not exceed
1,000,000. The input is terminated by end-of-file (EOF).
Output
For each test case, print the k smallest sums, in ascending order.
Sample Input
3
1 8 5
9 2 5
10 7 6
2
1 1
1 2
Sample Output
9 10 12
2 2
给了n*n的数,每一行只能选一个数,然后n列选出n个数,他们的和记录下来,求最小的k个和(选的方式不同就算不一样的)
一开始想法就是优先队列,因为n的n次方肯定会炸,一定有的东西是要不搜的,就想到先对每行排序,然后最小的就可以出来,把每行后面的数与这行最小的数进行比较,然后根据差值入队,找次小,但是很快发现,这是没有一个标准比较的,意思就是根据入队不能将满足条件的数用队列表示出来。
然后看了lrj的想法- -,先是简化版的,只有两列数,那么就会有一个表(先对每一行排序)
a1 + b1 <= a1 + b2 <= a1 + b3 <= ……<=a1 + bn
a2 + b1 <= a2 + b2 <= a3 + b3<= ……<= a2 + bn
……
an + b1 <= an + b2 <= an + b3 <= …… <= an + bn
这样做的目的就是可以每次找出n个数,来代表所有的数,因为最小值一定在这n个数中产生。
这里还有一个存储技巧,就是用(s,b)来进行操作
(s,b)就是s = Aa +Bb;那么(s,b + 1) = (s,b) - Bb + B(b + 1);就可以了
现在这里是n行,就需要把这个想法推广。
假设我之前找了n个最小值,那么再来一行n个数,就算我只用了这里面的一个数,我也可以产生n个最小值,意思就是说,无论多么极限,更新之后的n个最小值一定是从这n个里面产生的。
数学归纳的思想。
#include<cstdio>#include<cstring>#include<iostream>#include<queue>#include<vector>#include<algorithm>#include<string>#include<cmath>#include<set>#include<map>#include<vector>using namespace std;typedef long long ll;const int inf = 0x3f3f3f3f;const int maxn = 1005;int a[800],b[800];int ans[800];int leap[800];struct Node{ int y,num; bool operator < (Node a)const { return num > a.num; }};void work(int n){ priority_queue<Node>p; Node temp; Node now; for(int i = 1;i <= n;i++) { temp.num = a[i] + b[1]; temp.y = 1; p.push(temp); } for(int i = 1;i <= n;i++) { temp = p.top();p.pop(); a[i] = temp.num; if(temp.y < n) { now.num = temp.num - b[temp.y] + b[temp.y + 1]; now.y = temp.y + 1; p.push(now); } }}int main(){#ifdef LOCAL freopen("C:\\Users\\ΡΡ\\Desktop\\in.txt", "r", stdin); //freopen("C:\\Users\\ΡΡ\\Desktop\\out.txt","w",stdout);#endif // LOCAL int n; while (scanf("%d", &n) != EOF) { for(int i = 1;i <= n;i++) scanf("%d",&a[i]); sort(a + 1,a + 1 + n); for(int i = 2;i <= n;i++) { for(int j = 1;j <= n;j++)scanf("%d",&b[j]); sort(b + 1,b + 1 + n); work(n); } for(int i = 1;i <= n;i++) { printf("%d",a[i]); if(i < n)printf(" "); else printf("\n"); } } return 0;}
- K Smallest Sums(优先队列+思维推广)
- (优先队列)K Smallest Sums
- UVA 11997 K Smallest Sums(优先队列)
- UVA11997 K Smallest Sums(并归,优先队列)
- UVA 11997 K Smallest Sums(优先队列)
- UVA 11997 K Smallest Sums(优先队列)
- UVa 11997 K Smallest Sums / 优先队列
- UVA 11997 K Smallest Sums 优先队列
- UVA 11997 K Smallest Sums(优先队列)
- K Smallest Sums uva+优先队列
- uva 11997 - K Smallest Sums(优先队列)
- uva_11997,K Smallest Sums优先队列
- UVA 11997 - K Smallest Sums 优先队列
- uva11997 K Smallest Sums(优先队列)
- uva 11997 K Smallest Sums 优先队列
- UVa 11997 K Smallest Sums (优先队列)
- UVa11997 K Smallest Sums 归并 + 优先队列
- uva11997 K Smallest Sums(优先队列)
- 简单易学代码块(block)
- PostgreSQL中使用动态SQL-实现自动按时间创建表分区
- C# 委托实例(跨窗体操作控件)
- 数据库SQL中Like的用法总结
- Tomcat关闭过程(Tomcat源码解析四)
- K Smallest Sums(优先队列+思维推广)
- 数据库课堂笔记
- 某公司的面试题目
- NSCache在AFNetworking及SDWebImage中的使用流程
- div+css 如何让背景层半透明 内容不透明 并且能兼容ie6,ie7,ie8,ff
- Python 2.7.x 与 Python 3.x 的主要差异( Published Nov 10 2014 BY ShinChan)
- exp/imp expdp/impdp乱码
- java.sql.Date 和java.util.Date 互转
- 计算几何与图形学有关的几种常用算法(二)