131. Palindrome Partitioning--back tracking 和93. Restore IP Addresses本质一样

给你一个字符串， 输出他的pattitioning 都是 palindrome 的组合

Input: "aab"
Output:
[
  ["aa","b"],
  ["a","a","b"]
]

 和93 题实际上一样， 给你个字符串长度比较是 3，  你相当于有一个nums = {1,2,3} 的数组，每次去取 一个数求和，当和的长度恰好等于 字符串长度 并且都是palindrome 时就是结果。

但这道题不能先 把长度数组求出来 再去判断是否符合要求， 比如 先求出 [1,1,1] 这样的长度数组，再判断就会TLE， 因为大部分字符都是不符合palindrome 的，所以产生一个长度就判断是否palindrome 不符合要求就continue才不会TLE。 

以下当input = "seeslaveidemonstrateyetartsnomedievalsees" 会TLE。 

class Solution {
    public List<List<String>> partition(String s) {
        int[] nums = new int[s.length()];
        for(int i=1; i<=s.length(); i++) nums[i-1] = i;
        List<List<String>> result = new ArrayList<>();
        dfs(new ArrayList<>(), nums, result,s,s.length());
        return result;
        
    }
    
    private void dfs(List<Integer> curResult, int[] nums, List<List<String>> result, String s, int remain){
        if(remain == 0){
          List<String> cur = new ArrayList<>();  
          boolean flag = false;  
          int start = 0;
          int end = 0;  
          for(int num: curResult){
              end += num;
              String substr = s.substring(start,end);
              if(!isValid(substr)) return;
              cur.add(substr);
              start = end;
          } 
          result.add(cur);  
          return;  
        }
        
        if(remain <0) return;
        
        for(int i=0; i<nums.length;i++){
           curResult.add(nums[i]) ;
           dfs(curResult,nums,result,s,remain-nums[i]);
           curResult.remove(curResult.size()-1);
        }
    }
    

    private boolean isValid(String s){
        int len = s.length();
        for(int i=0; i<len/2;i++){
            if(s.charAt(i) != s.charAt(len-i-1)) return false;
        }
        return true;
    }    
}

 

改进后的code:  这段code 一开始对于 "efe" 怎么 都产生不了 "efe"自身这个output, 然后发现for 循环里i 怎么都达不到 2 .

                         原因在于 31行用了 return 而不是continue ， 对于DFS 真正能return 是在 达到了叶子节点时，如果该节点还有兄弟节点，return 则会导致 不继续访问兄弟节点。对于 26行，应该用return 而不是continue, 虽然continue 可以work，但改成return 效率会提升很多。

 

class Solution {
    public List<List<String>> partition(String s) {
        int[] nums = new int[s.length()];
        for(int i=1; i<=s.length(); i++) { nums[i-1] = i;}
        
        List<List<String>> result = new ArrayList<>();
        dfs(new ArrayList<>(), nums, result,s,0);
        return result;
        
    }
    
    private void dfs(List<String> curResult, int[] nums, List<List<String>> result, String s, int start){
       
        if(start == s.length()){
             result.add(new ArrayList<>(curResult));  
             return;
          } 
        
        if(start >s.length()) return;
        
     
        
        for(int i=0; i<nums.length ;i++){
         
          
           if(start+nums[i]>s.length()) continue;
           
           String substr = s.substring(start,start+nums[i]);
        
           if(!isValid(substr)) {
             continue;  
           } 
           curResult.add(substr); 
           start += nums[i];
           dfs(curResult,nums,result,s,start); 
           curResult.remove(curResult.size()-1);
           start -= nums[i];    
        }
    }
    

    private boolean isValid(String s){
        int len = s.length();
        for(int i=0; i<len/2;i++){
            if(s.charAt(i) != s.charAt(len-i-1)) return false;
        }
        return true;
    }    
}