320. Generalized Abbreviation-- back tracking

把一个字符串中字母用数字代替,产生所有的组合数
Input: "word" Output: ["word", "1ord", "w1rd", "wo1d", "wor1", "2rd", "w2d", "wo2", "1o1d", "1or1", "w1r1", "1o2", "2r1", "3d", "w3", "4"]

和78 subsets本质上也是一样的: https://leetcode.com/problems/subsets/description/

画出递归数吧: 假设word = "abc", 递归数如下:

需要注意的是, 如果某个节点已经是数字了,则接下来不能继续放数字 ,例如 "a11" 是不合法的。

先写了如下code ,在 input = "interaction" 时WA了, 而其他短字符串都可以过。 

因为 最后一行 “curResult.deleteCharAt(curResult.length()-1);”   input 字符串长度超过10后,每次加的长度 就是2位数了, 所以不能 只 减去1。 

class Solution {
    public List<String> generateAbbreviations(String word) {
        List<String> result = new ArrayList<>();
        
        dfs(new StringBuilder(), result, word,0);
        return result;
    }
    
    private void dfs(StringBuilder curResult,List<String> result, String word, int cur_index){
        if(curResult.length() == word.length() || cur_index == word.length()){
            result.add(curResult.toString());
            return;
        }
        
        // put cur letter
        curResult.append(word.charAt(cur_index));
        dfs(curResult,result, word,cur_index+1);
        curResult.deleteCharAt(curResult.length()-1);
        
        for(int i=1; i<=word.length()-cur_index; i++){ // wrd: cur_index = 0, len =3, 可以放 1,2,3
            // 当前result 里 最后一个字符得是字母才能 放数字
            if(curResult.length()==0 || Character.isLetter(curResult.charAt(curResult.length()-1)) ){
                curResult.append(i);
                dfs(curResult,result,word,cur_index+i);
                curResult.deleteCharAt(curResult.length()-1);
            }
            
        }
    }
    
    
}

修改后的code, 每次记录 上一次的stringbuild 长度,然后  重新setLeng 到之前的长度即可:

class Solution {
    public List<String> generateAbbreviations(String word) {
        List<String> result = new ArrayList<>();
        dfs(new StringBuilder(), result, word,0);
        return result;
    }
    
    private void dfs(StringBuilder curResult,List<String> result, String word, int cur_index){
        if(curResult.length() == word.length() || cur_index == word.length()){
            result.add(curResult.toString());
            return;
        }
       
        int len = curResult.length(); // 每次记录长度
        // put cur letter
        curResult.append(word.charAt(cur_index));
        dfs(curResult,result, word,cur_index+1);
        curResult.setLength(len);
        //put the 剩下的可能的长度
        for(int i=1; i<=word.length()-cur_index; i++){ // wrd: cur_index = 0, len =3, 可以放 1,2,3
            // 当前result 里 最后一个字符得是字母才能 放数字
            if(curResult.length()==0 || Character.isLetter(curResult.charAt(curResult.length()-1)) ){
                curResult.append(i);
                dfs(curResult,result,word,cur_index+i);
                curResult.setLength(len); //恢复之前的长度
            }   
        }
    }  
}
 


猜你喜欢

转载自www.cnblogs.com/keepAC/p/9944641.html