leetcode **336. 回文对(待研究)

【题目】**336. 回文对

给定一组唯一的单词, 找出所有不同 的索引对(i, j),使得列表中的两个单词, words[i] + words[j] ,可拼接成回文串。

示例 1:

输入: ["abcd","dcba","lls","s","sssll"]
输出: [[0,1],[1,0],[3,2],[2,4]] 
解释: 可拼接成的回文串为 ["dcbaabcd","abcddcba","slls","llssssll"]

示例 2:

输入: ["bat","tab","cat"]
输出: [[0,1],[1,0]] 
解释: 可拼接成的回文串为 ["battab","tabbat"]

【解题思路1】枚举前缀后缀

假设存在两个字符串 s1和 s2,s1+s2是一个回文串,记这两个字符串的长度分别为 len1和 len
2,我们分三种情况进行讨论:

  • len1=len2 , s1是 s2的翻转。
  • len1>len2,可以将 s1拆成左右两部分:t1和 t2,其中 t1是 s2的翻转,t2是一个回文串。
  • len1<len2,可以将 s2 拆成左右两部分:t1 和 t2,其中 t2 是 s1的翻转,t1 是一个回文串。

这样,对于每一个字符串,我令其为 s1 和 s2中较长的那一个,然后找到可能和它构成回文串的字符串即可。
具体地说,枚举每一个字符串 k,令其为 s1 和 s2中较长的那一个,那么 k 可以被拆分为两部分,t1和 t2。

  • 当 t1 是回文串时,符合情况 3,只需要查询给定的字符串序列中是否包含 t2 的翻转。
  • 当 t2 是回文串时,符合情况 2,我们只需要查询给定的字符串序列中是否包含 t1 的翻转。

也就是说,枚举字符串 k 的每一个前缀和后缀,判断其是否为回文串。如果是回文串,就查询其剩余部分的翻转是否在给定的字符串序列中出现即可。

注意到空串也是回文串,所以我们可以将 k 拆解为 k+∅ 或 ∅+k,这样就能将情况 1 也解释为特殊的情况 2 或情况 3。

而要实现这些操作,只需要设计一个能够在一系列字符串中查询「某个字符串的子串的翻转」是否存在的数据结构,有两种实现方法:

  • 可以使用字典树存储所有的字符串。在进行查询时,将待查询串的子串逆序地在字典树上进行遍历,即可判断其是否存在。

  • 可以使用哈希表存储所有字符串的翻转串。在进行查询时,判断带查询串的子串是否在哈希表中出现,就等价于判断了其翻转是否存在。

字典树

class Solution {
    
    
    class Node {
    
    
        int[] ch = new int[26];
        int flag;

        public Node() {
    
    
            flag = -1;
        }
    }

    List<Node> tree = new ArrayList<Node>();

    public List<List<Integer>> palindromePairs(String[] words) {
    
    
        tree.add(new Node());
        int n = words.length;
        for (int i = 0; i < n; i++) {
    
    
            insert(words[i], i);
        }
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        for (int i = 0; i < n; i++) {
    
    
            int m = words[i].length();
            for (int j = 0; j <= m; j++) {
    
    
                if (isPalindrome(words[i], j, m - 1)) {
    
    
                    int leftId = findWord(words[i], 0, j - 1);
                    if (leftId != -1 && leftId != i) {
    
    
                        ret.add(Arrays.asList(i, leftId));
                    }
                }
                if (j != 0 && isPalindrome(words[i], 0, j - 1)) {
    
    
                    int rightId = findWord(words[i], j, m - 1);
                    if (rightId != -1 && rightId != i) {
    
    
                        ret.add(Arrays.asList(rightId, i));
                    }
                }
            }
        }
        return ret;
    }

    public void insert(String s, int id) {
    
    
        int len = s.length(), add = 0;
        for (int i = 0; i < len; i++) {
    
    
            int x = s.charAt(i) - 'a';
            if (tree.get(add).ch[x] == 0) {
    
    
                tree.add(new Node());
                tree.get(add).ch[x] = tree.size() - 1;
            }
            add = tree.get(add).ch[x];
        }
        tree.get(add).flag = id;
    }

    public boolean isPalindrome(String s, int left, int right) {
    
    
        int len = right - left + 1;
        for (int i = 0; i < len / 2; i++) {
    
    
            if (s.charAt(left + i) != s.charAt(right - i)) {
    
    
                return false;
            }
        }
        return true;
    }

    public int findWord(String s, int left, int right) {
    
    
        int add = 0;
        for (int i = right; i >= left; i--) {
    
    
            int x = s.charAt(i) - 'a';
            if (tree.get(add).ch[x] == 0) {
    
    
                return -1;
            }
            add = tree.get(add).ch[x];
        }
        return tree.get(add).flag;
    }
}

哈希表

class Solution {
    
    
    List<String> wordsRev = new ArrayList<String>();
    Map<String, Integer> indices = new HashMap<String, Integer>();

    public List<List<Integer>> palindromePairs(String[] words) {
    
    
        int n = words.length;
        for (String word: words) {
    
    
            wordsRev.add(new StringBuffer(word).reverse().toString());
        }
        for (int i = 0; i < n; ++i) {
    
    
            indices.put(wordsRev.get(i), i);
        }

        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        for (int i = 0; i < n; i++) {
    
    
            String word = words[i];
            int m = words[i].length();
            if (m == 0) {
    
    
                continue;
            }
            for (int j = 0; j <= m; j++) {
    
    
                if (isPalindrome(word, j, m - 1)) {
    
    
                    int leftId = findWord(word, 0, j - 1);
                    if (leftId != -1 && leftId != i) {
    
    
                        ret.add(Arrays.asList(i, leftId));
                    }
                }
                if (j != 0 && isPalindrome(word, 0, j - 1)) {
    
    
                    int rightId = findWord(word, j, m - 1);
                    if (rightId != -1 && rightId != i) {
    
    
                        ret.add(Arrays.asList(rightId, i));
                    }
                }
            }
        }
        return ret;
    }

    public boolean isPalindrome(String s, int left, int right) {
    
    
        int len = right - left + 1;
        for (int i = 0; i < len / 2; i++) {
    
    
            if (s.charAt(left + i) != s.charAt(right - i)) {
    
    
                return false;
            }
        }
        return true;
    }

    public int findWord(String s, int left, int right) {
    
    
        return indices.getOrDefault(s.substring(left, right + 1), -1);
    }
}

【解题思路2】字典树 + manacher

猜你喜欢

转载自blog.csdn.net/XunCiy/article/details/107830837
今日推荐