题目描述
回忆一下祖玛游戏。现在桌上有一串球,颜色有红色(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;
}
};
过不了了,
input:”RRWWRRBBRR”
“WB”
Output:-1
Expected:2