Beta Round #43 (ACM-ICPC Rules), problem: (E) Comb 优先队列+晦涩题意

题意:给定n个数列,每个数列的长度为m,在i数列中从头取连续的ci个数字,最后要满足c1>c2<c3....问这些数字最大和是多少。

做法:根据行数的不同,构造队列的方法要适当改变,可以优先队列还不是很会用,总是把cmp写错

#include <iostream>
#include <queue>
#include <vector>
#include <cstdio>
#define LL long long
#define eps -1e17
using namespace std;
const int LMT=1502;
LL dp[2][LMT],sum[LMT];
int n,m,tag;
struct cmp
{
    bool operator()(const int a,const int b)
    {
        return dp[tag^1][a]<dp[tag^1][b];
    }
};
priority_queue<int,vector<int>,cmp>que;
int main(void)
{
    LL ans=eps;
    tag=1;
    scanf("%d%d",&n,&m);
    for(int j=0;j<=m;j++)dp[tag][j]=eps;
    for(int i=1;i<=n;i++)
    {
        while(!que.empty())que.pop();
        for(int j=1;j<=m;j++)
        {
            scanf("%I64d",&sum[j]);
            sum[j]+=sum[j-1];
        }
        if(i&1)
        {
            for(int j=1;j<=m;j++)
            {
                if(dp[tag^1][j-1]!=eps)que.push(j-1);
                if(!que.empty())dp[tag][j]=dp[tag^1][que.top()]+sum[j];
            }
        }
        else
        {
            for(int j=2;j<=m;j++)que.push(j);
            for(int j=1;j<=m;j++)
            {
                while(!que.empty()&&que.top()<=j)que.pop();
                if(!que.empty())dp[tag][j]=dp[tag^1][que.top()]+sum[j];
            }
        }
        tag^=1;
        for(int j=0;j<LMT;j++)dp[tag][j]=eps;
    }
    for(int j=1;j<=m;j++)ans=max(ans,dp[tag^1][j]);
    printf("%I64d\n",ans);
    return 0;
}


你可能感兴趣的:(Beta Round #43 (ACM-ICPC Rules), problem: (E) Comb 优先队列+晦涩题意)