DZY Loves Modification CodeForces - 447D

题意：有一个n * m的矩阵，你必须执行k次操作，每执行一次操作，你可以获得当前行（列）所有值的和，并且当前行（列）所有值减去p , 求执行k次所能获得的所有值 ，和的最大值；

思路：对于确定的对行执行i次，对列执行j次，可以确定所能获得的最大值，枚举i从0~k，求出max,减去 k*(k-i)p 就是最后答案。因为在对行和列单独进行贪心的适合，对答案的贡献少减了k(k-1)*p !

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e3+5;
ll sumr[maxn];
ll sumc[maxn];
ll dpsumr[maxn*maxn];
ll dpsumc[maxn*maxn];
priority_queue <ll> r,c;

int main(void)
{
    memset(sumr,0,sizeof(sumr));
    memset(sumc,0,sizeof(sumc));
    ll n,m,k,p;
    cin >> n >> m >> k >> p;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            ll x;
            scanf("%I64d",&x);
            sumr[i]+=x;
            sumc[j]+=x;
        }
    }
    /*for(int i=1;i<=m;i++)
        printf("%I64d\n",sumc[i]);*/
    for(int i=1;i<=n;i++)
        r.push(sumr[i]);
    for(int j=1;j<=m;j++)
        c.push(sumc[j]);
    /*while(!c.empty())
    {
        printf("c=%I64d\n",c.top());
        c.pop();
    }*/
    dpsumc[0]=dpsumr[0]=0;
    for(int i=1;i<=k;i++)
    {
        ll rr=r.top();r.pop();
        ll cc=c.top();c.pop();
        //printf("rr=%I64d  cc=%I64d\n",rr,cc);
        dpsumc[i]=dpsumc[i-1]+rr;
        dpsumr[i]=dpsumr[i-1]+cc;
        r.push(rr-p*m);
        c.push(cc-p*n);
    }
    ll maxx=-LONG_LONG_MIN;
    for(int i=0;i<=k;i++)
    {
        ll ans=dpsumr[i]+dpsumc[k-i]-1LL*i*(k-i)*p;
        maxx=max(maxx,ans);
    }
    cout << maxx << endl;
}