【Leetcode】599. Minimum Index Sum of Two Lists

题目地址：

https://leetcode.com/problems/minimum-index-sum-of-two-lists/

给定两个字符串数组$A$和$B$，要求返回两个数组的交集中下标和最小的那些字符串。有可能有多个答案，所以要求返回数组。

不妨设$A$的长度小于等于$B$，接着用一个哈希表记录$A$中字符串及每个字符串下标，然后遍历$B$，若发现哈希表存了当前字符串，则判断下标和是否更小，若更小，则更新最小下标和，并清空答案，并把这个字符串加进去；若相等，则不用清空答案，直接加进去；若更大，则略过。代码如下：

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Solution {
    public String[] findRestaurant(String[] list1, String[] list2) {
        if (list1.length > list2.length) {
            String[] tmp = list1;
            list1 = list2;
            list2 = tmp;
        }
        
        // 此时list1的长度更小，记录一下每个字符串及其出现的下标
        Map<String, Integer> map = new HashMap<>();
        for (int i = 0; i < list1.length; i++) {
            map.put(list1[i], i);
        }
        
        List<String> ans = new ArrayList<>();
        int minIdxSum = Integer.MAX_VALUE;
        for (int i = 0; i < list2.length; i++) {
            if (map.containsKey(list2[i])) {
                if (map.get(list2[i]) + i < minIdxSum) {
                	// 若下标和更小，则更新最小下标和，清空答案并将当前字符串加进去
                    minIdxSum = map.get(list2[i]) + i;
                    ans.clear();
                    ans.add(list2[i]);
                } else if (map.get(list2[i]) + i == minIdxSum) {
                	// 若一样大，则直接加入
                    ans.add(list2[i]);
                }
            }
        }
        
        String[] res = new String[ans.size()];
        for (int i = 0; i < res.length; i++) {
            res[i] = ans.get(i);
        }
        
        return res;
    }
}

时间复杂度$O((m+n)l)$，$m$和$n$是两个数组的长度，$l$是最长字符串长度（事实上put和get需要$O(l)$的时间），空间$O(\min (m,n))$。