leetcode 336. Palindrome Pairs 回文数字符串对 + Map加速暴力查询

Given a list of unique words, find all pairs of distinct indices (i, j) in the given list, so that the concatenation of the two words, i.e. words[i] + words[j] is a palindrome.

Example 1:
Given words = [“bat”, “tab”, “cat”]
Return [[0, 1], [1, 0]]
The palindromes are [“battab”, “tabbat”]
Example 2:
Given words = [“abcd”, “dcba”, “lls”, “s”, “sssll”]
Return [[0, 1], [1, 0], [3, 2], [2, 4]]
The palindromes are [“dcbaabcd”, “abcddcba”, “slls”, “llssssll”]

这道题暴力求解肯定超时,我先了好久没有想到更好的办法,网上看了个答案,它是使用HashMap做查询加速,本质还是循环来实现。

本体需要注意的地方是,要考虑到空字符串的特殊情况,所以内层循环是从0开始的,这个时候会造成重复,所以要使用set来保存结果

代码如下:

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;


/*
 * 使用暴力循环求解肯定超时,就是不断地拆解一个字符串,然后查询等等
 * 
 * 利用字典wmap保存单词 -> 下标的键值对
 * 遍历单词列表words,记当前单词为word,下标为idx:
 * 1). 若当前单词word本身为回文,且words中存在空串,则将空串下标bidx与idx加入答案
 * 2). 若当前单词的逆序串在words中,则将逆序串下标ridx与idx加入答案
 * 3). 将当前单词word拆分为左右两半left,right。
 *   3.1) 若left为回文,并且right的逆序串在words中,则将right的逆序串下标rridx与idx加入答案
 *   3.2) 若right为回文,并且left的逆序串在words中,则将left的逆序串下标idx与rlidx加入答案
 * 
 * */
class Solution 
{
    public List<List> palindromePairs(String[] words) 
    {
        if(words==null || words.length<=0)
            return new ArrayList<List>();      
        Map map=new HashMap<>();
        Set<List> set=new HashSet<>();

        //记录查询下标
        for(int i=0;ifor(int i=0;ifor(int j=0;j<=words[i].length();j++)
            {
                String left=words[i].substring(0,j);
                String right=words[i].substring(j);
                if(isPalindrome(left))
                {
                    int one=map.getOrDefault(new StringBuilder(right).reverse().toString(), -1);
                    if(one!=-1 && one!=i)
                    {
                        List list=new ArrayList<>();
                        list.add(one);
                        list.add(i);
                        set.add(list);
                    }
                }               
                if(isPalindrome(right))
                {
                    int one=map.getOrDefault(new StringBuilder(left).reverse().toString(), -1);
                    if(one!=-1 && one!=i)
                    {
                        List list=new ArrayList<>();
                        list.add(i);
                        list.add(one);
                        set.add(list);
                    }
                }
            }
        }
        List<List> res=new ArrayList<>(set);
        return res;
    }



    /*
     * 暴力循环求解
     * */
    public List<List> palindromePairsByBaoLi(String[] words) 
    {
        List<List> res=new ArrayList<>();
        if(words==null || words.length<=0)
            return res;
        for(int i=0;ifor(int j=i+1;jif(words[i].length()==words[j].length())
                {
                    if(isPalindrome(words[i]+words[j]))
                    {
                        List one=new ArrayList<>();
                        one.add(i);
                        one.add(j);
                        res.add(one);

                        List one1=new ArrayList<>();
                        one1.add(j);
                        one1.add(i);
                        res.add(one1);
                    }
                }else 
                {
                    if(isPalindrome(words[i]+words[j]))
                    {
                        List one=new ArrayList<>();
                        one.add(i);
                        one.add(j);
                        res.add(one);
                    }
                    if(isPalindrome(words[j]+words[i]))
                    {
                        List one=new ArrayList<>();
                        one.add(j);
                        one.add(i);
                        res.add(one);
                    }
                }
            }
        }
        return res;
    }

    public boolean isPalindrome(String s) 
    {
        int i=0,j=s.length()-1;
        while(iif(s.charAt(i)!=s.charAt(j))
                return false;   
            i++;
            j--;
        }       
        return true;
    }
}

下面是C++的做法

本质上还是暴力去做,不过使用了Map等数据机构来加速计算

代码如下:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;



class Solution 
{
public:
    vector<vector<int>> palindromePairs(vector<string>& words) 
    {
        set<vector<int>> res;
        unordered_map<string, int> index;
        for (int i = 0; i < words.size(); i++)
            index[words[i]] = i;

        for (int i=0;istring one = words[i];
            for (int j = 0; j <= one.length(); j++)
            {
                string left = one.substr(0,j);
                string right = one.substr(j);
                if (isPalindrome(left) == true)
                {
                    string tmp = right;
                    reverse(tmp.begin(), tmp.end());
                    if (index.find(tmp) != index.end() && index[tmp] != i)
                        res.insert({ index[tmp],i });
                }

                if (isPalindrome(right) == true)
                {
                    string tmp = left;
                    reverse(tmp.begin(), tmp.end());
                    if (index.find(tmp) != index.end() && index[tmp] != i)
                        res.insert({ i,index[tmp] });
                }
            }
        }
        return vector<vector<int>>(res.begin(),res.end());
    }

    bool isPalindrome(string s)
    {
        int i = 0, j = (int)s.length() - 1;
        while (i < j)
        {
            if (s[i] != s[j])
                return false;
            i++;
            j--;
        }
        return true;
    }
};

你可能感兴趣的:(leetcode,For,Java,leetcode,For,C++,需要好好想一下的题目)