【leetcode】336. Palindrome Pairs

题目如下:

解题思路:对于任意一个word,要找出在wordlist中是否存在与之能组成回文的其他words,有两种思路。一是遍历wordlist;二是对word本身进行分析,找出能组成回文的words,并判断是否存在于wordlist中。显然,第二种思路比较的次数要少很多。怎么找出能组成回文的words呢?只要把后面的字符依次往前复制,并判断是否为回文,直到全部字符复制完成为止,这就能得到所有能与之组成回文的words。以abcd作为前缀为例,首先把d复制到abcd前面得到dabcd,接下来依次是dcabcd,dcbabdc,dcbaabcd。其中dcbabdc和dcbaabcd是回文,再判断dcb 和 dcba 是否存在于wordlist中即可;同理,abcd作为后缀是一样的。

代码如下:

class Solution(object):
    def isPalindrome(self,s):
        return s == s[::-1]

    def palindromePairs(self, words):
        dic = {}
        for i,v in enumerate(words):
            dic[v] = i
        res = []
        for i,v in enumerate(words):
            if v == 'ab':
                pass
            subs = ''
            for j in v[::-1]:
                subs += j
                if subs in dic and dic[subs] != i and self.isPalindrome(subs + v):
                    res.append([dic[subs],i])

            subs = ''
            #考虑abcd和dcbd都存在于list中的情况,这里要少判断一位,避免出现重复的组合
            for j in v[:-1]:
                subs = j + subs
                if subs in dic and dic[subs] != i and self.isPalindrome(v + subs):
                    res.append([i,dic[subs]])

            # space,空能与其他本身就回文word组成回文word,这里单独考虑
            space = ''
            if space in dic and dic[space] != i and self.isPalindrome(v):
                res.append([dic[space],i])
                res.append([i,dic[space]])
        return res

猜你喜欢

转载自www.cnblogs.com/seyjs/p/9569247.html