给定一个整数 M,对于任意一个整数集合 S,定义“校验值”如下:
从集合 S 中取出 M 对数(即 2×M 个数,不能重复使用集合中的数,如果 S 中的整数不够 M 对,则取到不能取为止),使得“每对数的差的平方”之和最大,这个最大值就称为集合 S 的“校验值”。
现在给定一个长度为 N 的数列 A 以及一个整数 T。
我们要把 A 分成若干段,使得每一段的“校验值”都不超过 T。
求最少需要分成几段。
输入格式
第一行输入整数 K,代表有 K 组测试数据。
对于每组测试数据,第一行包含三个整数 N,M,T 。
第二行包含 NN 个整数,表示数列A1,A2…AN。
输出格式
对于每组测试数据,输出其答案,每个答案占一行。
数据范围
1≤K≤12
1≤N,M≤500000
0≤T≤1018
0≤Ai≤220
输入样例:
2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9
输出样例:
2
1
解题思路:
(倍增 ++ 归并) O(nlogn)
上述代码中,在处理 [start,end) 的时候,已经将 [start,end) 排好序了,所以不需要在处理 [start,end+len) 时再排序。
处理 [start,end+len)时,只需要将 [end,end+len)排序,然后将 [start,end) 与[end,end+len) 这两段区间进行归并即可。
详见代码注释。
时间复杂度
假设一共将数组划分成了 k 个区间(这里的区间指的是每次二分里面check的区间总和,并非题目中所指的区间),每个区间的长度分别为 len1,len2,⋯,lenk。
那么按上述方法只需要将每个区间排序一遍,所以时间复杂度为 O(len1loglen1+len2loglen2+⋯+lenkloglenk)≤O(nlogn)
加上每次归并的时间复杂度为 O(n),总的时间复杂度为 O(n+nlogn)=O(nlogn)
好了,啊,废话不多说,直接上废话:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=500005;
int n,m;
int ans;
ll T;
ll w[N],t[N];
ll tmp[N];
ll sq(ll x){
return x*x;
}
bool check(int l,int mid,int r){
for(int i=mid;i<r;i++)
t[i]=w[i];
sort(t+mid,t+r);
int i=l,j=mid,k=0;
while(i!=mid&&j!=r)
if(t[i]<t[j])
tmp[k++]=t[i++];
else
tmp[k++]=t[j++];
while(i!=mid)tmp[k++]=t[i++];
while(j!=r)tmp[k++]=t[j++];
ll sum=0;
for(i=0;i<m&&i<k;i++,k--)
sum+=sq(tmp[i]-tmp[k-1]);
return sum<=T;
}
int main(){
int K;
scanf("%d",&K);
while(K--){
scanf("%d%d%lld\n",&n,&m,&T);
for(int i=0;i<n;i++)
scanf("%lld",&w[i]);
ans=0;
int len;
int start=0,end=0;
while(end<n){
len=1;
while(len){
if(end+len<=n&&check(start,end,end+len)){
end+=len,len<<=1;
if(end>=n)break;
for(int i=start;i<end;i++)
t[i]=tmp[i-start];
}
else len>>=1;
}
start=end;
ans++;
}
printf("%d\n",ans);
}
}
本周内容就这些下期看啥,我也不知道诶嘿,玩原神吗?? 反正我玩诶嘿,再见!!
————玩原神的唐桑