力扣每日一题(难的我也不会)2038. 如果相邻两个颜色均相同则删除当前颜色(2022.3.22)
难度中等62收藏分享切换为英文接收动态反馈
总共有 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
。
首先想到的就是模拟,然后直接超时...,贴上代码
class Solution {
public boolean winnerOfGame(String colors) {
int n=colors.length();
if(n<3) return false;
boolean isAlice=true;
ArrayList<Character> list=new ArrayList<>();
for (int i = 0; i < n; i++) {
list.add(colors.charAt(i));
}
boolean A=false;
boolean B=false;
while (true){
n=list.size();
if(n<3) return !isAlice;
for (int i = 1; i < n-1; i++) {
if(isAlice){
if(list.get(i)=='A'&&list.get(i-1)=='A'&&list.get(i+1)=='A'){
list.remove(i);
A=true;
B=false;
break;
}
}else {
if(list.get(i)=='B'&&list.get(i-1)=='B'&&list.get(i+1)=='B'){
list.remove(i);
B=true;
A=false;
break;
}
}
}
if(isAlice&&!A) return false;
if(!isAlice&&!B) return true;
isAlice=!isAlice;
}
}
}
然后发现这个题可以采用贪心,将所有可以删除的A加起来,在将所有可以删除的B加起来,比较哪个大,贴上代码
class Solution {
public boolean winnerOfGame(String colors) {
int n=colors.length();
if(n<3) return false;
int a=0;
int b=0;
for (int i = 0; i < n-2; i++) {
String s=colors.substring(i,i+3);
if("AAA".equals(s)) a++;
if("BBB".equals(s)) b++;
}
return a>b;
}
}
然后这个其实可以优化,里面调用了太多的substring
优化版,采用滑动窗口
class Solution {
public boolean winnerOfGame(String colors) {
int n=colors.length();
if(n<3) return false;
int a=0;
int b=0;
for (int i = 1; i < n-1; i++) {
if('A'==colors.charAt(i-1)&&'A'==colors.charAt(i)&&'A'==colors.charAt(i+1)) a++;
if('B'==colors.charAt(i-1)&&'B'==colors.charAt(i)&&'B'==colors.charAt(i+1)) b++;
}
return a>b;
}
}
结束!!!