首页 > 代码库 > Sequence(priority_queue)

Sequence(priority_queue)

这题很智慧。

VJ上4000多ms

#include<cstdio>#include<algorithm>#include<queue>#include <stdio.h>#include <vector>using namespace std;int main(){    int T,n,m,sum;    int a[2005];    scanf("%d",&T);    while(T--)    {        priority_queue<int ,vector<int>, greater<int> >q;        priority_queue<int ,vector<int>, less<int> >p;        scanf("%d%d",&n,&m);        for(int i=0; i<m; i++)        {            scanf("%d",&a[i]);            q.push(a[i]);        }        for(int i=1; i<n; i++)        {            for(int j=0; j<m; j++)                scanf("%d",&a[j]);            while(!q.empty())            {                sum=q.top();                q.pop();                for(int j=0; j<m; j++)                {                    if(p.size()==m&&p.top()>sum+a[j])                    {                        p.pop();                        p.push(sum+a[j]);                    }                    else if(p.size()<m)                    {                        p.push(sum+a[j]);                    }                }            }            while(!p.empty())            {                sum=p.top();                p.pop();                q.push(sum);            }        }        int flag=0;        while(!q.empty())        {            sum=q.top();            q.pop();            if(flag==1)            {                printf(" %d",sum);            }            else            {                printf("%d",sum);                flag=1;            }        }        printf("\n");    }    return 0;}

大神的代码(VJ上才刚900ms)

#include <iostream>#include<stdio.h>#include<math.h>#include<string.h>#include<algorithm>#include<queue>#include<set>#include<string>using namespace std;int main(){    priority_queue<int,vector<int>,less<int> >p;    int n,m,i,j,a[2011],b[2011],t,k;    scanf("%d",&t);    while(t--)    {        scanf("%d%d",&n,&m);        for(i=0; i<m; i++)            scanf("%d",&a[i]);        sort(a,a+m);        for(k=1; k<n; k++)        {            for(i=0; i<m; i++)            {                scanf("%d",&b[i]);                p.push(a[0]+b[i]);            }            sort(b,b+m);            for(i=1; i<m; i++)            {                for(j=0; j<m; j++)                {                    if(a[i]+b[j]>p.top())                        break;                    p.pop();                    p.push(a[i]+b[j]);                }            }            for(i=0; i<m; i++)            {                a[m-i-1]=p.top();                p.pop();            }        }        for(i=0; i<m-1; i++)            printf("%d ",a[i]);        printf("%d\n",a[m-1]);    }}