codeforces 446B（优先队列）

最新推荐文章于 2022-10-06 09:46:43 发布

原创最新推荐文章于 2022-10-06 09:46:43 发布 · 561 阅读

0 ·

CC 4.0 BY-SA版权

矩阵同时被 2 个专栏收录

3 篇文章

订阅专栏

优先队列

2 篇文章

订阅专栏

本文提供了一道CodeForces B级题目（编号446/B）的详细解答过程，该题涉及矩阵操作、优先队列及最优化算法。通过使用C++实现，文章展示了如何读取矩阵数据并计算每行每列的总和，再利用优先队列来找出最优的k个元素。

题目链接：http://codeforces.com/problemset/problem/446/B

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e3+3;
int matrix[N][N];
ll row[N],col[N];
ll dpr[N*N],dpc[N*N];
priority_queue <ll> q;
int main()
{
    ll n,m,k,p;
    scanf("%I64d%I64d%I64d%I64d",&n,&m,&k,&p);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&matrix[i][j]);
            row[i]+=matrix[i][j];
            col[j]+=matrix[i][j];
        }
    }
    for(int i=1;i<=n;i++)
        q.push(row[i]);
    for(int i=1;i<=k;i++)
    {
        ll temp=q.top();
        q.pop();
        dpr[i]=dpr[i-1]+temp;
        temp-=m*p;
        q.push(temp);
    }
    while(!q.empty()) q.pop();
    for(int i=1;i<=m;i++)
        q.push(col[i]);
    for(int i=1;i<=k;i++)
    {
        ll temp=q.top();
        q.pop();
        dpc[i]=dpc[i-1]+temp;
        temp-=n*p;
        q.push(temp);
    }
    ll ans=dpr[0]+dpc[k];
    for(ll i=1;i<=k;i++)
        ans=max(ans,dpr[i]+dpc[k-i]-i*(k-i)*p);
    printf("%I64d\n",ans);
    return 0;
}