Talented Chef
Time Limit: 2 Seconds Memory Limit: 65536 KB
As we all know, Coach Gao is a talented chef, because he is able to cook M dishes in the same time. Tonight he is going to have a hearty dinner with his girlfriend at his home. Of course, Coach Gao is going to cook all dishes himself, in order to show off his genius cooking skill to his girlfriend.
To make full use of his genius in cooking, Coach Gao decides to prepare N N N dishes for the dinner. The i i i-th dish contains A i A_i Ai steps. The steps of a dish should be finished sequentially. In each minute of the cooking, Coach Gao can choose at most M M M different dishes and finish one step for each dish chosen.
Coach Gao wants to know the least time he needs to prepare the dinner.
Input
There are multiple test cases. The first line of input contains an integer
T
T
T indicating the number of test cases. For each test case:
The first line contains two integers
N
N
N and
M
M
M
(
1
<
=
N
,
M
<
=
40000
)
(1 <= N, M <= 40000)
(1<=N,M<=40000). The second line contains
N
N
N integers
A
i
A_i
Ai
(
1
<
=
A
i
<
=
40000
)
(1 <= Ai <= 40000)
(1<=Ai<=40000).
Output
For each test case, output the least time (in minute) to finish all dishes.
Sample Input
2
3 2
2 2 2
10 6
1 2 3 4 5 6 7 8 9 10
Sample Output
3
10
题意
有 n n n个菜,做每个菜需要 a i a_i ai步,每次最多可以做 m m m个菜,求做完这 n n n个菜最少需要多少时间
Solve
先提供一个超时的思路:将
n
n
n个菜按照
a
i
a_i
ai降序排序,排序后每次取出前
m
m
m个菜,然后让这
m
m
m个菜一直减到取出的最小的
a
i
a_i
ai小于未取出的
n
−
m
n-m
n−m个菜的最大的
a
i
a_i
ai,然后将减少后的不等于
0
0
0的
a
i
a_i
ai放入序列中,重复上述操作,直到
n
n
n个数全部为
0
0
0。
时间复杂度大概为
O
(
n
×
m
)
O(n\times m)
O(n×m)
不超时的算法:
将所有的
a
i
a_i
ai加起来,
s
u
m
=
∑
a
i
sum=\sum a_i
sum=∑ai,然后
s
u
m
sum
sum除以
m
m
m向上取整。
但是只有这样是不正确的:如果出现了
⌈
s
u
m
m
⌉
<
m
a
x
(
a
i
)
\lceil \dfrac {sum}{m}\rceil <max(a_i)
⌈msum⌉<max(ai)的情况,那么所需的时间一定不会小于
a
i
a_i
ai中的最大值,所以我们需要比较
⌈
s
u
m
m
⌉
\lceil \dfrac {sum}{m}\rceil
⌈msum⌉和
m
a
x
(
a
i
)
max(a_i)
max(ai)的值
证明:点这里吧,不太会证明。程序是照着超时的思路对拍出来的
Code
超时代码在最后
AC代码
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define ms(a,b) memset(a,b,sizeof(a))
const int inf=(1<<30);
const ll INF=(1LL*1<<60);
const int maxn=1e6+10;
const int mod=1e9+7;
const int maxm=1e3+10;
using namespace std;
int main(int argc, char const *argv[])
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
freopen("out1.txt", "w", stdout);
#endif
ios::sync_with_stdio(false);
cin.tie(0);
int t;
cin>>t;
while(t--)
{
int n,m;
ll x;
cin>>n>>m;
ll sum=0;
ll maxx=0;
for(int i=0;i<n;i++)
{
cin>>x;
maxx=max(maxx,x);
sum+=x;
}
ll _=(sum - 1)/(1LL*m)+1;
cout<<max(maxx,_)<<endl;
}
return 0;
}
超时代码
/*************************************************************************
> Author: WZY
> School: HPU
> Created Time: 2019-04-20 09:07:07
************************************************************************/
#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e4 + 10;
int a[maxn];
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
freopen("out2.txt", "w", stdout);
#endif
int t;
int x;
scanf("%d",&t);
while (t --)
{
int ans=0;
int n,m;
priority_queue<int,vector<int>,less<int> > que;
scanf("%d %d",&n,&m);
for (int i = 0;i < n;i ++)
{
scanf("%d",&x);
que.push(x);
}
while(!que.empty())
{
int cnt=0;
int minn;
for(int i=0;i<m;i++)
{
minn=que.top();
que.pop();
a[cnt++]=minn;
if(que.empty())
break;
}
int maxx;
if(que.empty())
maxx=0;
else
maxx=que.top();
int __=max(1,minn-maxx);
ans+=max(1,minn-maxx);
for(int i=0;i<cnt;i++)
{
a[i]=max(a[i]-__,0);
if(a[i])
que.push(a[i]);
}
}
printf("%d\n",ans);
}
return 0;
}