[leetcode] https://leetcode.com/problems/shortest-word-distance/
For example,
Assume that words = ["practice", "makes", "perfect", "coding", "makes"]
.
Given word1 = “coding”
, word2 = “practice”
, return 3.
Given word1 = "makes"
, word2 = "coding"
, return 1.
需要注意:
该数组中得字符串有可能出现多次,比如例子中的makes;
这个问题比较简单,按照下面的步骤可以简单的做出来:
记录给定两个字符串出现的位置,会得到两个数组;假设为a和b;
然后计算abs(a[i] - b[j])的最小值即可;
第一步扫描一遍字符串数组即可;第二步,如果简单的用两层循环来做,因为a和b的长度都有可能是n(比如各占一半),那么用O(n * n)的时间;
事实上,有意思的地方就在于,第二步可以在线性时间内解出来;这里有一个隐藏的特性,扫描原数组的时候得到的两个字符的位置数组,是有序的;比如数组 [a, b, a, b, b, a], 那么a的位置数组为[0, 2, 5], b的为[1, 3, 4]; 假设我们比较到了位置2和位置3,我们是没有必要去比较位置2和4的;因为后者的距离肯定大于前者;用下面的图可能更清楚;
所以,可以得到这样的关系,当a < b的时候,增加a的坐标,当a > b的时候,增加b的坐标;这样就可以在o(n)的时间内完成;
public int shortestDistance(String[] words, String word1, String word2) {
int[] indexes1 = new int[words.length];
int[] indexes2 = new int[words.length];
int m = 0, n = 0;
for (int i = 0; i < words.length; i++) {
String word = words[i];
if (word.equals(word1)) {
indexes1[m++] = i;
} else if (word.equals(word2)) {
indexes2[n++] = i;
}
}
int dist = Integer.MAX_VALUE;
for (int i = 0, j = 0; i < m && j < n; ) {
int x = indexes1[i];
int y = indexes2[j];
dist = Math.min(dist, Math.abs(x - y));
if (x < y) {
i++;
} else {
j++;
}
}
return dist;
}