其他分享
首页 > 其他分享> > 【打卡】数组的最长前缀

【打卡】数组的最长前缀

作者:互联网

描述
给定两个正整数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 都出现且出现次数相同的情况
from typing import (
    List,
)

class Solution:
    """
    @param x: a integer
    @param y: a integer
    @param nums: a list of integer
    @return: return the maximum index of largest prefix
    """
    def longest_prefix(self, x: int, y: int, nums: List[int]) -> int:
        # write your code here
        l = len(nums)
        icount = 0
        jcount = 0
        result = -1
        if not nums:
            return -1
        for i in range(l):
            if nums[i] == x:
                icount += 1
            if nums[i] == y:
                jcount += 1
            if icount != 0 and icount == jcount:
                result = i
        return result

标签:index,return,前缀,nums,int,42,数组,icount,打卡
来源: https://blog.csdn.net/weixin_44708254/article/details/123129249