算法解析:如果相邻两个颜色均相同则删除当前颜色

135 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第1天,点击查看活动详情

问题

总共有 n 个颜色片段排成一列,每个颜色片段要么是 'A' 要么是 'B' 。给你一个长度为 n 的字符串 colors ,其中 colors[i] 表示第 i 个颜色片段的颜色。

A 和 B 在玩一个游戏,他们 轮流 从这个字符串中删除颜色。A 先手 。

如果一个颜色片段为 'A' 且 相邻两个颜色 都是颜色 'A' ,那么 A 可以删除该颜色片段。A 不可以 删除任何颜色 'B' 片段。 如果一个颜色片段为 'B' 且 相邻两个颜色 都是颜色 'B' ,那么 B 可以删除该颜色片段。B 不可以 删除任何颜色 'A' 片段。 A 和 B 不能 从字符串两端删除颜色片段。 如果其中一人无法继续操作,则该玩家 输 掉游戏且另一玩家 获胜 。 假设 A 和 B 都采用最优策略,如果 A 获胜,请返回 true,否则 B 获胜,返回 false。

输入:colors = "AAABBBA"
输出:false
解释:
AAABABB -> AABABB
A 先操作。
她删除从左数第二个 'A' ,这也是唯一一个相邻颜色片段都是 'A' 的 'A' 。

现在轮到 B 操作。
B 可以删除第五个 'B'

再轮到 A 操作:A 无法再执行操作了
所以 B 胜利,返回fal

解题思路

根据题意,当colors 有连续 N(N >= 3)个时,可以删除中间的数,也就是N-2个数。 因为只有当左右两个颜色一致时才会删除中间的颜色,所以A和B操作删除原字符串后都不会影响另一方。

所以我们思路是,可以通过一次遍历统计符合A和B各自的数量,又因为A先开始操作,所以最终结果就是对比 两个人谁的符合数量更多即获胜。

因为A先开始,所以数量一致时是B赢

代码

var winnerOfGame = function(colors) {
  const map = {
    'A': 0,
    'B': 0
  };
  for (let i = 1; i < colors.length-1; i++) {
    const isSame = colors[i] === colors[i-1] &&colors[i] === colors[i+1];
    if(isSame && colors[i] === 'A') {
      map['A']++
    } else if(isSame && colors[i] === 'B') {
      map['B']++
    }
  }
 // 数量一致时是 B 赢,因为 A 先开始
  return map['A'] > map['B'];
};