力扣需要vip,没有vip的朋友可以去牛客网做,一样的题
题目描述
给定一个string数组article(有单词构成)和数组中元素的个数n,同时给定数组中的两个单词x和y。请返回这两个单词的最短距离(比如两个单词分别在第1和第3个位置,则最短距离为2)。保证两个单词不相同且均在数组中出现,同时保证数组中单词数小于等于1000。
我的题解
import java.util.*;
public class Distance {
public int getDistance(String[] article, int n, String x, String y) {
// write code here
int len = n;
Map<String,Integer> map = new HashMap<>();
for(int i=0;i<article.length;i++){
map.put(article[i],i);
if(map.containsKey(x)&&map.containsKey(y)){
if(Math.abs(map.get(x)-map.get(y))<len){
len = Math.abs(map.get(x)-map.get(y));
}
}
}
return len;
}
}
思路:
1.因为map.containsKey(x)&&map.containsKey(y)的key表示最后一次出现的x或y,与最新出现的x或y,所以2中相减一定是目前距离最短。
2.因为map.get(x)-map.get(y)总是能返回遍历目前为止的最短距离,但不是总体遍历完后的最短距离,所以还需要继续遍历完
3.求出整体最短返回就好了