Codeforces Round #FF/#255 D DZY Loves Modification --贪心+优先队列

题意：给你一个矩阵，每次选某一行或者某一列，得到的价值为那一行或列的和，然后该行每个元素减去p。问连续取k次能得到的最大总价值为多少。

解法：

如果p=0，即永远不减数，那么最优肯定是取每行或每列那个最大的取k次，所以最优解由此推出。

如果不管p，先拿，最后再减去那些行列交叉点，因为每个点的值只能取一次，而交叉点的值被加了两次，所以要减掉1次，如果取行A次，取列B次，那么最后答案为：

res = dp1[A] + dp2[B] - B*(k-A)*p，可以细细体会一下后面那部分。

其中：

dp1[A]表示在行中取A次得到的最大和（取完要减去p哦）

dp2[B]表示在列中取B次得到的最大和（~）

每次取最大的可以用优先队列来实现，思想就是这么个思想。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#define lll __int64
using namespace std;
#define N 1007
#define M 33

lll dp1[N*N],dp2[N*N];
lll RS[N],CS[N];
lll a[N][N];
priority_queue<lll> que;

int main()
{
    int n,m,k,p;
    int i,j;
    while(scanf("%d%d%d%d",&n,&m,&k,&p)!=EOF)
    {
        memset(RS,0,sizeof(RS));
        memset(CS,0,sizeof(CS));
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                scanf("%I64d",&a[i][j]);
                RS[i] += a[i][j];
                CS[j] += a[i][j];
            }
        }
        dp1[0] = dp2[0] = 0;
        while(!que.empty())
            que.pop();
        for(i=1;i<=n;i++)
            que.push(RS[i]);
        for(i=1;i<=k;i++)
        {
            lll tmp = que.top();
            que.pop();
            dp1[i] = dp1[i-1]+tmp;
            tmp -= (lll)m*p;
            que.push(tmp);
        }
        while(!que.empty())
            que.pop();
        for(i=1;i<=m;i++)
            que.push(CS[i]);
        for(i=1;i<=k;i++)
        {
            lll tmp = que.top();
            que.pop();
            dp2[i] = dp2[i-1]+tmp;
            tmp -= (lll)n*p;
            que.push(tmp);
        }
        ll res = dp1[0]+dp2[k];
        for(i=1;i<=k;i++)
        {
            ll tmp = dp1[i]+dp2[k-i]-1LL*i*(k-i)*p;
            res = max(res,tmp);
        }
        printf("%I64d
",res);
    }
    return 0;
}