lintcode:单词切分

import java.util.Iterator;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;
// write your code here
public class Main{
    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        Main m = new Main();
        while(in.hasNext()){
            String s = in.nextLine();
            String[] str = in.nextLine().split(" ");
            Set<String> dict = new TreeSet<String>();
            for(int i = 0;i<str.length;i++)
                dict.add(str[i]);
            int start = 0;
            boolean flag = m.wordBreak(s, dict, start);
            System.out.println(flag);
        }
    }
    public boolean wordBreak(String s,Set<String> dict,int start){

　　　　　　if((s==null ||s.length() ==0) && (dict == null || dict.size()==0))
　　　　　　　　return true;

        Iterator it = dict.iterator(); 
        if(start == s.length())
            return true;
        while(it.hasNext()){
            String t = (String)it.next();
            int end = start + t.length();
            if(end > s.length())
                continue;
            if(s.substring(start,end).equals( t )){
                if(wordBreak(s,dict,end)){
                    return true;
                }
            }
        }
        return false;
    }
}

public class Solution {
    /**
     * @param s: A string s
     * @param dict: A dictionary of words dict
     */
    public boolean wordBreak(String s, Set<String> dict) {
        // write your code here   
        if((s==null ||s.length() ==0) && (dict == null || dict.size()==0))
            return true;
        return wordBreak(s,dict,0);
    }
     public boolean wordBreak(String s,Set<String> dict,int start){
        boolean dp[] = new boolean[s.length() + 1];
        dp[0] = true;//初始值
        for(int i = 0;i<s.length();i++){
            if(!dp[i])
                continue;
            for(String t:dict){
                int len = t.length();
                int end = i+ len;
                if(end > s.length())
                    continue;
                if(s.substring(i,end).equals(t)){
                    dp[end] = true;
                }
            }
        }
        return dp[s.length()];
    }
    
}