题目描述
题目链接:https://leetcode-cn.com/problems/find-the-winner-of-an-array-game/
解题思路
我们先不考虑k
值的情况,以示例1arr = [2,1,3,5,4,6,7]
,画出每个回合列表
的变化情况如下:
第一回合:2
和1
比较,原始列表(绿色背景)中1
紧跟2
,2
胜
第二回合:原始列表中3
紧跟1
,3
和当前赢家2
比较, 3
胜,3
成为新的当前赢家
第三回合:原始列表中5
紧跟3
,5
和当前赢家3
比较, 5
胜,5
成为新的当前赢家
第四回合:原始列表中4
紧跟5
,4
和当前赢家5
比较,5
胜,保持
第五回合:原始列表中6
紧跟4
,6
和当前赢家5
比较, 6
胜,6
成为新的当前赢家
第六回合:原始列表中7
紧跟6
,7
和当前赢家6
比较, 7
胜,7
成为新的当前赢家
之后7
就一直是当前赢家了。,。。
…
规律:每次比较的值,索引都紧跟上一个,因此假设我们以arr[0]
为初始赢家,那么显然我们只要从arr[1:]
中按顺序进行遍历,如果碰到比当前赢家大的,那么它就替换当前赢家。(从上面也可以看出,在一次遍历结束之前,每个元素只会跟它后面的元素进行比较(在绿色背景原始列表中的位置),而跟每回合结束之后列表的具体位置似乎没有多大关系)。
此外我们发现,当最大值7
到达列表首部的时候(而最大值到达首部最多也只需要一次遍历),之后它就永远位于首部(因为没有人会比它大了),因此此时无论输入k
值为多少,最大值的连胜次数一定可以满足k
值。
因此,程序结束有两种情况,一是在一次遍历结束之前,某个数的连胜次数到达了k
值,那么我们就返回当前赢家cwinner
(如示例1中的5
),二是一次遍历结束,此时cwinner
一定是首部元素,而且它一定是最大值,从而无论多大k
值都能满足,如示例2
解答代码如下:
class Solution:
def getWinner(self, arr: List[int], k: int) -> int:
cwinner = arr[0] # 初始赢家
countn = 0 # 连胜次数
for x in arr[1:]:
if cwinner>x:
countn += 1 # 连胜次数+1
else:
cwinner = x # 碰到比它大的,进行替换
countn = 1 # 连胜终结
if countn >=k: # 达到k值,返回当前赢家
return cwinner
return cwinner # 一趟遍历结束之后,最大值到达首部,此时返回的一定是最大值