每日一题-如果相邻两个颜色均相同则删除当前颜色

发表于 更新于 分类于 Waline：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-colored-pieces-if-both-neighbors-are-the-same-color

题目

总共有 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。

思路

Alice 和 Bob能操作的次数分别记为A和B，如果Alice先开始只要A严格大于B即可获胜。假如连续‘A’长度为a则当$a>=3$时Alice可操作次数$A = A+a-2$。Bob同理。结尾添加一个无关字符减少特殊情况判断。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
class Solution {
	public:
	 bool winnerOfGame(string colors) {
		int a = 0;
		int b = 0;
		int A = 0;
		int B = 0;
		if (colors[0] == 'A') {
			++a;
		} else {
			++b;
		}
		colors.push_back('C');
		int n = colors.size();
		for (int i = 1 ; i < n ; ++i) {
			if (colors[i] != colors[i-1]) {
				if (colors[i-1] == 'B') {
					if (b >= 3) {
						B += b-2;
					}
					b = 0;
					a = 1;
				}
				if (colors[i-1] == 'A') {
					if (a >= 3) {
						A += a-2;
					}
					a = 0;
					b = 1;
				}
			} else if (colors[i] == 'A') {
				++a;
			} else {
				++b;
			}
		}
		return A > B;
	}
}
;