CF446B DZY Loves Modification 思维/优先队列By cellur925

Posted 2021-01-16 nopartyfoucaodong

题目传送门

题目大意：给一个 (n*m) 的矩阵，并进行 (k) 次操作，每次操作将矩阵的一行或一列的所有元素的值减 (p) ，得到的分数为这次修改之前这一列/一行的元素和，求分数最大值。

我开始的意识流想法是用一个优先队列维护，先把所有元素插入，然后(k)次每次取出堆顶，减去乘(p)的什么东西，再塞回队中。但是...行与列是会互相影响的鸭，那么怎么搞呢？然后就不会了hh。

正解是努力打破了行与列之间的相互影响，把行与列分开计算。我们考虑行与列是怎么互相影响的：选择(i)个行，(j)个列，那么这里(j=k-i)。显然会产生(i)*((k-i))个交点，我们需要另减去这些交点的贡献。

我们分别处理出行&列后，枚举最终的(i)就好了：）

#include<cstdio>
#include<algorithm>
#include<queue>

using namespace std;
typedef long long ll;

int n,m,k,p;
int f[2018][2018];
ll hang[2000],lie[2000],hang_cnt[1000900],lie_cnt[1000900];
ll ans=-1e18;
priority_queue<ll>q;

int main()
{
    scanf("%d%d%d%d",&n,&m,&k,&p);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&f[i][j]);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            lie[j]+=f[i][j],hang[i]+=f[i][j];
    for(int i=1;i<=m;i++) q.push(lie[i]);
    for(int i=1;i<=k;i++)
    {
        ll tmp=q.top();q.pop();
        lie_cnt[i]=lie_cnt[i-1]+tmp;
        tmp-=1ll*n*p;
        q.push(tmp);
    }
    while(!q.empty()) q.pop();
    for(int i=1;i<=n;i++) q.push(hang[i]);
    for(int i=1;i<=k;i++)
    {
        ll tmp=q.top();q.pop();
        hang_cnt[i]=hang_cnt[i-1]+tmp;
        tmp-=1ll*m*p;
        q.push(tmp);
    }
    for(int i=0;i<=k;i++)
        ans=max(ans,hang_cnt[i]+lie_cnt[k-i]-1ll*i*(k-i)*p);
    printf("%lld
",ans);
    return 0;
}

注意开始数组开小了==！(k)是(1e5)级别的。到现在还犯这种智障错误==。