相等序列

[编程题]相等序列

热度指数：5949 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

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

输入描述:

输入第一行为一个整数k，代表有k个序列(k<100)，接下来有2*k行:
偶数行为一个整数n，代表给定序列的长度(1<=n<=100,000)
奇数行包含n个元素，a₁,a₂...a_n，代表序列中的元素(0<=a_i<=100,000)

输出描述:

输出k行，每行一个YES或者NO

示例1

输入

1
5
1 3 3 2 1

输出

YES

算法知识视频讲解

Java

小谷围做题家

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n=in.nextInt();
        while(n>0){
            n--;
            int m=in.nextInt();
            Set<Integer> set=new HashSet();
            while(m>0){
                set.add(in.nextInt());
                m--;
            }
            if(set.size()>3){
                System.out.println("NO");
                continue;
            }else if(set.size()<=2){
                System.out.println("YES");
            }else{
                int[] arr=new int[3];
                int idx=0;
                for(int x:set){
                    arr[idx++]=x;
                }
                Arrays.sort(arr);
                if(arr[0]+arr[2]==arr[1]*2){
                    System.out.println("YES");
                }else{
                    System.out.println("NO");
                }
            }
        }
    }
}

发表于 2023-09-06 21:41:17 回复(0)

Java

你的offer对我打了烊

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int k = Integer.parseInt(br.readLine());
        for(int i = 0;i < k;i++){
            int n = Integer.parseInt(br.readLine());
            String[] str = br.readLine().split(" ");
            int[] arr = new int[n];
            for(int j = 0;j < n;j++){
                arr[j] = Integer.parseInt(str[j]);
            }
            System.out.println(eql(arr));
        }
    }
    public static String eql(int[] arr){
        Arrays.sort(arr);
        int a = arr[0];
        int b = arr[arr.length-1];
        if(arr.length == 2){
            return "YES";
        }
        if((a+b) % 2 != 0){
            return "NO";
        }
        int mid = (a+b)/2;
        int offset = b - mid;
        for(int i = 0;i < arr.length;i++){
            if(!(arr[i] + offset == mid || arr[i] - offset == mid || arr[i] == mid)){
                return "NO";
            }
        }
        return "YES";
    }
}

很黄很暴力

发表于 2020-04-19 21:33:01 回复(0)