DZY Loves Modification CodeForces 447D 贪心

最新推荐文章于 2021-02-13 22:32:19 发布

litmxs

最新推荐文章于 2021-02-13 22:32:19 发布

阅读量349

点赞数

文章标签： codeforces 贪心

本文链接：https://blog.csdn.net/litmxs/article/details/75675200

版权

-----贪心专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题目链接: DZY Loves Modification

题目大意

一个n*m的数字矩阵, 每次操作可以选取一行或一列, 让这一行(列)每个数字都减去p, 每次得到的收益是减去p前这一行(列)数字和, 求操作k次最大收益
$1 ≤ n, m ≤ 10^3; 1 ≤ k ≤ 10^6; 1 ≤ p ≤ 100$

思路

贪心
预处理出r[i](操作行i次后能得到的最大收益, 贪心, 利用优先队列每次取最大收益的, 减去后再放回优先队列), c[i](操作列i次能得到的最大收益)
因为操作i行之后, 每操作一任何列, 都会让这次操作(相对于没有操作行之前, 也就是c[i]里面存的值)收益减少p*i
所以枚举操作行的次数, 找到最大值

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e3;
typedef long long ll;
int n, m, k, p, a[maxn][maxn];
ll r[2000000], c[2000000];
int main()
{
    cin >> n >> m >> k >> p;
    for(int i=0; i<n; ++i)
        for(int j=0; j<m; ++j) scanf("%d", &a[i][j]);
    priority_queue<int> que;
    for(int i=0; i<n; ++i) 
    {
        int sum = 0;
        for(int j=0; j<m; ++j) sum += a[i][j];
        que.push(sum);
    }
    r[0] = 0;
    for(int i=1; i<=k; ++i) 
    {
        int c = que.top(); que.pop();
        r[i] = r[i-1] + c;
        c -= p*m;
        que.push(c);
    }
    while(!que.empty()) que.pop();
    for(int i=0; i<m; ++i)
    {
        int sum = 0;
        for(int j=0; j<n; ++j) sum += a[j][i];
        que.push(sum);
    }
    c[0] = 0;
    for(int i=1; i<=k; ++i)
    {
        int x = que.top(); que.pop();
        c[i] = c[i-1] + x;
        que.push(x-p*n);
    }
    // for(int i=0; i<=k; ++i) cout << r[i] << " "; cout << endl;
    // for(int i=0; i<=k; ++i) cout << c[i] << " "; cout << endl;
    ll ans = -1e18;
    for(int i=0; i<=k; ++i)
    {
        ans = max(ans, r[i]+c[k-i]-1ll*p*i*(k-i));
    }
    cout << ans << endl;

    return 0;
}