LeetCode-Shortest Word Distance II

最新推荐文章于 2024-08-08 14:17:25 发布

原创最新推荐文章于 2024-08-08 14:17:25 发布 · 576 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 3 个专栏收录

220 篇文章

订阅专栏

hashmap

15 篇文章

订阅专栏

two pointers

9 篇文章

订阅专栏

本文介绍了一种高效算法，用于计算给定字符串数组中两个特定单词出现位置之间的最短距离。通过使用哈希映射存储每个单词的所有位置，并采用双指针技术在这些位置列表中寻找最小距离，该算法能够在O(n)时间内解决问题。

因为需要多次call 所以就存成hashmap 存position list

关键是找min的时候实际上就是给两个list 找其中数字最小difference

方法就是每次更新那个小的让他变大去靠近大的（因为list是sorted 从小到大）移动数字比较小的那个的指针

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

    public int shortest(String word1, String word2) {
        int min = Integer.MAX_VALUE;
        int p1 = 0;
        int p2 = 0;
        List<Integer> list1 = map.get ( word1 );
        List<Integer> list2 = map.get ( word2 );
        while ( p1 < list1.size() && p2 < list2.size() ){
            min = Math.min( min, Math.abs ( list1.get ( p1 ) - list2.get ( p2 ) ));
            if ( list1.get ( p1 ) > list2.get ( p2 ) )
                p2 ++;
            else 
                p1 ++;
        }
        return min;
    }
}