Question

Formatted question description: https://leetcode.ca/all/244.html

 244	Shortest Word Distance II

 This is a follow up of Shortest Word Distance.
 The only difference is now you are given the list of words and
 your method will be called repeatedly many times with different parameters.

 How would you optimize it?

 Design a class which receives a list of words in the constructor,
 and implements a method that takes two words word1 and word2
 and return the shortest distance between these two words in the list.

 For example,
 Assume that words = ["practice", "makes", "perfect", "coding", "makes"].

 Given word1 = “coding”, word2 = “practice”, return 3.
 Given word1 = "makes", word2 = "coding", return 1.

 Note:
 You may assume that word1 does not equal to word2, and word1 and word2 are both in the list.

 @tag-array

Algorithm

Need to call the function to find the shortest word distance multiple times, then the solution to the previous problem is very inefficient.

Use HashMap to create a mapping between each word and all its appearing positions, and then when looking for the shortest word distance, you only need to take out the word’s mapped position array in HashMap for pairwise comparison.

Code

Java

import java.util.ArrayList;
import java.util.HashMap;

public class Shortest_Word_Distance_II {

    public class WordDistance {

        HashMap<String, ArrayList<Integer>> map; // @note: duplicated cases

        public WordDistance(String[] words) {

            map = new HashMap<String, ArrayList<Integer>>();

            for (int i = 0; i < words.length; i++) {
                if (map.containsKey(words[i])) {
                    map.get(words[i]).add(i);
                } else {
                    ArrayList<Integer> list = new ArrayList<Integer>();
                    list.add(i);
                    map.put(words[i], list);
                }
            }
        }

        public int shortest(String word1, String word2) {
            ArrayList<Integer> l1 = map.get(word1);
            ArrayList<Integer> l2 = map.get(word2);

            int result = Integer.MAX_VALUE;

            // better than 2 for-loop
            int p1 = 0;
            int p2 = 0;
            while (!l1.isEmpty() || !l2.isEmpty()) {
                int v1 = l1.isEmpty() ? Integer.MAX_VALUE : l1.get(p1);
                int v2 = l2.isEmpty() ? Integer.MAX_VALUE : l2.get(p2);
                result = Math.min(result, Math.abs(v1 - v2));

                if (v1 < v2) {
                    p1++;
                } else {
                    p2++;
                }
            }
//            for (int i1 : l1) {
//                for (int i2 : l2) {
//                    result = Math.min(result, Math.abs(i1 - i2));
//                }
//            }

            return result;
        }
    }
}

All Problems

All Solutions