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");