题意 定义了一个集合的检验最大值也就是满足选出m对不同数的差平方和的最大值 不足m则取完 那我们知道只要排序 每次对应选最大的和最小的 次大的和次小的 第三大的 和第三小的 ......这样就可以了 暴力的做法就是n*n*logn *logn(二分
然后不用归并排序用递增 因为要暴力排序 暴力取值 所以复杂度是n*logn*logn
但是这题这样还过不了 所以要用递增加归并排序
以前没写过归并排序 对着某博客的写的//自己写了一个小时的版本竟然归并排序炸了 意思就是 你更新的时候先判断再决定更新不更新 然后延迟更新 不然会超时 思路不难 码的时候注意点
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAX_N = 5e5 +10;
inline ll read()
{
ll re=0;char ch=getchar();
while(ch<'0'||ch>'9') ch=getchar();
while(ch>='0'&&ch<='9') re=re*10+(ch^48),ch=getchar();
return re;
}
ll n,m,k,r,p,ans,l;
int a[MAX_N],b[MAX_N],c[MAX_N];
void Merge(int l,int r,int mid){
int i = l,j = mid;
for(int k = l;k<=r;++k)
if(j>r||(i<mid&&b[i]<b[j])) c[k] = b[i++];
else c[k] = b[j++];
}
void mysort(int al,int ar){
for(int i = r+1;i<=ar;++i) b[i] = a[i];
sort(b+1+r,b+1+ar);
Merge(al,ar,r+1);
}
ll cal(int al,int ar){
mysort(al,ar);
ll re = 0,cnt = 0;
for(int i = al,j = ar;i<j;i++,j--){
re+=(ll)(c[j]-c[i])*(c[j]-c[i]);cnt++;
if(cnt>=m) break;
}
return re;
}
int main(){
int T;scanf("%d",&T);
while(T--){
n = read();m = read();k = read();
for(int i=1;i<=n;++i) a[i] = read();
ans = 0,l = 1;
while(l<=n){
ans++;
r = l;p = 1;
b[l] = a[l];
while(p){
if(r+p<=n&&cal(l,r+p)<=k){
r+=p;
p*=2;
for(int i = l;i<=r;++i) b[i] = c[i];
}
else p/=2;
}
l = r + 1;
}
printf("%lld\n",ans);
}
return 0;
}