DZY Loves Modification CodeForces

题意：有一个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;}