牛客网练习赛7-B-购物

在遥远的东方,有一家糖果专卖店。
这家糖果店将会在每天出售一些糖果,它每天都会生产出m个糖果,第i天的第j个糖果价格为C[i][j]元。
现在的你想要在接下来的n天去糖果店进行选购,你每天可以买多个糖果,也可以选择不买糖果,但是最多买m个。(因为最多只生产m个)买来糖果以后,你可以选择吃掉糖果或者留着之后再吃。糖果不会过期,你需要保证这n天中每天你都能吃到至少一个糖果。
这家店的老板看你经常去光顾这家店,感到非常生气。(因为他不能好好睡觉了)于是他会额外的要求你支付点钱。具体来说,你在某一天购买了 k 个糖果,那么你在这一天需要额外支付 k2 的费用。

那么问题来了,你最少需要多少钱才能达成自己的目的呢?

 

输入描述:

第一行两个正整数n和m,分别表示天数以及糖果店每天生产的糖果数量。
接下来n行(第2行到第n+1行),每行m个正整数,第x+1行的第y个正整数表示第x天的第y个糖果的费用。

输出描述:

输出只有一个正整数,表示你需要支付的最小费用。

思路:首先这道题有个地方就是每天要加上k^2的额外费用,所以我们先处理一下这个,把每地方处理一下,就是每天价格排序,从小到大,每个糖果价格一次加上1,3,5...,因为k^2为首项1,差为2的等差数列前n项和,然后优先队列处理下就行了;
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<set>
#include<map>
#include<queue>
#include<vector>
#define ll long long int
#define mod 1000000007
#define me(a,b) memset(a,b,sizeof(a))
const int inf=0x7fffffff;
using namespace std;
/*struct node
{
    int n;
    int pos;
    friend bool operator<(node a,node b)
    {
        return a.n>b.n;
    }
}node,temp;*/
int main()
{
 
    int a[305][305];
    int n,m;
    int i,j;
    int cnt=1;
    int ans;
    priority_queue<int,vector<int>,greater<int> >q;
    ans=0;
    cin>>n>>m;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++)
            cin>>a[i][j];
    for(i=1;i<=n;i++)
        sort(a[i]+1,a[i]+1+m);
    for(i=1;i<=n;i++)
    {
        cnt=1;
        for(j=1;j<=m;j++)
        {
            a[i][j]+=cnt;
            cnt+=2;
        }
    }
    for(i=1;i<=m;i++)
        q.push(a[1][i]);
    ans=q.top();
    q.pop();
    for(i=2;i<=n;i++)
    {
        for(j=1;j<=m;j++)
        {
            q.push(a[i][j]);
        }
        ans+=q.top();
        q.pop();
    }
    cout<<ans<<endl;
    return 0;
}
原文地址:https://www.cnblogs.com/huangdao/p/8082158.html