字符串数组中两个字符的最短距离

[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.

需要注意:

  1. 该数组中得字符串有可能出现多次,比如例子中的makes;

这个问题比较简单,按照下面的步骤可以简单的做出来:

  1. 记录给定两个字符串出现的位置,会得到两个数组;假设为a和b;

  2. 然后计算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的;因为后者的距离肯定大于前者;用下面的图可能更清楚;

字符串数组中两个字符的最短距离_第1张图片

字符串数组中两个字符的最短距离_第2张图片

字符串数组中两个字符的最短距离_第3张图片

字符串数组中两个字符的最短距离_第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;
}


你可能感兴趣的:(LeetCode)