C. Make It Equal
There is a toy building consisting of n towers. Each tower consists of several cubes standing on each other. The i-th tower consists of hi cubes, so it has height hi.
Let’s define operation slice on some height H as following: for each tower i, if its height is greater than H, then remove some top cubes to make tower’s height equal to H. Cost of one “slice” equals to the total number of removed cubes from all towers.
Let’s name slice as good one if its cost is lower or equal to k (k≥n).
Calculate the minimum number of good slices you have to do to make all towers have the same height. Of course, it is always possible to make it so.
Input
The first line contains two integers n and k (1≤n≤2⋅105, n≤k≤109) — the number of towers and the restriction on slices, respectively.
The second line contains n space separated integers h1,h2,…,hn (1≤hi≤2⋅105) — the initial heights of towers.
Output
Print one integer — the minimum number of good slices you have to do to make all towers have the same heigth.
Examples
inputCopy
5 5
3 1 2 2 4
outputCopy
2
inputCopy
4 5
2 3 4 5
outputCopy
2
Note
In the first example it’s optimal to make 2 slices. The first slice is on height 2 (its cost is 3), and the second one is on height 1 (its cost is 4).
这题的技巧在于使用前后缀计算已经覆盖的长度,避免超时
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,k,t[2*100000+5]={0},h[2*100000+5],sum[2*100000+5]={0};
long long ans=0,minn=0x3f3f3f,maxn=0,x;
int main()
{
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&x);
h[i]=x;
t[x]++;
minn=min(x,minn);
maxn=max(x,maxn);
}
sum[maxn]=t[maxn];
for(int i=maxn;i>=1;i--)
{
sum[i]=sum[i+1]+t[i];
}
int cost=0;
for(int i=maxn;i>minn;i--)
{
cost+=sum[i];
if(cost>k)
{
ans++;
cost=sum[i];
}
if(i==minn+1&&cost<=k)
ans++;
}
cout<<ans<<endl;
return 0;
}
加粗样式