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