码迷,mamicode.com
首页 > 其他好文 > 详细

uva 11997 (基础数据结构)

时间:2014-06-18 17:46:28      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:style   class   blog   code   http   tar   

题意: 有一个k*k 的方阵,让你从当中每一行挑选一个数字相加最后能得到K^K次方的和,输出其中最小的k个。

思路:先对每一行排序然后两两归并,每次取前k个再和下一行再进行归并。在归并的时候用一个优先队列维护最大的k个值每次先放k个进去然后一次每行和队顶比较,若是小则替换否则break最后输出即可。

代码如下:

bubuko.com,布布扣
 1 #include <stdio.h>
 2 #include <algorithm>
 3 #include <iostream>
 4 #include <queue>
 5 #include <set>
 6 #include <string.h>
 7 #include <map>
 8 #include <vector>
 9 #include <string>
10 #include <math.h>
11 using namespace std;
12 
13 const int LEN = 2010;
14 int Num[LEN][LEN];
15 int s[LEN], n;
16 
17 void Merge(int a, int b){
18     int top = 0;
19     priority_queue<int> q;
20     for(int i=0; i<n; i++){
21         q.push(Num[a][i] + Num[b][0]);
22     }
23     for(int i=0; i<n; i++){
24         for(int j=1; j<n; j++){
25             int ls = Num[a][i] + Num[b][j];
26             int nv = q.top(); q.pop();
27             if(nv > ls){ 
28                 swap(nv, ls);
29                 q.push(nv);
30             }else {
31                 q.push(nv);
32                 break;
33             }
34         }
35     }
36     while(!q.empty()){
37         s[top++] = q.top();q.pop();
38     }
39     for(int i=top-1, j=0; i>=0; i--, j++){
40         Num[b][j] = s[i];
41     }
42 }
43 
44 int main()
45 {
46 //    freopen("in.txt","r",stdin);
47     //freopen("out.txt","w",stdout);
48 
49     ios::sync_with_stdio(false);
50     while(cin >> n){
51         for(int i=0; i<n; i++){
52             for(int j=0; j<n; j++){
53                 cin >> Num[i][j];
54             }
55             sort(Num[i], Num[i]+n);
56         }
57         for(int i=1; i<n; i++){
58             Merge(i-1, i);
59         }
60         for(int i=n-1; i>=0; i--){
61             cout << s[i];
62             if(i != 0) cout <<  ;
63         }cout << endl;
64     }
65     return 0;
66 }
View Code

 

uva 11997 (基础数据结构),布布扣,bubuko.com

uva 11997 (基础数据结构)

标签:style   class   blog   code   http   tar   

原文地址:http://www.cnblogs.com/shu-xiaohao/p/3790938.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!