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+哈希。我们先将玩家手中的球统计出来。然后往桌上的塞。每次更新新的球即可。

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;
    }
};

 

posted @ 2019-07-10 14:40  YF-1994  阅读(587)  评论(0编辑  收藏  举报