csp模拟第二周A HRZ的序列_[csp模拟]造船-优快云博客

本文探讨了一种数据平衡问题的解决方案，通过寻找特定数值K，使得在一组数据中，部分数值加K，部分减K，其余保持不变的情况下，所有数值能够达到相等。文章提供了详细的解题思路和实现代码，采用排序和中间值策略，适用于数据科学和算法竞赛爱好者。

题意：

给定一组数据，如果有一个数K，在这组数中的某些数加上K，某些数减去K，某些数不变，可以这组数据全部相等，则输出YES，否则输出“NO”。

input:

第一个数据t(t<10)，表示有几组数据。随后每组数据两行，第一行为这组数据中的数据个数（<1e4，随后一行为这组数据(<1e15。

output:

如果有这样的数据则输出YES，没有输出“NO”。

思路：

考虑到数据大小，需要使用long long或者double，虽然题目给的说使整数。但使用double可以在之后少判断情乱（比如只有两个数，一个数不动，一个数加上K，需要特别判断，使用double，可以在这直接得到）。先将数据读入数组，然后排序，以最大值减去最小值除以2得到k（如果是用的long long红好像不能直接这么算k了，因为k貌似需要整数）和中间大小m=a[1]+k。然后从头开始遍历每一个数a[i]，如果a[i]!=m && a[i]-k!=m && a[i]+k!=m，则数组已经失败。如果全部成立，则输出K。

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int t;
int n;
double a[10010]; 
double  m, k; 

int main(){
	ios::sync_with_stdio(0);
	cin>>t;
	for(int i=0; i<t;i++){
		m=0,k=0;
		cin>>n;
		for(int j=1; j<=n; j++){
			cin>>a[j];
		} 
		sort(a+1,a+1+n);
		k=(a[n]-a[1])/2;
		m=a[1]+k;
		int ret=0;
		for(int j=1; j<=n; j++){
			if(a[j]!=m && (a[j]+k)!=m && (a[j]-k)!=m){
				ret=1;
				break;
			} 
		} 
		if(ret==0) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
		
	}
	return 0;
}