1.题目描述
给定一组唯一的单词, 找出所有不同 的索引对(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"]
2.解题思路
对于示例1的一个单词 lls,如何找其他单词把它拼成回文串。
两种方式,一种是在左边加,比如有个单词s,那么可以拼车 slls;一种右边加,单词为sssll,拼成 llssssll
由上面简单想法,可以把单词分成两部分考虑,看代码很好理解!
作者:powcai
链接:https://leetcode-cn.com/problems/palindrome-pairs/solution/python-zi-dian-by-powcai/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
3.代码实现
class Solution:
def palindromePairs(self, words: List[str]) -> List[List[int]]:
# 本身就是回文串单词
palidStr = []
# 翻转单词记录位置
rev_words = {}
# 结果
res = []
for idx, word in enumerate(words):
rev_words[word[::-1]] = idx
# 为了防止数组里有空字符串("")
if word == word[::-1]:
palidStr.append(idx)
for idx, word in enumerate(words):
if word:
# 这里没有 len(word) + 1
for i in range(len(word)):
left, right = word[:i], word[i:]
# 是否存在在单词左边加 使得成为回文串
if left == left[::-1] and right in rev_words and idx != rev_words[right]:
res.append([rev_words[right], idx])
# 同理
if right == right[::-1] and left in rev_words and idx != rev_words[left]:
res.append([idx, rev_words[left]])
else:
# 对于空字符串
for loc in palidStr:
if loc != idx:
res.append([idx, loc])
return res