LeetCode 488.祖玛游戏

Posted 2022-04-01 programyang

回忆一下祖玛游戏。现在桌上有一串球，颜色有红色(R)，黄色(Y)，蓝色(B)，绿色(G)，还有白色(W)。 现在你手里也有几个球。

每一次，你可以从手里的球选一个，然后把这个球插入到一串球中的某个位置上（包括最左端，最右端）。接着，如果有出现三个或者三个以上颜色相同的球相连的话，就把它们移除掉。重复这一步骤直到桌上所有的球都被移除。

找到插入并可以移除掉桌上所有球所需的最少的球数。如果不能移除桌上所有的球，输出 -1 。

示例:
输入: "WRRBBW", "RB"
输出: -1
解释: WRRBBW -> WRR[R]BBW -> WBBW -> WBB[B]W -> WW （翻译者标注：手上球已经用完，桌上还剩两个球无法消除，返回-1）

输入: "WWRRBBWW", "WRBRW"
输出: 2
解释: WWRRBBWW -> WWRR[R]BBWW -> WWBBWW -> WWBB[B]WW -> WWWW -> empty

输入:"G", "GGGGG"
输出: 2
解释: G -> G[G] -> GG[G] -> empty

输入: "RBYYBBRRB", "YRBGB"
输出: 3
解释: RBYYBBRRB -> RBYY[Y]BBRRB -> RBBBRRB -> RRRB -> B -> B[B] -> BB[B] -> empty

标注:

    你可以假设桌上一开始的球中，不会有三个及三个以上颜色相同且连着的球。
    桌上的球不会超过20个，输入的数据中代表这些球的字符串的名字是 "board" 。
    你手中的球不会超过5个，输入的数据中代表这些球的字符串的名字是 "hand"。
    输入的两个字符串均为非空字符串，且只包含字符 ‘R‘,‘Y‘,‘B‘,‘G‘,‘W‘。
算法：dfs+哈希。我们先将玩家手中的球统计出来。然后往桌上的塞。每次更新新的球即可。

class Solution 
public:
    string del(string board)
        for(int i=0;i<board.size();)
            int j=i;
            while(j<board.size()&&board[i]==board[j])j++;
            if(j-i>=3)
                return del(board.substr(0,i)+board.substr(j));
            else i=j;
        
        return board;
    
    int dfs(string board, unordered_map<char,int>&hash)
        board=del(board);
        if(board.size()==0)return 0;
        int rs=6,need=0;
        for(int i=0;i<board.size();)
            int j=i;
            while(j<board.size()&&board[i]==board[j])j++;
            need=3-(j-i);
            if(hash[board[i]]>=need)
                hash[board[i]]-=need;
                rs=min(rs,need+dfs(board.substr(0,i)+board.substr(j),hash));
                hash[board[i]]+=need;
            
            i=j;
        
        return rs;
    
    int findMinStep(string board, string hand) 
        unordered_map<char,int>hash;
        for(auto x:hand)hash[x]++;
        int res=dfs(board,hash);
        return res==6?-1:res;
    
;