leetcode1535. 找出数组游戏的赢家（Python3、c++）

leetcode1535. 找出数组游戏的赢家

给你一个由 不同 整数组成的整数数组 arr 和一个整数 k 。

每回合游戏都在数组的前两个元素（即 arr[0] 和 arr[1] ）之间进行。比较 arr[0] 与 arr[1] 的大小，较大的整数将会取得这一回合的胜利并保留在位置 0 ，较小的整数移至数组的末尾。当一个整数赢得 k 个连续回合时，游戏结束，该整数就是比赛的 赢家 。

返回赢得比赛的整数。

题目数据 保证 游戏存在赢家。

示例 1：

输入：arr = [2,1,3,5,4,6,7], k = 2
输出：5
解释：一起看一下本场游戏每回合的情况：

因此将进行 4 回合比赛，其中 5 是赢家，因为它连胜 2 回合。

示例 2：

输入：arr = [3,2,1], k = 10
输出：3
解释：3 将会在前 10 个回合中连续获胜。

示例 3：

输入：arr = [1,9,8,2,3,7,6,4,5], k = 7
输出：9

示例 4：

输入：arr = [1,11,22,33,44,55,66,77,88,99], k = 1000000000
输出：99

提示：

• 2 <= arr.length <= 10^5
• 1 <= arr[i] <= 10^6
• arr 所含的整数 各不相同 。
• 1 <= k <= 10^9

---

方法：模拟

思路：

这道题其实是比较容易的。

首先，我们思考，如果k大于数组的长度n，那么最后返回的答案一定是数组中的最大值。因为需要的获胜次数大于n，遍历一次数组进行比较是不够的，肯定需要继续比较（即把输的数字放到末尾），那么要想达到k次，说明获胜的数字为数组中的最大值（不然连获胜n次都达不到）。

所以，我们需要考虑的就是，当n<k的时候，还没有遍历到最大值，就满足了获胜次数等于k，这时候的答案。

使用res来保存答案，times保存获胜次数。首先res为arr[0]和arr[1]中较大的，即第一次的获胜者，times初始化为1；接下来开始遍历arr数组，

• 如果arr[i] > res，更新res，重置times=1；
• else：则times += 1，如果此时times = k，则返回res，即为答案。

如果遍历到最后，times还是小于k，那么说明此时的res即为数组的最大值，那么直接返回res即可。

• 只遍历了数组一次，时间复杂度为O(n)。
• 只是用了常数个变量，空间复杂度为O(1)。

代码：

Python3：

class Solution:
    def getWinner(self, arr: List[int], k: int) -> int:
        n = len(arr)
        # 首先初始为第一次比较的胜者，times为获胜的次数，初始为1
        res = max(arr[0],arr[1])
        times = 1
        # 如果k==1，那么直接返回即可
        if k == times:
            return res
        # 遍历剩余数组中的数
        for i in range(2,n):
            # 如果这个数大于目前的胜者，胜者更新，次数更新
            if arr[i] > res:
                res = arr[i]
                times = 1
            # res还是获胜者，那么次数更新
            else:
                times += 1
            # 次数==k，返回胜者
            if times == k:
                return res
        # 如果遍历结束还没有返回，那么说明目前的res为数组中的最大值，就是答案
        return res

cpp：

class Solution {
public:
    int getWinner(vector<int>& arr, int k) {
        int n = arr.size();
        // 首先初始为第一次比较的胜者，times为获胜的次数，初始为1
        auto res = max(arr[0],arr[1]);
        int times = 1;
        // 如果k==1，那么直接返回即可
        if (k == times)
            return res;
        // 遍历剩余数组中的数
        for (int i = 2; i < n; ++i){
            // 如果这个数大于目前的胜者，胜者更新，次数更新
            if (arr[i] > res){
                res = arr[i];
                times = 1;
            }    
            // res还是获胜者，那么次数更新
            else times += 1;
            // 次数==k，返回胜者
            if (times == k)
                return res;
        }    
        // 如果遍历结束还没有返回，那么说明目前的res为数组中的最大值，就是答案
        return res;

    }
};

结果：