Java实现 LeetCode 488 祖玛游戏
488. 祖玛游戏
回忆一下祖玛游戏。现在桌上有一串球,颜色有红色®,黄色(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’。
class Solution {
int DFS(String board,Map<Character,Integer> num){
if(board.length()==0) return 0;
int ans = 0x3f3f3f3f,l = board.length(),need,i,j;
for(i = 0;i < l;i = j){
j = i + 1;
while(j < l && board.charAt(j) == board.charAt(i)) j++;
need = 3 - (j - i);//当前位置要满足三个球消除需要几个额外的。
if(need <= num.get(board.charAt(i))){
need = Math.max(0,need);//可能存在有连续消除的情况。此时need为负数。例如WWWW
num.put(board.charAt(i),num.get(board.charAt(i))-need);
String subStr = board.substring(0,i) + board.substring(j);
int cnt = DFS(subStr,num);
if(cnt >= 0)
ans = Math.min(ans , need + cnt);
num.put(board.charAt(i), num.get(board.charAt(i))+need);
}
}
if(ans == 0x3f3f3f3f) return -1;
else return ans;
}
int findMinStep(String board, String hand) {
Map <Character,Integer> num = new HashMap<Character,Integer>();//这里要定义char
num.put('R', 0);
num.put('Y', 0);
num.put('B', 0);
num.put('G', 0);
num.put('W', 0);
int l = hand.length();
for(int i = 0;i < l;i++){
num.put(hand.charAt(i), num.get(hand.charAt(i))+1);
}
return DFS(board,num);
}
}