Every day a Leetcode
题目来源:1535. 找出数组游戏的赢家
解法1:模拟
由于较小的元素会移至数组的末尾,在重新遇到被移至数组末尾的元素之前,arr 的每个元素都会被访问到。
考察游戏执行的流程,本质上是在从左到右遍历 arr,求数组最大值(打擂台)。我们要找首个连续 k 回合都是最大值的数。
如果遍历完 arr 也没找到这样的数,那么答案就是 arr 的最大值 mx,因为此时比 mx 小的数都会移到 mx 的右边,所以后面比大小都是 mx 胜利。
具体算法如下:
- 初始化 mx=arr[0],从 arr[1] 开始遍历数组。其中 win 用来统计 mx 连续多少个回合是最大值(获胜)。
- 如果 arr[i]>mx,更新 mx=arr[i] 以及 win=0。
- 把 win 加一,如果 win=k 就退出循环。
- 遍历结束(或者中途退出循环),返回 mx,也就是 arr[mx_i]。
代码:
/*
* @lc app=leetcode.cn id=1535 lang=cpp
*
* [1535] 找出数组游戏的赢家
*/
// @lc code=start
class Solution
{
public:
int getWinner(vector<int> &arr, int k)
{
int n = arr.size();
int mx_i = 0, win = 0;
for (int i = 1; i < n && win < k; i++)
{
if (arr[i] > arr[mx_i])
{
mx_i = i;
win = 0;
}
win++;
}
return arr[mx_i];
}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O(n),其中 n 是数组 arr 的长度。
空间复杂度:O(1)。