給定一組單詞words,編寫一個程序,找出其中的最長單詞,且該單詞由這組單詞中的其他單詞組合而成。若有多個長度相同的結果,返回其中字典序最小的一項,若沒有符合要求的單詞則返回空字符串。
示例:
輸入: [“cat”,“banana”,“dog”,“nana”,“walk”,“walker”,“dogwalker”]
輸出: “dogwalker”
解釋: "dogwalker"可由"dog"和"walker"組成。
代碼
class Solution {public String longestWord(String[] words) {Map<Character,Set<String>> map=new HashMap<>();for(String s:words)//記錄首字母和對應的字符串{ if(s.length()==0) continue;if(!map.containsKey(s.charAt(0)))map.put(s.charAt(0),new HashSet<>());map.get(s.charAt(0)).add(s);}Arrays.sort(words,((o1, o2) -> o1.length()==o2.length()?o1.charAt(0)-o2.charAt(0):o2.length()-o1.length()));//排序,長的在前面,字典序小的在前面for(String s:words){if(s.length()==0) continue;if(map.get(s.charAt(0)).size()>1)//有對應的首字母{if(getLongestWord(s,map,0))return s;}}return "";}public boolean getLongestWord(String s,Map<Character,Set<String>> map,int loc) {if(loc==s.length()) return true;if(!map.containsKey(s.charAt(loc))) return false;//沒用對應首字母for(String t:map.get(s.charAt(loc)))//嘗試對應首字母的字符串{if(t.length()>=s.length()||loc+t.length()>s.length()) continue;if(s.substring(loc,loc+t.length()).equals(t)){if(getLongestWord(s,map,loc+t.length()))return true;}}return false;}
}