Question

2660. 保龄球游戏的获胜者

English Version

题目描述

给你两个下标从 0 开始的整数数组 player1 和 player2 ，分别表示玩家 1 和玩家 2 击中的瓶数。

保龄球比赛由 n 轮组成，每轮的瓶数恰好为 10 。

假设玩家在第 i 轮中击中 xi 个瓶子。玩家第 i 轮的价值为：

• 如果玩家在该轮的前两轮的任何一轮中击中了 10 个瓶子，则为 2xi 。
• 否则，为 xi 。

玩家的得分是其 n 轮价值的总和。

返回

• 如果玩家 1 的得分高于玩家 2 的得分，则为 1 ；
• 如果玩家 2 的得分高于玩家 1 的得分，则为 2 ；
• 如果平局，则为 0 。

 

示例 1：

输入：player1 = [4,10,7,9], player2 = [6,5,2,3]
输出：1
解释：player1 的得分是 4 + 10 + 2*7 + 2*9 = 46 。
player2 的得分是 6 + 5 + 2 + 3 = 16 。
player1 的得分高于 player2 的得分，所以 play1 在比赛中获胜，答案为 1 。

示例 2：

输入：player1 = [3,5,7,6], player2 = [8,10,10,2]
输出：2
解释：player1 的得分是 3 + 5 + 7 + 6 = 21 。
player2 的得分是 8 + 10 + 2*10 + 2*2 = 42 。
player2 的得分高于 player1 的得分，所以 play2 在比赛中获胜，答案为 2 。

示例 3：

输入：player1 = [2,3], player2 = [4,1]
输出：0
解释：player1 的得分是 2 + 3 = 5 。
player2 的得分是 4 + 1 = 5 。
player1 的得分等于 player2 的得分，所以这一场比赛平局，答案为 0 。

 

提示：

• n == player1.length == player2.length
• 1 <= n <= 1000
• 0 <= player1[i], player2[i] <= 10

解法

方法一：模拟

我们可以用一个函数 $f(arr)$ 计算出两个玩家的得分，分别记为 $a$ 和 $b$，然后根据 $a$ 和 $b$ 的大小关系返回答案即可。

时间复杂度 $O(n)$，其中 $n$ 是数组的长度。空间复杂度 $O(1)$。

class Solution:
  def isWinner(self, player1: List[int], player2: List[int]) -> int:
    def f(arr: List[int]) -> int:
      s = 0
      for i, x in enumerate(arr):
        k = 2 if (i and arr[i - 1] == 10) or (i > 1 and arr[i - 2] == 10) else 1
        s += k * x
      return s

    a, b = f(player1), f(player2)
    return 1 if a > b else (2 if b > a else 0)

class Solution {
  public int isWinner(int[] player1, int[] player2) {
    int a = f(player1), b = f(player2);
    return a > b ? 1 : b > a ? 2 : 0;
  }

  private int f(int[] arr) {
    int s = 0;
    for (int i = 0; i < arr.length; ++i) {
      int k = (i > 0 && arr[i - 1] == 10) || (i > 1 && arr[i - 2] == 10) ? 2 : 1;
      s += k * arr[i];
    }
    return s;
  }
}

class Solution {
public:
  int isWinner(vector<int>& player1, vector<int>& player2) {
    auto f = [](vector<int>& arr) {
      int s = 0;
      for (int i = 0, n = arr.size(); i < n; ++i) {
        int k = (i && arr[i - 1] == 10) || (i > 1 && arr[i - 2] == 10) ? 2 : 1;
        s += k * arr[i];
      }
      return s;
    };
    int a = f(player1), b = f(player2);
    return a > b ? 1 : (b > a ? 2 : 0);
  }
};

func isWinner(player1 []int, player2 []int) int {
  f := func(arr []int) int {
    s := 0
    for i, x := range arr {
      k := 1
      if (i > 0 && arr[i-1] == 10) || (i > 1 && arr[i-2] == 10) {
        k = 2
      }
      s += k * x
    }
    return s
  }
  a, b := f(player1), f(player2)
  if a > b {
    return 1
  }
  if b > a {
    return 2
  }
  return 0
}

function isWinner(player1: number[], player2: number[]): number {
  const f = (arr: number[]): number => {
    let s = 0;
    for (let i = 0; i < arr.length; ++i) {
      s += arr[i];
      if ((i && arr[i - 1] === 10) || (i > 1 && arr[i - 2] === 10)) {
        s += arr[i];
      }
    }
    return s;
  };
  const a = f(player1);
  const b = f(player2);
  return a > b ? 1 : a < b ? 2 : 0;
}

impl Solution {
  pub fn is_winner(player1: Vec<i32>, player2: Vec<i32>) -> i32 {
    let f = |arr: &Vec<i32>| -> i32 {
      let mut s = 0;
      for i in 0..arr.len() {
        let mut k = 1;
        if (i > 0 && arr[i - 1] == 10) || (i > 1 && arr[i - 2] == 10) {
          k = 2;
        }
        s += k * arr[i];
      }
      s
    };

    let a = f(&player1);
    let b = f(&player2);
    if a > b {
      1
    } else if a < b {
      2
    } else {
      0
    }
  }
}