Codeforces 447D - DZY Loves Modification

447D - DZY Loves Modification

思路：将行和列分开考虑。用优先队列，计算出行操作i次的幸福值r[i]，再计算出列操作i次的幸福值c[i]。然后将行取i次操作和列取k-i次操作，那么多加的幸福值就是i*(k-i)*p，因为无论先操作行还是列，每操作一次一个格子只减一次p。这样记录下最大的幸福值。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e6+;
const int M=1e3+;
const ll INF=1e18;
priority_queue<ll>row,col;
ll r[N],c[N];
int a[M][M];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie();
    int n,m,k,p;
    cin>>n>>m>>k>>p;
    for(int i=;i<n;i++)
    {
        ll sum=;
        for(int j=;j<m;j++)
        {
            cin>>a[i][j];
            sum+=a[i][j];
        }
        row.push(sum);
    }
    for(int j=;j<m;j++)
    {
        ll sum=;
        for(int i=;i<n;i++)
        sum+=a[i][j];
        col.push(sum);
    }
    for(int i=;i<=k;i++)
    {
        ll temp=row.top();
        row.pop();
        r[i]=r[i-]+temp;
        row.push(temp-m*p);
    }
    for(int i=;i<=k;i++)
    {
        ll temp=col.top();
        col.pop();
        c[i]=c[i-]+temp;
        col.push(temp-n*p);
    }
    ll ans=-INF;
    for(int i=;i<=k;i++)
    {
        ans=max(ans,r[i]+c[k-i]-1ll*i*(k-i)*p);
    }
    cout<<ans<<endl;
    return ;
}