字节青训营 小C的排列询问

问题描述

小C拿到了一个排列，她想知道在这个排列中，元素 xx 和 yy 是否是相邻的。排列是一个长度为 nn 的数组，其中每个数字从 11 到 nn 恰好出现一次。

你的任务是判断在给定的排列中，xx 和 yy 是否是相邻的。

---

测试样例

样例1：

输入：n = 4, a = [1, 4, 2, 3], x = 2, y = 4
输出：True

样例2：

输入：n = 5, a = [3, 4, 5, 1, 2], x = 3, y = 2
输出：False

样例3：

输入：n = 6, a = [6, 1, 5, 2, 4, 3], x = 5, y = 2
输出：True

public class Main {
    public static boolean solution(int n, int[] a, int x, int y) {
    for (int i = 0; i < a.length; i++) {
        if(a[i]==x){
            if(i==0){
                if(a[i+1]==y){
                    return true;
                }
            }else if (i==n-1){
                if(a[i-1]==y){
                    return true;
                }
                
            }else{
                if(a[i-1]==y||a[i+1]==y){
                    return true;
                }
            }
        }
    }
        return false;
    }

    public static void main(String[] args) {
        System.out.println(solution(4, new int[]{1, 4, 2, 3}, 2, 4) == true);
        System.out.println(solution(5, new int[]{3, 4, 5, 1, 2}, 3, 2) == false);
        System.out.println(solution(6, new int[]{6, 1, 5, 2, 4, 3}, 5, 2) == true);
    }
}