Shortest Word Distance

Given a list of words and two words word1 and word2, 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.

 

解法一：

public class Solution {
    public int shortestDistance(String[] words, String word1, String word2) 
    { 
        int idx1=-1;
        int idx2=-1;
        int min=Integer.MAX_VALUE;
        for(int i=0;i<words.length;i++)
        {
            if(words[i].equals(word1))
            {
                idx1=i;
                if(idx2!=-1)min=Math.min(min,idx1-idx2);
            }
            if(words[i].equals(word2))
            {
                idx2=i;
                if(idx1!=-1)min=Math.min(min,idx2-idx1);
            }
            
        }
        return min;
        
    }
}

 

解法二：

hashtable

public class Solution {
    public int shortestDistance(String[] words, String word1, String word2) 
    {
        HashMap<String, ArrayList<Integer>> map = new HashMap<String,ArrayList<Integer>>();
        for(int i=0;i<words.length;i++)
        {
            String word = words[i];
            if(!map.containsKey(word))
            {
                map.put(word,new ArrayList<Integer>());
            }
            map.get(word).add(i);
        }
        
        ArrayList<Integer> list1 = map.get(word1);
        ArrayList<Integer> list2 = map.get(word2);
        int i = 0;
        int j = 0;
        int min = words.length-1;
        while(i<list1.size()&&j<list2.size())
        {
            int index1 = list1.get(i);
            int index2 = list2.get(j);
            
                min = Math.min(Math.abs(index1-index2),min);
                if(index1<index2)
                {
                    i++;
                    
                }
                else
                {
                    j++;
                }
            
        }
        return min;
    }
}