Codeforces ラウンド #FF (ディビジョン 2) D. DZY は変更が大好き + 貪欲 + 優先キュー_html/css_WEB-ITnose

リンク: http://codeforces.com/problemset/problem/447/D

質問: n*m 行列は k 回演算でき、各演算室は特定の行または列の数を減算します。両方から p を取得し、得られるスコアは、この行または列の元の数値の合計になります。 N 回の操作後に得られた最高スコアを見つけます。

アイデア: まず、各行と列の数値の合計を数えます。

実行される k 回の操作のうち、i 個の操作は行に対して実行され、残りの k-i 個の操作は列に対して実行されます。まず、各操作で列に対する行の影響を無視し、列を計算するときに、最終的に合計の影響が i*(k-i)*p であると計算できます。各 i 操作に対して選択された最高値を見つけて、cn[i]、rn[i] として記録される最高スコアを計算します (優先キューを使用して最初のものを取得します)。異なる i については、ans=max(cn[i]+rn[k-i]-i*(k-i)*p) をとります。

データは int を超えるため、ans の初期値は非常に小さい必要があることに注意してください。

コード:

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<map>#include<queue>#include<stack>#include<vector>#include<ctype.h>#include<cstdlib>#include<algorithm>#include<string>#define PI acos(-1.0)#define maxntypedef long long ll;using namespace std;long long INF =(1LL << 60);priority_queue < long long > c,r;int main(){    int n,m,k,p,x;    long long y;    scanf("%d%d%d%d",&n,&m,&k,&p);    int col[1005],row[1005];    long long cn[1000005],rn[1000005];    for(int i=0; i<n; i++)        for(int j=0; j<m; j++)        {            scanf("%d",&x);            col[j]+=x;            row[i]+=x;        }    for(int i=0; i<n; i++)        r.push(row[i]);    for(int i=0; i<m; i++)        c.push(col[i]);    cn[0]=rn[0]=0;    for(int i=1; i<=k; i++)    {        y=c.top();        cn[i]=cn[i-1]+y;        c.pop();        c.push(y-n*1LL*p);        y=r.top();        rn[i]=rn[i-1]+y;        r.pop();        r.push(y-m*1LL*p);    }    long long ans=-INF;    for(int i=0; i<=k; i++)        ans=max(ans,cn[i]+rn[k-i]-i*1LL*(k-i)*p);    printf("%I64d\n",ans);    return 0;}