相等序列--牛客网

题目给定a1,a2...an，这样一个长度为n的序列，现在你可以给其中一些元素加上一个值x（只能加一次），然后可以给另外一些值减上一个值x（只能减一次），剩下的元素不能再进行操作。问最后有没有可能找到一个值x使所有元素的值相等。

思路：把元素放入set中，有三种情况：

（1）长度小于3，直接输出Yes

（2）长度等于3，若开始值与末尾值的一半是中间值，则输出Yes,否则为No;

（3）否则为No.

程序：

#include <iostream>
#include <set>
#include <string>
using namespace std;
 
int main(){
    int k;  //k个序列
    cin >> k;
    while(k--){
        int n;  //序列长度
        int num; //元素值
        cin >> n;
        set<int> res;  //存储每一行的元素，但是每种类型只能存1个，并且会排序
        while(n--)
        {
            cin>>num;
            res.insert(num);
        }
        set<int>::iterator iter = res.begin();
        string ans;
        if(res.size() < 3){ 
            ans = "YES";
        }
        else if(res.size() == 3){
            //只能一增一减，又因为是有序的，所以第一个增，第三个减，第二个不变。
            //比如1 5 8 就是不可以的，1 5 9就是可以的，利用等差序列性质
            //注意这里不能用res.end,因为end是指向最后一个元素的下一个位置
            if((*res.begin() + *res.rbegin())/2 == *(++iter)){
                ans = "YES";
            }
            else{
                ans = "NO";
            }
        }
        else{
            ans = "NO";
        }
        cout << ans << endl;
    }
    return 0;
}