1823. 数组的最长前缀

1823. 数组的最长前缀

 

 

给定两个正整数X和Y，以及正整数数组nums。

我们需要找到一个最大的index，使得在nums[0], nums[1], .... , nums[index]中，出现X、Y的次数相等，且至少均出现一次，返回该index。

若不存在这样的index，则返回-1。

样例

示例 1:

输入:

X = 2

Y = 4

nums: [1, 2, 3, 4, 4, 3]

输出: 3

解释: 保证 2 和 4 出现相同次数的最长前缀是： {1, 2, 3, 4}，所以你应该返回3。

示例 2：

输入：

X = 7

Y = 42

nums = [7、42、5、6、42、8、7、5、3、6、7]

输出：9

解释：保证7和42出现相同次数的最长前缀是：{7, 42, 5, 6, 42, 8, 7, 5, 3, 6}，所以你应该返回9。

示例 3:

输入:

X = 1

Y = 10

nums: [2, 3, 1]

输出：-1

解释：不存在前缀使得 1 和 10 都出现且出现次数相同的情况

注意事项

• nums 的数组长度范围是: [0, 1000000]
• nums[i], X 以及 Y 的整数范围是: [1, 1 000000]

输入测试数据 (每行一个参数)如何理解测试数据？

 

 

public class Solution {

    /**

     * @param X: a integer

     * @param Y: a integer

     * @param nums: a list of integer

     * @return: return the maximum index of largest prefix

     */

    public int LongestPrefix(int X, int Y, int[] nums) {

            int result=0;

            int a=0,b=0;

            for (int i = 0; i < nums.length; i++) {

                if (nums[i]==X){

                    a++;

                }else if (nums[i]==Y){

                    b++;

                }

                if (a!=0&&b!=0&&a==b)result=i;

            }

            return result==0?-1:result;

    }

}