牛客等级之题N1(8.4场)购物
题目描述
在遥远的东方,有一家糖果专卖店。
这家糖果店将会在每天出售一些糖果,它每天都会生产出m个糖果,第i天的第j个糖果价格为C[i][j]元。
现在的你想要在接下来的n天去糖果店进行选购,你每天可以买多个糖果,也可以选择不买糖果,但是最多买m个。(因为最多只生产m个)买来糖果以后,你可以选择吃掉糖果或者留着之后再吃。糖果不会过期,你需要保证这n天中每天你都能吃到至少一个糖果。
这家店的老板看你经常去光顾这家店,感到非常生气。(因为他不能好好睡觉了)于是他会额外的要求你支付点钱。具体来说,你在某一天购买了 k 个糖果,那么你在这一天需要额外支付
k
2
k^2
k2 的费用。
那么问题来了,你最少需要多少钱才能达成自己的目的呢?
输入描述:
第一行两个正整数n和m,分别表示天数以及糖果店每天生产的糖果数量。
接下来n行(第2行到第n+1行),每行m个正整数,第x+1行的第y个正整数表示第x天的第y个糖果的费用。
输出描述:
输出只有一个正整数,表示你需要支付的最小费用。
示例1
输入
3 2
1 1
100 100
10000 10000
输出
107
示例2
输入
5 5
1 2 3 4 5
2 3 4 5 1
3 4 5 1 2
4 5 1 2 3
5 1 2 3 4
输出
10
看到不少人用 DP,这题其实可以用优先队列解决,我们首先要知道一个公式:
n
2
=
1
+
3
+
5
+
⋯
+
2
∗
n
−
1
n^2=1+3+5+\cdots+2*n-1
n2=1+3+5+⋯+2∗n−1
那么我们把每一天的商品先按价格排序,然后依次加
2
∗
i
−
1
2*i-1
2∗i−1 存入优先队列,每天去一个队头即可,AC代码如下:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[305],c[305],n,m,sum=0;
int main(){
priority_queue<int,vector<int>,greater<int> >q;
cin>>n>>m;
for(int i=1;i<=300;i++) c[i]=2*i-1;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++) cin>>a[j];
sort(a,a+m);
for(int j=0;j<m;j++) q.push(a[j]+c[j+1]);
sum+=q.top();
q.pop();
}
cout<<sum;
return 0;
}