原题链接
Description
一天,树树买了N个容量可以认为是无限大的瓶子,初始时每个瓶子里有 1 升水。树树发现瓶子实在太多了,于是他决定 保留不超过 K 个瓶子。每次他选择两个当前含水量相同的瓶子合并,把一个瓶子的水全部倒进另一个瓶,然后把空瓶丢弃(不能丢弃有水的瓶子)。
显然在某些情况下树树无法达到目标比如N=3,K=1。此时树树会重新买一些新的瓶子(新瓶子容量无限,开始时有 1 升水),以达到目标。
现在树树想知道,最少需要买多少新瓶子才能达到目标呢?
Input
一行两个正整数N,K(1≤N≤109,K≤1000)。
Output
一个非负整数,表示最少需要买多少新瓶子。
Input
3 1
Output
1
Hint
对于30%的数据,N≤3∗105
对于100%的数据如题目。
分析:贪心问题,首先,只要是2的次方个瓶子都能合成一个,那么,接下来就要考虑如果2的次方分别合并以后,瓶子数量多于k,如何将多余的瓶子再次合并。
思路:首先将能凑齐2次方数的瓶子合成一个,再将剩下的再凑…
如果最终剩余的小于k,那就不需要新瓶子;如果大于k,那就需要新瓶子来和相对较小的凑:先将不用新瓶子的凑好的瓶子按水量从大到小排序,那,最后肯定是要和第k大水量的瓶子和并(因为水量越少,水量相同才能合并,那需要的新瓶子就越少),接着就把第k+1到最后一个瓶子里的水量加起来,看还差多少水量能和第k个瓶子水量持平,就需要多少个新瓶子。
Code:
#include<bits/stdc++.h>
using namespace std;
long long n,k,sum,a[100000],cnt,flag,summ;
int main(){
cin>>n>>k;
sum=n;
while(1) //将n分成尽可能少的几个部分,每个部分都是n的次方数
{
for(int i=1;;i++){
if(pow(2,i)>sum){
a[++cnt]=pow(2,i-1);
flag=i;
break;
}
}
if(flag!=1) sum-=a[cnt];
else break;
}
if(sum>0) a[cnt]=sum;
else cnt--;
if(cnt<=k){ //分块之后,总快数小于k,说明用不到新瓶子
cout<<0;
return 0;
}
for(int i=k+1;i<=cnt;i++) summ+=a[i]; //将第k个瓶子之后的所有瓶子的水量加起来,
//看还差多少和第k个瓶子持平
cout<<a[k]-summ; //差多少需要多少个新瓶子
return 0;
}