【Leetcode】245. Shortest Word Distance III

题目地址：

https://leetcode.com/problems/shortest-word-distance-iii/description/

给定一个字符串数组$A$，再给定两个字符串$s_1,s_2$，求这两个字符串在$A$中的最短距离。如果$s_1=s_2$，则需要将它们视为在两个不同位置。题目保证两个字符串在$A$中，并且如果$s_1=s_2$的时候，其在$A$中出现多于等于两次。

代码如下：

class Solution {
 public:
  int shortestWordDistance(vector<string>& v, string s1, string s2) {
    int res = v.size();
    if (s1 == s2) {
      int last = -1;
      for (int k = 0; k < v.size(); k++)
        if (v[k] == s1) {
          if (~last) res = min(res, k - last);
          last = k;
        }
    } else {
      for (int k = 0, i = -1, j = -1; k < v.size(); k++) {
        if (v[k] == s1) i = k;
        if (v[k] == s2) j = k;
        if (~i && ~j) res = min(res, abs(i - j));
      }
    }

    return res;
  }
};

时间复杂度$O({\sum }_i{l}_{A[i]})$，空间$O(1)$。