【JZOJ4790】选数问题

题目描述
在给定的 N 个数中选出 R×C 个数，然后填入 R×C 的矩阵中，每一行的 D(i) 定义为本行最大值与最小值的差，然后要令所有行中 D(i) 的最大值 F 尽量小，其中1≤i≤R。请你求出满足条件的 F 。


输入格式
第一行是三个整数：N，R，C，其中, 1≤R,C≤104 ，R×C≤N≤5×105 。
第二行是 N 个整数 Pi ，0＜Pi≤109


输出格式
输出一个整数，即满足条件的最小的 F。


样例数据 1
输入　 [复制] 


7 2 3 
170 205 225 190 260 225 160
输出


30
备注
【样例说明】


输入样例可以选出以下数字构成2×3的矩阵


190 170 160  —> D(1) = 190 – 160 = 30
225 225 205  —> D(2) = 225 – 205 = 20
    

所以 D(i) 的最大值 F = max（D1,D2）= max（30,20）= 30 ，而且 F 是所有方案中最小的。

--------------------------------------------------------------------------------------------------------------------------------

二分答案是很好想到的——因为要求的答案有“最大的最小”这样的性质

首先先将n个数排序 然后明确：我们要求的这个答案 一定是来自一段连续的序列 

这运用了贪心的思想 具体证明见下（看不懂无所谓 只要能够大概知道为什么是连续序列就行）

假设对于某一行Ri，是排序之后不连续的C个数，那么必然存在一个数Pj min(Ri)≤Pj≤max(Ri)（其中，min(Ri)表示行Ri的最小值，max(Ri)表示行Ri的最大值）不属于Ri。假如Pj未被选入R×C个数中，那么把Ri替换为Pj，D(i)不会变大，只会变小，对其他行没有影响，从而F值不会变大。假如Pj被Rk选中，那么：

    ① 如果min(Rk)≤min(Ri)，那么交换Pj和min(Ri)不会增大D(k)，也不会增大D(i)，从而不会增大F值；


    ② 否则如果max(Rk)≥max(Ri)，那么交换Pj和max(Ri)不会增大D(k)，也不会增大D(i)，从而不会增大F值；

    ⑧ 否则如果min(Ri)≤min(Rk)，max(Rk)≤max(Ri))，假如交换max(Ri)和max(Rk)，那么改变之后D'(i)≤D(i)，D'(j)≤D(j)，所以max(D'(i),D'(j))≤D(i)，故F值不会增大。通过以上步骤可以把所有行的数字都变成连续顺序的。

所以对于当前二分出来的mid 如果发现了一个位置p:num[p+c-1]-num[p]<=ans 说明这一串数字可以填入矩阵 tot+1 p+=c-1 

如果大于ans 说明不能填入 那么p++ 即跳过一个数字

然后等检查完了整个序列后 若tot>=r 说明ans还有减小的潜力（ans减小，会有更多满足要求的数字串）

看代码吧

#include<bits/stdc++.h>
using namespace std;
int n,r,c,real_ans;
int num[500000]; 
bool check(int ans)
{
	int tot=0;
	for(int i=1;i+c-1<=n;)
	{
		if(num[i+c-1]-num[i]<=ans)
		{
			tot++;
			i+=c;
		}
		else i++;
	}
	if(tot>=r)	return true;//ans可以减小 
	else return false;
}
int main()
{
	cin>>n>>r>>c;
	for(int i=1;i<=n;i++)
	{
		cin>>num[i];
	}
	sort(num+1,num+n+1);
	int l=0,r=1000000000,mid;
	while(l<r)
	{
		mid=(l+r)/2;
		if(check(mid))
		{
			r=mid;
			real_ans=mid;
		}
		else
		{
			l=mid+1; 
		}
	}
	cout<<l<<" "<<endl;
	return 0;
}