LeetCode Shortest Word Distance

LeetCode Shortest Word Distance

题目

思路

直接对两个目标字符串进行定位，然后更新最小距离即可。O(n)，注意循环最好倒过来，这是因为这样的话words.size()方法只需要调用一次，如果写成i < words.size()的话会增加不必要的耗时。

代码

class Solution {
public:
    int shortestDistance(vector<string>& words, 
        string word1, string word2) {

        int ans = words.size(), p = -1, q = -1;

        for (int i = words.size() - 1; i >= 0; i--) {
            if (word1 == words[i]) p = i;
            if (word2 == words[i]) q = i;
            if (p != -1 && q != -1) ans = min(abs(p - q), ans);
        }

        return ans;
    }
};

更多

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