【打卡】数组的最长前缀
作者:互联网
描述
给定两个正整数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