题意:
解法:
考虑对行操作一次的影响:
1.对其他行无影响.
2.每一列的总值减少了p.
发现p是常数,列操作总贡献和行的操作次数有关,
那么可以根据行的操作次数来判断列的总贡献减少了多少,
这样就可以消去先进行行操作,对之后列操作的影响了.
考虑枚举行操作i次,列操作k-i次能获得的分数最大值,
令ansr[i]表示行操作i次能获得的最大值,用堆存每行总值,贪心地操作即可,
同理,预处理出ansc[i],表示列操作i次能获得的最大值.
对行操作i次,列操作k-i次,此时重复操作的格子有i*(k-i)*p个,
枚举i,用ansr[i]+ansc[k-i]-i*(k-i)*p更新答案.
code:
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=2e3+5;
int a[maxm][maxm];
int ansr[1000000+5];
int ansc[1000000+5];
int r[maxm];
int c[maxm];
int n,m,k,p;
signed main(){
ios::sync_with_stdio(0);cin.tie(0);
cin>>n>>m>>k>>p;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>a[i][j];
r[i]+=a[i][j];
c[j]+=a[i][j];
}
}
priority_queue<int,vector<int>,less<int> >q,qq;
for(int i=1;i<=n;i++){
q.push(r[i]);
}
for(int i=1;i<=m;i++){
qq.push(c[i]);
}
for(int i=1;i<=k;i++){
int x=q.top();q.pop();
ansr[i]=ansr[i-1]+x;
x-=m*p;
q.push(x);
int y=qq.top();qq.pop();
ansc[i]=ansc[i-1]+y;
y-=n*p;
qq.push(y);
}
int ans=-1e18;
for(int i=0;i<=k;i++){
ans=max(ans,ansr[i]+ansc[k-i]-i*(k-i)*p);
}
cout<<ans<<endl;
return 0;
}