题目:
Given a list of unique words. Find all pairs of distinct indices (i, j) in the given list, so that the concatenation of the two words, i.e. words[i] + words[j] is a palindrome. Example 1: Given words = ["bat", "tab", "cat"] Return [[0, 1], [1, 0]] The palindromes are ["battab", "tabbat"] Example 2: Given words = ["abcd", "dcba", "lls", "s", "sssll"] Return [[0, 1], [1, 0], [3, 2], [2, 4]] The palindromes are ["dcbaabcd", "abcddcba", "slls", "llssssll"]
错误思路:words里面所有的String组合,判断两两连接后的String是否回文。TLE!
正确思路:针对每个String,单独分析。
String 截取两半 |-----str1------|-----str2-----|
如果str1是回文,那么 把str2翻转,放在左边,新的String也是回文,只要判断翻转后的str2是否在words里即可。 |-----2rts------|------str1-----|-----str2-----|
如果str2是回文,那么 把str1翻转,放在右边,新的String也是回文,只要判断翻转后的str1是否在words里即可。|------str1-----|-----str2-----|------1rts------|
但是此过程会出现重复。举例,如果 words里面有abc 和 cba ,分析abc的时候,可能会得到把cba放在abc的左边;分析cba的时候可能会得到把abc放在右边。
所以要过滤重复的。可以在第二个判断中,也就是如果str2是回文的情况下,如果str2为空,长度为0,则不添加结果。
因为str2为空时,新String为|------str1-----|------1rts------|,这个会和 分析[翻转后的str1]时,第一个判断中str1为空重复。
代码:
public List<List<Integer>> palindromePairs(String[] words) { List<List<Integer>> results = new ArrayList<>(); Map<String,Integer> map = new HashMap<>(); for(int i=0;i<words.length;i++) map.put(words[i], i); for(int i=0;i<words.length;i++){ for(int j=0;j<=words[i].length();j++){ String str1 = words[i].substring(0, j); String str2 = words[i].substring(j); if(isPalindrome(str1)){ String needAddToLeft = new StringBuffer(str2).reverse().toString(); if(map.containsKey(needAddToLeft)&&map.get(needAddToLeft)!=i){ results.add(new ArrayList<>(Arrays.asList(map.get(needAddToLeft),i))); } } if(isPalindrome(str2)){ String needAddToRight = new StringBuffer(str1).reverse().toString(); if(map.containsKey(needAddToRight)&&map.get(needAddToRight)!=i&&str2.length()!=0){ results.add(new ArrayList<>(Arrays.asList(i,map.get(needAddToRight)))); } } } } return results; } private boolean isPalindrome(String s){ if(s.equals("")) return true; int start =0; int end = s.length()-1; while(start<=end){ if(s.charAt(start)!=s.charAt(end)) return false; else{ start++; end--; } } return true; }