码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces Round #FF/#255 D DZY Loves Modification --贪心+优先队列

时间:2014-07-14 10:26:55      阅读:226      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   os   art   

题意:给你一个矩阵,每次选某一行或者某一列,得到的价值为那一行或列的和,然后该行每个元素减去p。问连续取k次能得到的最大总价值为多少。

解法:

如果p=0,即永远不减数,那么最优肯定是取每行或每列那个最大的取k次,所以最优解由此推出。

如果不管p,先拿,最后再减去那些行列交叉点,因为每个点的值只能取一次,而交叉点的值被加了两次,所以要减掉1次,如果取行A次,取列B次,那么最后答案为:

res = dp1[A] + dp2[B] - B*(k-A)*p,可以细细体会一下后面那部分。

其中:

dp1[A]表示在行中取A次得到的最大和(取完要减去p哦)

dp2[B]表示在列中取B次得到的最大和(~)

每次取最大的可以用优先队列来实现,思想就是这么个思想。

代码:

bubuko.com,布布扣
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#define lll __int64
using namespace std;
#define N 1007
#define M 33

lll dp1[N*N],dp2[N*N];
lll RS[N],CS[N];
lll a[N][N];
priority_queue<lll> que;

int main()
{
    int n,m,k,p;
    int i,j;
    while(scanf("%d%d%d%d",&n,&m,&k,&p)!=EOF)
    {
        memset(RS,0,sizeof(RS));
        memset(CS,0,sizeof(CS));
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                scanf("%I64d",&a[i][j]);
                RS[i] += a[i][j];
                CS[j] += a[i][j];
            }
        }
        dp1[0] = dp2[0] = 0;
        while(!que.empty())
            que.pop();
        for(i=1;i<=n;i++)
            que.push(RS[i]);
        for(i=1;i<=k;i++)
        {
            lll tmp = que.top();
            que.pop();
            dp1[i] = dp1[i-1]+tmp;
            tmp -= (lll)m*p;
            que.push(tmp);
        }
        while(!que.empty())
            que.pop();
        for(i=1;i<=m;i++)
            que.push(CS[i]);
        for(i=1;i<=k;i++)
        {
            lll tmp = que.top();
            que.pop();
            dp2[i] = dp2[i-1]+tmp;
            tmp -= (lll)n*p;
            que.push(tmp);
        }
        ll res = dp1[0]+dp2[k];
        for(i=1;i<=k;i++)
        {
            ll tmp = dp1[i]+dp2[k-i]-1LL*i*(k-i)*p;
            res = max(res,tmp);
        }
        printf("%I64d\n",res);
    }
    return 0;
}
View Code

 

Codeforces Round #FF/#255 D DZY Loves Modification --贪心+优先队列,布布扣,bubuko.com

Codeforces Round #FF/#255 D DZY Loves Modification --贪心+优先队列

标签:style   blog   http   color   os   art   

原文地址:http://www.cnblogs.com/whatbeg/p/3841984.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!