LeetCode打卡day22——如果相邻两个颜色均相同则删除当前颜色

471 阅读2分钟

“Offer 驾到,掘友接招!我正在参与2022春招系列活动-刷题打卡任务,点击查看活动详情。”

一、题目描述:

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 。
 

提示:

1 <= colors.length <= 105
colors 只包含字母 'A' 和 'B'

二、思路分析:

  1. 贪心 因为无法删除两端元素,故当给定的数组长度<=2时,直接返回Alice输。 接下来进行一般情况————当有n个相同元素连接时的判断:
  • n>2时,因为删除一个数例如AAAA,删除第二个A,对于第三个A还是视为左右两边相同,可以删除,故可删除n-2次
  • n<=2时,不能删除,删除次数为0 由上可知,Alice和Bob获胜条件的判断就转换为————Alice的删除次数大于Bob?大于获胜否则失败。因为Alice先手所以不能等于

按照这个思路我们用a,b存储对应的删除次数,cur取值为A或B,判断当前数组值是否等于cur,等于对应的字符出现次数count++,当不等于时,删除次数a或b进行相应的加法。

/**
 * @param {string} colors
 * @return {boolean}
 */
var winnerOfGame = function(colors) {
    let n=colors.length,counta=0,countb=0,cur="A";
    let a=0,b=0;
    if(n<3)return false
    for(let i=0;i<n;i++){
        if(colors[i]===cur){
            if(cur==='A')counta++
            else if(cur==='B')countb++
        }else{
            if(cur==='A'){
                if(counta>2)a+=counta-2
                counta=0
                countb=1
            }
            else {
                if(countb>2)b+=countb-2
                countb=0
                counta=1
            }
            cur=colors[i]
        }
    }
if(counta>2)a+=counta-2
if(countb>2)b+=countb-2
    return a>b 
};
  1. 优化 对于最初的设想,发现可以进行优化.我们可以只用一个count进行判断,这样我们就要区分count是哪一个,因为数组中只有A,B,我们设置cur='A',用当前的值对应的码数减即可,不用对cur频繁判断
var winnerOfGame = function(colors) {
    const freq = [0, 0];
    let cur = 'A';
    let cnt = 0;
    for (let i = 0; i < colors.length; i++) {
        const c = colors[i];
        if (c !== cur) {
            if (cnt >= 3) {
                freq[cur.charCodeAt() - 'A'.charCodeAt()] +=cnt-2 ;
            }
            cur = c;
            cnt = 1;
        } else {
            cnt += 1;
            
        }
    } 
        if (cnt >= 3) {
        freq[cur.charCodeAt() - 'A'.charCodeAt()] +=cnt-2 ;
    }           
    return freq[0] > freq[1];
};