数组题目：找出数组游戏的赢家

题目

标题和出处

标题：找出数组游戏的赢家

出处：1535. 找出数组游戏的赢家

难度

5 级

题目描述

要求

给你一个由不同整数组成的整数数组 $\text{arr}$ 和一个整数 $\text{k}$。

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

返回赢得比赛的整数。

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

示例

示例 1：

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

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

示例 2：

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

示例 3：

输入：$\text{arr = [1,9,8,2,3,7,6,4,5], k = 7}$
输出：$\text{9}$

示例 4：

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

数据范围

• $\text{2}\le \text{arr.length}\le {\text{10}}^{\text{5}}$
• $1\le \text{arr[i]}\le {\text{10}}^{\text{6}}$
• $\text{arr}$ 所含的整数各不相同
• $\text{1}\le \text{k}\le {\text{10}}^{\text{9}}$

解法

思路和算法

由于数组 $\text{arr}$ 中的整数各不相同，因此每回合游戏一定能分出胜负。

首先考虑最简单的情形。当 $k=1$ 时，第一回合结束后，$\text{arr}[0]$ 和 $\text{arr}[1]$ 中的较大的整数取得胜利，并赢得 $k$ 个连续回合。因此当 $k=1$ 时，返回 $\text{arr}[0]$ 和 $\text{arr}[1]$ 中的较大的整数。

当 $k>1$ 时，常规思路是，在每回合游戏之后更新数组 $\text{arr}$ 的状态，该做法在每回合游戏都需要 $O(n)$ 的时间更新数组，其中 $n$ 是数组的长度，最坏情况下的时间复杂度高达 $O(n^2+nk)$，显然会超出时间限制。

其实，并不需要对数组进行更新。在第一回合游戏之后，无论 $\text{arr}[0]$ 和 $\text{arr}[1]$ 当中的哪个整数取得胜利，第二回合游戏的另一个整数一定是 $\text{arr}$ 中的下一个整数，即 $\text{arr}[2]$。当 $2\le i<n$ 时，第 $i$ 回合游戏一定在第 $i-1$ 回合游戏中取得胜利的整数和 $\text{arr}[i]$ 之间进行。

因此，需要记录上一回合游戏中取得胜利的整数和该整数取得连续胜利的回合数。用 $\text{prev}$ 表示上一回合游戏中取得胜利的整数，用 $\text{consecutive}$ 表示该整数取得连续胜利的回合数。

第一回合游戏在 $\text{arr}[0]$ 和 $\text{arr}[1]$ 之间进行，第一回合游戏之后，$\text{prev}$ 为 $\text{arr}[0]$ 和 $\text{arr}[1]$ 中的较大值，$\text{consecutive}$ 的值为 $1$。

当 $2\le i<n$ 时，令 $\text{curr}=\text{arr}[i]$，第 $i$ 回合游戏在 $\text{prev}$ 和 $\text{curr}$ 之间进行，可能有以下两种情况：

• 如果 $\text{prev}>\text{curr}$，则 $\text{prev}$ 不变，将 $\text{consecutive}$ 的值加 $1$，如果 $\text{consecutive}$ 的值更新之后等于 $k$，则 $\text{prev}$ 赢得 $k$ 个连续回合，成为游戏的赢家，将 $\text{prev}$ 返回即可；

• 如果 $\text{prev}<\text{curr}$，则 $\text{curr}$ 取得胜利，令 $\text{prev}=\text{curr}$，并将 $\text{consecutive}$ 的值更新为 $1$。

如果在经过 $n-1$ 回合游戏之后，仍然没有整数赢得 $k$ 个连续回合，则不需要继续模拟游戏过程，而是可以直接得到答案。

注意到游戏过程实质为打擂台的过程。无论进行多少回合，最后一次取得胜利的整数一定是已经参与过游戏的整数中的最大值。在经过 $n-1$ 回合游戏之后，数组 $\text{arr}$ 中的所有整数都已经参与过游戏，因此第 $n-1$ 回合游戏中取得胜利的整数为数组 $\text{arr}$ 中的最大值。用 $\text{maxNum}$ 表示数组 $\text{arr}$ 中的最大值，则之后的每一回合游戏都在 $\text{maxNum}$ 和其他整数之间进行，$\text{maxNum}$ 将总是取得胜利，因此也将赢得 $k$ 个连续回合。

由此可以得到这道题的解法。

1. 从左到右遍历数组 $\text{arr}$ 的整数，模拟游戏过程，同时维护 $\text{arr}$ 的最大值 $\text{maxNum}$，如果存在一个整数赢得 $k$ 个连续回合，则将该整数返回。

2. 如果遍历结束数组 $\text{arr}$ 之后，没有出现一个整数赢得 $k$ 个连续回合，则将 $\text{maxNum}$ 返回。

代码

class Solution {
    public int getWinner(int[] arr, int k) {
        int prev = Math.max(arr[0], arr[1]);
        if (k == 1) {
            return prev;
        }
        int consecutive = 1;
        int maxNum = prev;
        int length = arr.length;
        for (int i = 2; i < length; i++) {
            int curr = arr[i];
            if (prev > curr) {
                consecutive++;
                if (consecutive == k) {
                    return prev;
                }
            } else {
                prev = curr;
                consecutive = 1;
            }
            maxNum = Math.max(maxNum, curr);
        }
        return maxNum;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{arr}$ 的长度。需要遍历数组 $\text{arr}$ 一次，对于每个整数，更新游戏状态和更新最大值的时间复杂度都是 $O(1)$，遍历结束之后返回最大值的时间复杂度也是 $O(1)$，因此总时间复杂度是 $O(n)$。

• 空间复杂度：$O(1)$。