POJ - 2442 Sequence(優先隊列/堆)

點我看題

大概今天狀态有點不對，有一點煩躁。

題意：給出一個m×n的矩陣，然後從矩陣的每一行跳出1個數可以形成n^m個序列，求出這些序列和中最小的n個。

分析：用優先隊列實作，用一個從大到小的臨時隊列來儲存前i行矩陣的最小和，然後不斷地更新。

參考代碼：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<iostream>

using namespace std;
#define inf 0x3f3f3f3f
const int maxn = 2e3+5;
const int maxm = 1e2+5;
int m,n;
int num[maxn];
//優先隊列預設從大到小
//定義從小到大
priority_queue<int,vector<int>,greater<int> > q;//存放結果
//定義從大到小
priority_queue<int,vector<int>,less<int> > tmp;//存放臨時結果

int main()
{
    int T;
    scanf("%d",&T);
    while( T--)
    {
        while( !q.empty())
            q.pop();
        while( !tmp.empty())
            tmp.pop();
        scanf("%d%d",&m,&n);
        for( int i = 1; i <= n; i++)
        {
            int x;
            scanf("%d",&x);
            q.push(x);
        }

        for( int i = 1; i < m; i++)
        {
            for( int j = 1; j <= n; j++)
            {
                scanf("%d",&num[j]);
                tmp.push(inf);//這個初始化很重要的
            }
            sort(num+1,num+1+n);
            while( !q.empty())
            {
                int tp = q.top();
                for( int j = 1; j <= n; j++)
                {
                    if( tp+num[j] < tmp.top())
                    {
                        tmp.pop();//保證tmp的大小恒為n
                        tmp.push(tp+num[j]);
                    }
                    else
                        break;
                }
                q.pop();
            }

            while( !tmp.empty())
            {
                q.push(tmp.top());
                tmp.pop();
            }

        }
        bool flag = 1;
        while( !q.empty())
        {
            if( flag)
            {
                printf("%d",q.top());
                flag = 0;
            }
            else
                printf(" %d",q.top());
            q.pop();
        }
        puts("");
    }

    return 0;
}