UVA11997求前k个和,多路归并问题
生活随笔
收集整理的這篇文章主要介紹了
UVA11997求前k个和,多路归并问题
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題意:
? ? ?給你一個(gè)二維矩陣,n*n的,每次從每一行中拿出來一個(gè),然后加起來組成一個(gè)和,一共可以得到n^n個(gè)和,要求求出這n^n個(gè)和中最小的那n個(gè)和。
思路:
? ? ?多路歸并問題,先說下多路歸并問題,我的理解是有個(gè)變量,每個(gè)變量都按照自己的變化規(guī)律在變化著,每次一旦選用每個(gè)變量,那么這個(gè)變量就會(huì)根據(jù)自己的變化規(guī)律變化,這種問題我們可以用優(yōu)先隊(duì)列來解決,首先我們可以把所有變量都扔進(jìn)隊(duì)列,然后在在里面取出一個(gè)最小的(或者最大的)作為當(dāng)前答案,然后把取出來的數(shù)值變化后在扔進(jìn)隊(duì)列里,如此反復(fù)操作,對(duì)于這個(gè)題目我們可以像白書上一樣這樣分析,首先我們想一個(gè)該問題的簡(jiǎn)化版,就是給你兩個(gè)數(shù)組,每個(gè)數(shù)組里面有n個(gè)數(shù)字,每次從兩個(gè)數(shù)組中任意取出一個(gè)數(shù)字加起來得到一個(gè)和,求前n個(gè)最小的和。
我們可以運(yùn)用多路歸并問題
A[1]+B[1] <= A[1]+B[2] <= A[1]+B[3]...
A[2]+B[1] <= A[2]+B[2] <= A[2]+B[3]...
A[3]+B[1] <= A[3]+B[2] <= A[3]+B[3]...
...
對(duì)于每一個(gè)集合的每一個(gè)變量我們可以這樣表示,node.s = A[a] + A[b] ,node.b =b;
這樣這個(gè)集合的下一個(gè)變量就是 node.s - B[node.b] + B[node.b + 1] ,node.b ++;
這樣說應(yīng)該沒問題吧? 然后我們就開始多路歸并,很容易理解和想到,我們直接先把第一列圈放進(jìn)隊(duì)列,然后取出一個(gè)最為最小的那個(gè),然后把取出的這個(gè)的右側(cè)(下一個(gè))放進(jìn)隊(duì)列,然后在取,在放,每次都把取出的下一個(gè)放進(jìn)去,執(zhí)行n此之后就得到了這最小的n個(gè)數(shù),這是兩個(gè)序列的,這個(gè)題目是n個(gè)序列,這樣我們可以吧第一個(gè)和第二個(gè)合并,然后在用合并的序列去合并第三個(gè),最后把所有的序列都合并,這里說的合并就是前面上多那個(gè)多路歸并。
#include<queue>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 750 + 5
using namespace std;
typedef struct NODE
{
? ? int s ,b;
? ? friend bool operator < (NODE a ,NODE b)
? ? {
? ? ? ?return a.s > b.s;
? ? }
}NODE;
int num[N][N];
int A[N] ,B[N] ,C[N];
NODE xin ,tou;
void duoluguibing(int n)
{
? ? priority_queue<NODE>q; ?
? ? for(int i = 1 ;i <= n ;i ++)
? ? {
? ? ? ?xin.b = 1;
? ? ? ?xin.s = A[i] + B[1];
? ? ? ?q.push(xin);
? ? }
? ? for(int i = 1 ;i <= n ;i ++)
? ? {
? ? ? ? tou = q.top();
? ? ? ? q.pop();
? ? ? ? C[i] = tou.s;
? ? ? ? xin.b = tou.b + 1;
? ? ? ? xin.s = tou.s - B[tou.b] + B[xin.b];
? ? ? ? if(xin.b <= n) q.push(xin);
? ? }
? ? return;
}
int main ()
{
? ? int n ,i ,j;
? ? while(~scanf("%d" ,&n))
? ? {
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?{
? ? ? ? ? for(j = 1 ;j <= n ;j ++)
? ? ? ? ? scanf("%d" ,&num[i][j]);
? ? ? ? ? sort(num[i] + 1 ,num[i] + n + 1);
? ? ? ?}
? ? ? ?
? ? ? ?
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?C[i] = num[1][i];
? ? ? ?for(i = 2 ;i <= n ;i ++)
? ? ? ?{
? ? ? ? ? for(j = 1 ;j <= n ;j ++)
? ? ? ? ? A[j] = C[j] ,B[j] = num[i][j];
? ? ? ? ? duoluguibing(n);
? ? ? ?}
? ? ? ?
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?if(i == n) printf("%d\n" ,C[i]);
? ? ? ?else printf("%d " ,C[i]);
? ? }
? ? return 0;
}
? ? ? ?
? ? ?給你一個(gè)二維矩陣,n*n的,每次從每一行中拿出來一個(gè),然后加起來組成一個(gè)和,一共可以得到n^n個(gè)和,要求求出這n^n個(gè)和中最小的那n個(gè)和。
思路:
? ? ?多路歸并問題,先說下多路歸并問題,我的理解是有個(gè)變量,每個(gè)變量都按照自己的變化規(guī)律在變化著,每次一旦選用每個(gè)變量,那么這個(gè)變量就會(huì)根據(jù)自己的變化規(guī)律變化,這種問題我們可以用優(yōu)先隊(duì)列來解決,首先我們可以把所有變量都扔進(jìn)隊(duì)列,然后在在里面取出一個(gè)最小的(或者最大的)作為當(dāng)前答案,然后把取出來的數(shù)值變化后在扔進(jìn)隊(duì)列里,如此反復(fù)操作,對(duì)于這個(gè)題目我們可以像白書上一樣這樣分析,首先我們想一個(gè)該問題的簡(jiǎn)化版,就是給你兩個(gè)數(shù)組,每個(gè)數(shù)組里面有n個(gè)數(shù)字,每次從兩個(gè)數(shù)組中任意取出一個(gè)數(shù)字加起來得到一個(gè)和,求前n個(gè)最小的和。
我們可以運(yùn)用多路歸并問題
A[1]+B[1] <= A[1]+B[2] <= A[1]+B[3]...
A[2]+B[1] <= A[2]+B[2] <= A[2]+B[3]...
A[3]+B[1] <= A[3]+B[2] <= A[3]+B[3]...
...
對(duì)于每一個(gè)集合的每一個(gè)變量我們可以這樣表示,node.s = A[a] + A[b] ,node.b =b;
這樣這個(gè)集合的下一個(gè)變量就是 node.s - B[node.b] + B[node.b + 1] ,node.b ++;
這樣說應(yīng)該沒問題吧? 然后我們就開始多路歸并,很容易理解和想到,我們直接先把第一列圈放進(jìn)隊(duì)列,然后取出一個(gè)最為最小的那個(gè),然后把取出的這個(gè)的右側(cè)(下一個(gè))放進(jìn)隊(duì)列,然后在取,在放,每次都把取出的下一個(gè)放進(jìn)去,執(zhí)行n此之后就得到了這最小的n個(gè)數(shù),這是兩個(gè)序列的,這個(gè)題目是n個(gè)序列,這樣我們可以吧第一個(gè)和第二個(gè)合并,然后在用合并的序列去合并第三個(gè),最后把所有的序列都合并,這里說的合并就是前面上多那個(gè)多路歸并。
#include<queue>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 750 + 5
using namespace std;
typedef struct NODE
{
? ? int s ,b;
? ? friend bool operator < (NODE a ,NODE b)
? ? {
? ? ? ?return a.s > b.s;
? ? }
}NODE;
int num[N][N];
int A[N] ,B[N] ,C[N];
NODE xin ,tou;
void duoluguibing(int n)
{
? ? priority_queue<NODE>q; ?
? ? for(int i = 1 ;i <= n ;i ++)
? ? {
? ? ? ?xin.b = 1;
? ? ? ?xin.s = A[i] + B[1];
? ? ? ?q.push(xin);
? ? }
? ? for(int i = 1 ;i <= n ;i ++)
? ? {
? ? ? ? tou = q.top();
? ? ? ? q.pop();
? ? ? ? C[i] = tou.s;
? ? ? ? xin.b = tou.b + 1;
? ? ? ? xin.s = tou.s - B[tou.b] + B[xin.b];
? ? ? ? if(xin.b <= n) q.push(xin);
? ? }
? ? return;
}
int main ()
{
? ? int n ,i ,j;
? ? while(~scanf("%d" ,&n))
? ? {
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?{
? ? ? ? ? for(j = 1 ;j <= n ;j ++)
? ? ? ? ? scanf("%d" ,&num[i][j]);
? ? ? ? ? sort(num[i] + 1 ,num[i] + n + 1);
? ? ? ?}
? ? ? ?
? ? ? ?
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?C[i] = num[1][i];
? ? ? ?for(i = 2 ;i <= n ;i ++)
? ? ? ?{
? ? ? ? ? for(j = 1 ;j <= n ;j ++)
? ? ? ? ? A[j] = C[j] ,B[j] = num[i][j];
? ? ? ? ? duoluguibing(n);
? ? ? ?}
? ? ? ?
? ? ? ?for(i = 1 ;i <= n ;i ++)
? ? ? ?if(i == n) printf("%d\n" ,C[i]);
? ? ? ?else printf("%d " ,C[i]);
? ? }
? ? return 0;
}
? ? ? ?
總結(jié)
以上是生活随笔為你收集整理的UVA11997求前k个和,多路归并问题的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: UVA11991第k次出现的v的下标
- 下一篇: LA3135简单多路归并(优先队列)