5分求调
  • 板块P10059 Choose
  • 楼主wo_hen_la
  • 当前回复0
  • 已保存回复0
  • 发布时间2024/3/24 21:05
  • 上次更新2024/3/25 09:11:30
查看原帖
5分求调
794701
wo_hen_la楼主2024/3/24 21:05

思路是先把最大的X求出来,再二分L,判断长度为L的区间 (满足极差>=最大的X) 的数量,是否有至少k个

#include <bits/stdc++.h>
#define int long long
#define N 100005
using namespace std;
int T,n,k,mxi;
int a[N],dp1[N][20],dp2[N][20];
void st()
{
	memset(dp1,-0x3f,sizeof(dp1)),memset(dp2,0x3f,sizeof(dp2));
	for(int i=1;i<=n;i++) dp1[i][0]=dp2[i][0]=a[i];
	for(int j=1;(1<<j)<=n;j++){
		for(int i=1;i+(1<<j)-1<=n;i++){
			dp1[i][j]=max(dp1[i][j-1],dp1[i+(1<<(j-1))][j-1]);
			dp2[i][j]=min(dp2[i][j-1],dp2[i+(1<<(j-1))][j-1]);
		}
	}
}
int query(int l,int r)
{
	int lg=trunc(log2(r-l+1));
	return max(dp1[l][lg],dp1[r-(1<<lg)+1][lg])-min(dp2[l][lg],dp2[r-(1<<lg)+1][lg]);
}
bool check(int x)
{
	int cnt=0;
	for(int i=1;i+x-1<=n;i++){
		if(query(i,i+x-1)>=mxi) cnt++;
		if(cnt==k) return 1;
	}
	return 0;
}
main()
{
	ios::sync_with_stdio(0);cin.tie(0),cout.tie(0);
	cin>>T;
	while(T--){
		cin>>n>>k;
		for(int i=1;i<=n;i++) cin>>a[i];
		st();
		mxi=0x3f;
		int len=n-k+1;
		for(int i=1;i+len-1<=n;i++) mxi=min(mxi,query(i,i+len-1));
		int l=1,r=n-k+1,ans;
		while(l<=r){
			int mid=l+r>>1;
			if(check(mid)) r=mid-1,ans=mid;
			else l=mid+1;
		}
		cout<<mxi<<" "<<ans<<"\n";
	}
	return 0;
}
2024/3/24 21:05
加载中...