ABC313C 扩展

简要题意：给定长为 $n$ 的序列，再给定 $k$，可以进行若干次以下操作：每次选定一个数 $i(1\le i\le n)$ 使得 $a_i\leftarrow a_i+k$ 或者 $a_i\leftarrow a_i-k$，最小化最终数组的最大值与最小值之差。

这个题是去年学校模拟赛的题目，这两天又给翻了出来（（

代码：

点击查看代码

#include<bits/stdc++.h>
using namespace std;
int n,m,ans=1e9,k,s;
int a[10005],v[10005];
struct ren{
	int id,v;
}d[20010];
bool cmp(ren a,ren b){
	return a.v<b.v;
}
int main(){
//	freopen("1.in","r",stdin);
//	freopen("1.out","w",stdout);
	scanf("%d",&n);
	for(int i=1; i<=n; i++) scanf("%d" ,&a[i]);
	scanf("%d",&k);
	for(int i=1;i<=n;i++){
		d[++m].id=i,d[m].v=a[i]-k;
		d[++m].id=i,d[m].v=a[i]+k;
	}
	sort(d+1,d+1+m,cmp);
//	for(int i=1;i<=m;i++) printf("%d %d\n",d[i].id,d[i].v);
	s=0;
	int i=0,j=1;
	while(i<m){
		if(s<n){
			i++;
			v[d[i].id]++;
			if(v[d[i].id]==1) s++;
		}
		else{
			while(1){
				ans=min(ans,d[i].v-d[j].v);
				v[d[j].id]--;
				if(v[d[j].id]==0){
					j++,s--;
					break;
				}
				j++;
			}
		}
	}
	printf("%d",ans);
	return 0;
}