题目:
给定一个整数 M,对于任意一个整数集合 S,定义“校验值”如下:
从集合 S 中取出 M 对数(即 2∗M 个数,不能重复使用集合中的数,如果 S 中的整 数不够 M 对,则取到不能取为止),使得“每对数的差的平方”之和最大,这个最大值 就称为集合 S 的“校验值”。
现在给定一个长度为 N 的数列 A 以及一个整数 T。我们要把 A 分成若干段,使得 每一段的“校验值”都不超过 T。求最少需要分成几段。
input
2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9
output
2
1
解题思路:
枚举每一个分块[L,R]的L,在L一定的情况下根据倍增思路不断更新R值,直到不能更新为止,确定一个分块。
在求分块的“校验值”时,若对整个分块进行sort,整体时间复杂度为 O(Nlog2N),需要对此进行优化。
当前分块左边界为L,已确定的满足条件的右边界为R,下一个需要检验的右边界为R+p。[L,R]
已有序,在对[R+1,R+p]
sort后,根据归并排序的思路将这两个区间合并为有序区间[L,R+p]
,再进行检验。据说整体时间复杂度可降到 O(NlogN)。(倍增的过程最多循环 O(logN)次,累计扩展长度为 O(N),好像差不多是这样)
a[]
纪录有序区段,[L,R]
(有序) 和[R+1,R+p]
(无序)
b[]
只截取a[]
中的[R+1,R+p]
,并进行sort
c[]
是归并排序时的临时数组,记录有序的[L,R+p]
若[L,R+p]
可行,更新a[]
中[L,R+p]
的元素排列(即c[]
中记录的内容),R+=p,p*=2;
否则p/=2,a[]
不做任何更新;
这样的话用到了3个数组。
还要加快读,否则只过90%
ac代码:
打扰了,同样的代码偶尔会超时。。👋
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e5+10;
ll n, m, t;
ll k;
ll a[maxn], b[maxn], c[maxn];
inline ll read() {
char ch = getchar(); ll x = 0;
while(ch<'0' || ch>'9') {
ch = getchar();
}
while(ch>='0' && ch<='9') {
x = x * 10 + ch - '0';
ch = getchar();
}
return x;
}
bool check(ll l, ll r)
{
if(l==r) return false;
ll sum = 0;
ll i = l, j = r;
for(int k = 1; k <= min((r-l+1)/2, m); k++, i++, j--)
sum += 1ll*(c[i]-c[j])*(c[i]-c[j]);
return sum <= k;
}
void Sort(ll L, ll R, ll p)
{
for(ll i = R+1; i <= R+p; i++) b[i] = a[i];//截取部分
sort(b+R+1, b+R+p+1);//只对新增的部分排序[) [L,R] [R+1, R+p]
ll i = L, j = R+1;
for(ll k = L; k <= R+p; k++)
{
if(j>R+p || (i<=R&&a[i]<=b[j])) c[k] = a[i++];
else c[k] = b[j++];
}
}
int main()
{
//freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
t = read();
while(t--)
{
n = read(); m = read(); k = read();
for(int i = 1; i <= n; i++) a[i] = read();
ll L = 1, R = 1, p = 1;
int ans = 0;
while(R<n)
{
while(p!=0)
{
Sort(L, R, p);
if(check(L, R+p))
{
for(ll i = L; i <= R+p; i++) a[i] = c[i];//接受排序
R+=p, p = min(p<<1, n-R);
}
else p>>=1;
}
ans++; R++;
if(R==n) ans++;
L = R; p = 1;
}
printf("%d\n", ans);
}
return 0;
}