题意:
给定一个整数 M,对于任意一个整数集合 S,定义“校验值”如下:
从集合 S 中取出 M 对数(即 2∗M 个数,不能重复使用集合中的数,如果 S 中的整 数不够 M 对,则取到不能取为止),使得“每对数的差的平方”之和最大,这个最大值就称为集合 S 的“校验值”。
现在给定一个长度为 N 的数列 A 以及一个整数 T。我们要把 A 分成若干段,使得 每一段的“校验值”都不超过 T。求最少需要分成几段。
输入格式:
第一行输入整数 K,代表有 K 组测试数据。
对于每组测试数据,第一行包含三个整数 N,M,T
第二行包含 N个整数,表示数列A1,A2…AN
输出格式:
对于每组测试数据,输出其答案,每个答案占一行。
输入样例:
2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9
输出样例:
2
1
算法思想:
题意就是求区间两数之差绝对值最大,整个数列分段越少越好,也就是区间包含的数越多越好。
倍增扩展区间,每个区间先排序,求出校验值,然后依次倍增扩展当前区间。
1、 要求每两个数的差的平方和最大,需要配对,最大配最小,次大配次小,依次类推,所以的排序。
2、开3个数组:
a[MaxN] :这个数组存原始数据,
b[MaxN] :这个数组,存放每一段已经排好序的数据;
c[MaxN] :归并排序用的临时数组;
3、 倍增:
寻找每一段满足差的平方和 sum <= k 的最长区间范围,L 和 R, 每次增加一段 p,一开始 p = 1, 如果区间 [L, R + p]满足条件,那么 p *= 2, 否则 p /= 2;
4 、快排和归并排序结合:
计算 区间[L, R] 差的平方和 sum的时候, 区间分为两段 [L, w] 和 [w + 1, R]
其中,区间 [L, w] 在数组 b中已经排好序了;
a) 用sort 对区间 [w + 1, R] 排序(在数组b上用sort)
b) 用归并对 两个有序区间 [L, w] 和 [w + 1, R]进行归并(区间归并的结果写到临时数组c)
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MaxN = 500010;
long long a[MaxN], b[MaxN], c[MaxN];
long long Test, n, m, k;
int w;
void merge(int L1, int R1, int R2) //合并区间[L1,R1][R1+1,R2]
{
int i = L1, j = R1 + 1;
int index = L1;
while(i <= R1 && j <= R2)
{
if(b[i] <= b[j])
{
c[index++] = b[i++];
}else{
c[index++] = b[j++];
}
}
while(i <= R1) c[index++] = b[i++];
while(j <= R2) c[index++] = b[j++];
}
long long calc_sum(int L, int R)
{
if(R > n)
{
R = n;
}
int t = m < (R - L + 1) >> 1? m : (R - L + 1) >> 1;
//如果区间大于m对,取m对,不够m对有多少取多少
for (int i = w + 1; i <= R; i++)
{
b[i] = a[i];
}
sort(b + w + 1, b + R + 1);
merge(L, w, R);
long long sum = 0;
for(int i = 0; i < t; ++i)
{
sum += (c[R - i] - c[L + i]) * (c[R - i] - c[L + i]);
}
return sum;
}
void solve()
{
int p = 1, L = 1, R = 1;
int cnt = 0;
b[1] = a[1];
w = 1;
while(L <= n)
{
p = 1;
while(p)
{
if(calc_sum(L, R + p) <= k)
{
w = R = R + p < n? R + p : n;//r+p超过n,w就是n
for(int i = L; i <= R; ++i)
{
b[i] = c[i];
}
p *= 2;
if(R == n)
{
break;
}
}else{
p /= 2;
}
}
++cnt;
L = R + 1;
}
printf("%d\n", cnt);
}
int main()
{
scanf("%lld", &Test);
while(Test--)
{
scanf("%lld%lld%lld", &n, &m, &k);
for(int i = 1; i <= n; ++i)
{
scanf("%lld", &a[i]);
}
solve();
}
return 0;
}
/*
2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9
*/
/*
2
1
*/