【Week8CSP模拟II-A】HRZ 的序列 Gym - 270737C

算法

模拟

题目描述

给出一个数列a，询问是否能够将序列中的某些数加K，某些数减K，某些数不变，使得该序列运算后的数值全部相等。

解题思路

即判断该序列中不同的数值的个数；若仅有1或2种不同数值，则必定可以；若有3种不同数值，则需要判断是否三个数值中，最大值与中间值之差是否等于中间值与最小值之差；若大于3种，则必定不可能。

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int main(){
    int t;
    cin>>t;
    while(t--){
        long long a[3];
        int n,cnt=0;
        cin>>n;
        long long x;
        for(int i=1;i<=n;i++){
            cin>>x;
            bool have=0;
            for(int j=0;j<min(cnt,3);j++){
                if(a[j]==x){
                    have=1;
                    break;
                }
            }
            if(have)    continue;
            if(cnt<3){
                a[cnt]=x;
            }
            cnt++;
        }
        if(cnt>3)
            cout<<"NO"<<endl;
        else if(cnt<3)
            cout<<"YES"<<endl;
        else{
            long long amin=min(a[0],min(a[1],a[2]));
            long long amax=max(a[0],max(a[1],a[2]));
            long long amid;
            for(int i=0;i<3;i++)
                if(a[i]!=amin&&a[i]!=amax)
                    amid=a[i];
            if(amax-amid==amid-amin)
                cout<<"YES"<<endl;
            else
                cout<<"NO"<<endl;
        }
    }
    return 0;
}
    

题目总结

在写这种简单题目的时候一定要注意细节，以免不必要的失分；如是否已将所有变量开为long long类型、数组是否可能超界、多组数据时是否已经清空等。

题目原文

题目描述

相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列a，他对这个序列产生了浓厚的兴趣，他好奇是否存在一个数K，使得一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等，其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。由于瑞神只会刷B站，所以他把这个问题交给了你！

输入格式

输入第一行是一个正整数t表示数据组数。接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。

输出格式

输出共包t含行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。（输出不包含引号）

样例输入

2
5
1 2 3 4 5
5
1 2 3 4 5

样例输出

NO
NO