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;

    }
};

结果:

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页