Leetcode 2038 如果相邻两个颜色均相同则删除当前颜色

2038. 如果相邻两个颜色均相同则删除当前颜色

题目描述

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

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

如果一个颜色片段为 'A' 且 相邻两个颜色 都是颜色 'A' ，那么 Alice 可以删除该颜色片段。Alice 不可以 删除任何颜色 'B' 片段。
如果一个颜色片段为 'B' 且 相邻两个颜色 都是颜色 'B' ，那么 Bob 可以删除该颜色片段。Bob 不可以 删除任何颜色 'A' 片段。
Alice 和 Bob 不能从字符串两端删除颜色片段。
如果其中一人无法继续操作，则该玩家输掉游戏且另一玩家获胜。

假设 Alice 和 Bob 都采用最优策略，如果 Alice 获胜，请返回 true，否则 Bob 获胜，返回 false。

实例1:

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

现在轮到 Bob 操作。
Bob 无法执行任何操作，因为没有相邻位置都是 'B' 的颜色片段 'B' 。
因此，Alice 获胜，返回 true 。

实例2：

输入：colors = "AA"
输出：false
解释：
Alice 先操作。
只有 2 个 'A' 且它们都在字符串的两端，所以她无法执行任何操作。
因此，Bob 获胜，返回 false 。

实例3：

输入：colors = "ABBBBBBBAAA"
输出：false
解释：
ABBBBBBBAAA -> ABBBBBBBAA
Alice 先操作。
她唯一的选择是删除从右数起第二个 'A' 。

ABBBBBBBAA -> ABBBBBBAA
接下来轮到 Bob 操作。
他有许多选择，他可以选择任何一个 'B' 删除。

然后轮到 Alice 操作，她无法删除任何片段。
所以 Bob 获胜，返回 false 。

方法：贪心

第一反应是模拟整个游戏的过程，代码如下：

class Solution {
    public boolean winnerOfGame(String colors) {
        while (true) {
            boolean flag = false;
            for (int i = 0; i < colors.length(); i++) {
                if (i + 2 < colors.length() && colors.charAt(i) == 'A'
                        && colors.charAt(i + 1) == 'A' && colors.charAt(i + 2) == 'A') {
                    flag = true;
                    colors = colors.substring(0, i + 1) + colors.substring(i + 2, colors.length());
                }
                if (flag) {
                    break;
                }
            }
            if (!flag) {
                return false;
            }
            flag = false;
            for (int i = 0; i < colors.length(); i++) {
                if (i + 2 < colors.length() && colors.charAt(i) == 'B'
                        && colors.charAt(i + 1) == 'B' && colors.charAt(i + 2) == 'B') {
                    flag = true;
                    colors = colors.substring(0, i + 1) + colors.substring(i + 2, colors.length());
                }
                if (flag) {
                    break;
                }
            }
            if (!flag) {
                return true;
            }
        }
    }
}

但是超时了，可恶 😢。

根据题目的要求，我们可以分别计算出 $\text{Alice}$ 和 $\text{Bob}$ 的操作数。当 $\text{Alice}$ 的操作数大于 $\text{Bob}$ 的操作数时， $\text{Alice}$ 获胜；否则， $\text{Bob}$ 获胜。

class Solution {
    public boolean winnerOfGame(String colors) {
        int[] ans = new int[]{0, 0};
        char curr = 'c';
        int num = 0;
        for (int i = 0; i < colors.length(); i++) {
            char c = colors.charAt(i);
            if (c != curr) {
                curr = c;
                num = 1;
            } else {
                ++num;
                if (num >= 3) {
                    ans[c - 'A']++;
                }
            }
        }
        return ans[0] > ans[1];
    }
}