WEEK 8 CSP-M2 A HRZ的序列

题目

样例

思路

思想十分简单，先判断输入的数据中有几种数字，如果大于三种则直接输出NO，小于三种则直接输出YES，等于三种则判断这三种数是不是等差的，如果是则输出YES，否则为NO。

反思

在课堂上做的时候，没有考虑到等差，直接等于三种就输出了，导致只拿了几十分（捂脸）。

注意

该题数据上限为10^15，应使用longlong类型，而不是int。

代码

#include <iostream>
#include <string.h>
#include <vector>
#include <algorithm>

using namespace std;

vector<long long> v,vt;
long long t, n, q, flag;

bool cmp1(long long a, long long b)
{
	return a < b;
}

int main()
{
	scanf("%d", &t);
	for (long long i = 1; i <= t; i++)
	{
		v.clear();
		vt.clear();
		flag = 0;
		scanf("%lld", &n);
		for (long long j = 1; j <= n; j++)
		{
			scanf("%lld", &q);
			v.push_back(q);
		}
		sort(v.begin(), v.end(), cmp1);
		for (long long j = 0; j < n-1; j++)
		{
			if (v[j] != v[j + 1])
			{
				vt.push_back(v[j]);
				flag++;				
			}
		}
		vt.push_back(v[n-1]);
		if (flag <= 1)
			printf("YES\n");
		else if(flag==2&&(vt[2]-vt[1])==(vt[1]-vt[0]))
			printf("YES\n");
		else
			printf("NO\n");
	}
    return 0; 
}