Leetcode 336 回文对

本文借鉴了https://discuss.leetcode.com/topic/51771/clean-c-implementation

原题回放

给定一个无重复字符串列表words，找出所有的序偶对（i,j），如果words[i]+words[j]是一个回文串的话。

例如：

words = [“bat”, “tab”, “cat”]

返回 [[0,1], [1,0]]

回文串即[“battab”, “tabbat”]

代码

class Solution{
public:
bool isPalindrome(string& s, int start, int end){
while(start < end)
if(s[start++] != s[end--])
return false;
}

vector<vector<int>> palindromePairs(vector<string> words){
vector<vector<int>> ans;
unordered_map<string, int> dict;
int len = words.size();
for(int i=0; i<len; i++)
dict[words[i]] = i;
for(int i=0; i<len; i++){
string cur = words[i];
int clen = cur.size();
for(int j=0; j<=clen; j++){
// 找后缀
if(isPalindrome(cur, j, clen-1)){
// 后缀可以为空
string suffix = cur.substr(0, j);
reverse(suffix.begin(), suffix.end());
if(dict.find(suffix)!=dict.end() && i!=dict[suffix])
ans.push_back({i, dict[suffix]});
}
// 找前缀
if(j>0 && isPalindrome(cur, 0, j-1)){
string prefix = cur.substr(j);
reverse(prefix.begin(), prefix.end());
if(dict.find[prefix]!=dict.end() && i!=dict[prefix])
ans.push_back({dict[prefix], i});
}
}
}
return ans;
}
};

单纯记录一下，方便以后复习。