LeetCode Shortest Word Distance II

LeetCode Shortest Word Distance II

题目

思路

跟上一题一样，但是在这题的背景下，求最短距离的函数将会被调用多次。因此我们可以先把字符串的每个位置都记录下来，这样找最短距离的时候就好很多。而且，记录位置的时候都是有序的，这样，在求最短距离的时候就类似于归并排序，只需要O(n+m)的时间复杂度即可，其中n为word1的位置数量，m为word2的位置数量。

那为什么不直接用一个二维数组来直接保存答案了？这样在每次调用最短距离函数的时候直接输出一个数不是更快吗？这是因为，虽然题目说到最短距离的函数会被多次调用，但是并不是调用n次，也就是说这个多次其实并不算多。如果直接保存答案，那我们需要O(n^2)的时间复杂度；如果按照上面说的做法，构造dis的时间复杂度是O(nlogn)，之后的求最短距离可以认为是常数时间（除非坑爹样例）。

当然，我们还可以维护一个容器来记录之前求过的答案，这样如果求最短距离的函数的参数和之前某一次调用是一样的话，效率会提高（一点点）。

还需要注意的是，在代码中，如果对dis有重复的访问，那么用迭代器来记录第一次访问的位置，第二次就可以直接访问迭代器了，这样也会带来效率上的提升。

其实如果追求极致，也可以将字符串转化为数字，这样在dis里面的访问就是O(1)，考虑到今晚就要期中考试，就不写那么复杂了。

代码

class WordDistance {
public:
    WordDistance(vector<string>& words) {
        dis.clear();
        for (int i = words.size() - 1; i >= 0; i--) {
            map<string, vector<int> >::iterator iter = dis.find(words[i]);
            if (iter != dis.end()) iter->second.push_back(i);
            else dis[words[i]] = vector<int>(1, i);
        }
    }

    int shortest(string word1, string word2) {
        int ans = INT_MAX;
        map<string, vector<int> >::iterator iter1 = dis.find(word1);
        map<string, vector<int> >::iterator iter2 = dis.find(word2);
        for (int p = iter1->second.size() - 1, q = iter2->second.size() - 1; p >= 0 && q >= 0;) {
            int pos1 = iter1->second[p];
            int pos2 = iter2->second[q];
            if (pos1 < pos2) {
                ans = min(ans, pos2 - pos1);
                p--;
            }
            else {
                ans = min(ans, pos1 - pos2);
                q--;
            }
        }
        return ans;
    }
private:
    map<string, vector<int> > dis;
};


// Your WordDistance object will be instantiated and called as such:
// WordDistance wordDistance(words);
// wordDistance.shortest("word1", "word2");
// wordDistance.shortest("anotherWord1", "anotherWord2");

更多

更多LeetCode题目（包括付费题）和题解。