245.最短单词距离III

class Solution {
    public int shortestWordDistance(String[] wordsDict, String word1, String word2) {
        int len = wordsDict.length;
        int ans = len;
        if(word1.equals(word2)){
            int prev = -1;
            for(int i = 0;i<len;i++){
                String word = wordsDict[i];
                if(word.equals(word1)){
                    if(prev >= 0){
                        ans = Math.min(ans,i - prev);
                    }
                    prev = i;
                }
            }
        }else{
            int index1 = -1,index2 = -1;
            for(int i = 0;i<len;i++){
                String word = wordsDict[i];
                if(word.equals(word1)){
                    index1 = i;
                }else if(word.equals(word2)){
                    index2 = i;
                }
                if(index1 >= 0 && index2 >= 0){
                    ans = Math.min(ans, Math.abs(index1 - index2));
                }
            }
        }
        return ans;
    }
}