LeetCode 336. Palindrome Pairs

题意：给出一个有n个单词的字典，问有多少不同的两个单词的组合使得新组合成的单词是回文串。

思路：对于每个单词，从左到右遍历一遍，对于任意一个位置，如果它的左边是回文串且右边的逆序在字典中出现，那么就存在这么一种组合，同理，如果它的右边是回文串且左边的逆序在字典中出现，那么也存在这么一种组合。

class Solution {
public:
    vector<vector<int>> palindromePairs(vector<string>& words) {
    	vector<vector<int>> ans;
    	unordered_map<string,int> dic;
    	for(int i = 0;i<words.size();++i)
    		dic[words[i]] = i;
    	int word_len = words.size();
    	for(int i = 0;i<word_len;++i)
    	{
    		int len = words[i].size();
    		if(len == 0)
    		    for(int k = 0;k<word_len;k++)
    		        if(k != i && palindrome(words[k])) ans.push_back(vector<int>{i,k});
    		        
    		
    		dic.erase(words[i]);
    		for(int j = 0;j<len;++j)
    		{
    			string subL = words[i].substr(0,j);
    			string subR = words[i].substr(j);
    			string revL,revR;
    			revL.assign(subL.rbegin(),subL.rend());
    			revR.assign(subR.rbegin(),subR.rend());

    			if((dic.find(revL) != dic.end())&&palindrome(subR))
    				ans.push_back(vector<int>{i,dic[revL]});
    			if((dic.find(revR) != dic.end())&&palindrome(subL))
    				ans.push_back(vector<int> {dic[revR], i});
    		}
    		dic[words[i]] = i;
    	}
    	return ans;
    }
private:
    bool palindrome(string str)
    {
    	int left = 0,right = str.size()-1;
    	while(left <= right)
    	{
    		if(str[left] != str[right])
    			return false;
    		left++,right--;
    	}
    	return true;
    }
};