建塔 (高级二分,再多思考一下)

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn=1e5+7;
int n,k;
ll a[maxn],b[maxn],c[maxn];
//判断能否建k座塔 
bool check(int x){    
	int i,j;
	//建底层 
	for(int i=1;i<=x;i++)
		b[i]=a[i],c[i]=1;//b记录每座塔顶层的数据,c记录塔高
	//一层层地给每座塔增高 
	for(int i=x+1,j=1;i<=n;i++)
	//i是现在这个块,b是塔顶,要加上去就得跟塔顶比较 
		if(a[i]>=b[j]*2){
			b[j]=a[i];
			c[j]++;
			j++;
			if(j>x) j=1;
		}
	//x座里只要有一座塔高度不达标,就失败	 
	for(int i=1;i<=x;i++) 
		if(c[i]<k) return 0;
	return 1;
}
int main(){
	int t,l,r,mid,ans;
	cin>>t;
	while(t--){
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++)
			scanf("%lld",&a[i]);
		sort(a+1,a+1+n); 
		l=1;r=n;ans=0;//初始一个塔也没有 0 
		while(l<=r){
			mid=(l+r)/2;
			if(check(mid)) 
				ans=mid,l=mid+1;//求>=
			else r=mid-1;	
		}	
		printf("%d\n",ans);
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_50904510/article/details/119877550