1823. 数组的最长前缀

1823. 数组的最长前缀

中文English

给定两个正整数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]

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

class Solution:
    """
    @param X: a integer
    @param Y: a integer
    @param nums: a list of integer
    @return: return the maximum index of largest prefix
    """
    '''
    大致思路:
    1.初始化p,和c_X和c_Y，如果两者相等，则更新p=i，并且c_X不能为0的前提，最后如果p为0，则返回-1，否则返回p。
    '''
    def LongestPrefix(self, X, Y, nums):
        p = 0
        c_X,c_Y = 0,0
        for i in range(len(nums)):
            if nums[i] == X:
                c_X += 1
            elif nums[i] == Y:
                c_Y += 1
            #不为0才进来
            if c_X == c_Y and c_X != 0:
                p = i
        return -1 if p == 0 else p