题目
总共有 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 。提示:
1 <= colors.length <= 10^5
colors 只包含字母 ‘A’ 和 ‘B’
思路
以为又是复杂的博弈问题,结果一想,两个人的操作不会相互造成影响,只要计算各自能走的步数就好了。不过这里怎么计算也有讲究。
自己写的时候是想通过计算连续子串的字符个数,然后就写的很不优雅,循环结束后还要判断一下。
代码
自己都看不下去的写法
class Solution {
public boolean winnerOfGame(String colors) {
int cntA = 0;
int cntB = 0;
int stepA = 0;
int stepB = 0;
int n = colors.length();
for (int i = 0; i < n; i++) {
if (colors.charAt(i) == 'A') {
cntA++;
// 遇到A了就更新一下B的步骤,但是当字符串最后是A时,循环结束还要再更新一下stepB,stepA同理
stepB += Math.max(0, cntB - 2);
cntB = 0;
} else {
cntB++;
stepA += Math.max(0, cntA - 2);
cntA = 0;
}
}
if (colors.charAt(n - 1) == 'A') {
stepA += Math.max(0, cntA - 2);
} else {
stepB += Math.max(0, cntB - 2);
}
return stepA > stepB;
}
}
优雅的写法
更好的写法,轮到谁就更新谁的,当
不小于
时,每遇到一个”己方”字母,
就加
class Solution {
public boolean winnerOfGame(String colors) {
int cntA = 0;
int cntB = 0;
int stepA = 0;
int stepB = 0;
int n = colors.length();
for (int i = 0; i < n; i++) {
if (colors.charAt(i) == 'A') {
cntA++;
cntB = 0;
if (cntA >= 3) {
stepA++;
}
} else {
cntB++;
cntA = 0;
if (cntB >= 3) {
stepB++;
}
}
}
return stepA > stepB;
}
}