LeetCode 488. 祖玛游戏

题目描述
回忆一下祖玛游戏。现在桌上有一串球，颜色有红色(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)
我们先将手中的球用哈希表来存储一下，然后看桌上的球能否在哈希表里的球添加后消除，然后消除后递归处理剩下的。中间记录需要的球数，用来更新需要球数的最小值。如果最小值超出了手中球的个数，则无法消除。

时间复杂度分析：桌上的球不会超过20个，手中的球不会超过5个，所以时间复杂度为O(m+n).

C++ 代码
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;
    }
};