AcWing 109. 天才ACM（倍增+归并）

来源：AcWing 109. 天才ACM（倍增+归并）

题目

给定一个整数 M，对于任意一个整数集合 S，定义“校验值”如下:

从集合 S中取出 M 对数(即 2∗M 个数，不能重复使用集合中的数，如果 S 中的整数不够 M 对，则取到不能取为止)，使得“每对数的差的平方”之和最大，这个最大值就称为集合 S的“校验值”。

现在给定一个长度为 N的数列 A 以及一个整数 T。

我们要把 A分成若干段，使得每一段的“校验值”都不超过 T。

求最少需要分成几段。

输入格式

第一行输入整数 K，代表有 K组测试数据。

对于每组测试数据，第一行包含三个整数 N,M,T。

第二行包含 N个整数，表示数列A1,A2…AN。

输出格式

对于每组测试数据，输出其答案，每个答案占一行。

数据范围

1≤K≤12,
1≤N,M≤500000,
0≤T≤10¹⁸,
0≤Ai≤2²⁰

输入样例

2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9

输出样例

2
1

题解

思路

校检值必然为排序后的集合的 最大值和最小值的差的平方+次大值和次小值的差的平方…

每一段尽量包含更多的元素。

考虑用倍增对每一段进行枚举，再把这一段排序，看看是否符合题意。

由于每一次枚举时右端点左侧的序列已经是排好序的，所以只用对新加的一段排序，然后把两端归并即可。

O(nlogn)

另外要注意sort的边界范围问题，左端点=a+你的起始位置，右端点=a+你的结束位置，若从1开始则右端点=a+你的结束位置+1。第一次交时就被这个坑了(为什么样例能过啊)。

代码

#include<bits/stdc++.h>
using namespace std;
long long k,n,m,t,a[500005],b[500005],c[500005];
bool check(long long l,long long r,long long p)
{
	long long sum=0;
	for(int i=r+1;i<=r+p;i++) b[i]=a[i];
	sort(b+r+1,b+r+p+1);
	long long x1=l,x2=r+1,x3=1;
	while(x1<=r&&x2<=r+p)
	{
		if(b[x1]>b[x2]) c[x3++]=b[x2++];
		else c[x3++]=b[x1++];
	}
	while(x1<=r) c[x3++]=b[x1++];
	while(x2<=r+p) c[x3++]=b[x2++];
	for(int i=1;i<=min(m,(x3-1)/2);i++) sum+=(c[x3-i]-c[i])*(c[x3-i]-c[i]);
	if(sum<=t)
	{
		for(int i=1;i<x3;i++) b[l+i-1]=c[i];
		return true;
	}
	return false;
}
inline long long mr()
{
	long long x=0;char ch=getchar();
	while(ch<'0'||ch>'9') ch=getchar();
	while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
	return x;
}
int main()
{
	k=mr();
	while(k--)
	{
		n=mr();
		m=mr();
		t=mr();
		for(int i=1;i<=n;i++) a[i]=mr();
		long long l=1,r=1,p=1,sum=0;
		while(r<=n)
		{
			b[l]=a[l];
			sum++;
			while(p>0)
			{
				if(r+p<=n&&check(l,r,p))
				{
					r+=p;
					p*=2;
				}
				else p/=2;
			}
			p=1;
			r++;
			l=r;
		}
		printf("%lld\n",sum);
	}
	return 0;
}