LeetCode第164场周赛：5273. 搜索推荐系统（set+map瞎搞）

给你一个产品数组 products 和一个字符串 searchWord ，products  数组中每个产品都是一个字符串。

请你设计一个推荐系统，在依次输入单词 searchWord 的每一个字母后，推荐 products 数组中前缀与 searchWord 相同的最多三个产品。如果前缀相同的可推荐产品超过三个，请按字典序返回最小的三个。

请你以二维列表的形式，返回在输入 searchWord 每个字母后相应的推荐产品的列表。

 

示例 1：

输入：products = ["mobile","mouse","moneypot","monitor","mousepad"], searchWord = "mouse"
输出：[
["mobile","moneypot","monitor"],
["mobile","moneypot","monitor"],
["mouse","mousepad"],
["mouse","mousepad"],
["mouse","mousepad"]
]
解释：按字典序排序后的产品列表是 ["mobile","moneypot","monitor","mouse","mousepad"]
输入 m 和 mo，由于所有产品的前缀都相同，所以系统返回字典序最小的三个产品 ["mobile","moneypot","monitor"]
输入 mou， mous 和 mouse 后系统都返回 ["mouse","mousepad"]
示例 2：

输入：products = ["havana"], searchWord = "havana"
输出：[["havana"],["havana"],["havana"],["havana"],["havana"],["havana"]]
示例 3：

输入：products = ["bags","baggage","banner","box","cloths"], searchWord = "bags"
输出：[["baggage","bags","banner"],["baggage","bags","banner"],["baggage","bags"],["bags"]]
示例 4：

输入：products = ["havana"], searchWord = "tatiana"
输出：[[],[],[],[],[],[],[]]
 

提示：

1 <= products.length <= 1000
1 <= Σ products[i].length <= 2 * 10^4
products[i] 中所有的字符都是小写英文字母。
1 <= searchWord.length <= 1000
searchWord 中所有字符都是小写英文字母。

思路：我们可以每次将前缀与待匹配串相同的串全部加进set里，这样我们拿的头三个一定是字典序最小的三个，题目一个坑点是有可能会有重复出现的字符串，因此我们可以考虑用map存一下每个字符串出现的次数。

class Solution {
    public List<List<String>> suggestedProducts(String[] products, String searchWord) {
        
    	Set<String> st=new TreeSet<>((o1,o2)->o1.compareTo(o2));
    	Map<String,Integer> map=new HashMap<>();
    	List<String> res=new ArrayList<>();
        List<List<String>> ans=new ArrayList<>();
        
        for(int i=0;i<searchWord.length();i++) {
        	st.clear();
        	res.clear();
        	map.clear();
        	for(int j=0;j<products.length;j++) {
        		if(products[j].length()>i && products[j].substring(0,i+1).equals(searchWord.substring(0,i+1))) {
        			st.add(products[j]);
        			map.put(products[j], map.getOrDefault(products[j], 0)+1);
        		}
        	}
        	//System.out.println
        	int num=0;
        	Iterator it=st.iterator();
        	while(it.hasNext() && num<3) {
        		String s=(String)it.next();
        		for(int j=0;j<map.get(s) && num<3;j++) {
        			num++;
        			res.add(s);
        		}
        	}
        	ans.add(new ArrayList<>(res));
        }
        
        return ans;
    }
}