题解 | 小红书推荐算法

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        
        int n = in.nextInt();
        int q = in.nextInt();
        // 存储关键词
        Set<String> set = new HashSet<>();
        for(int i = 0; i < q; i++){
            set.add(in.next());
        }
        // 存储商品名称
        String[] items = new String[n];
        // 一共 n 个商品，0 位置存储商品名称索引，1 位置存储关键词数量
        int[][] arrs = new int[n][2];
        for(int i = 0; i < n; i++){
            items[i] = in.next();
            // 记录当前商品名称的索引值
            arrs[i][0] = i;
            int count = in.nextInt();
            for(int j = 0; j < count; j++){
                String keyWord = in.next();
                if(set.contains(keyWord)){
                    arrs[i][1]++;
                }
            }
        }
        Arrays.sort(arrs,(a,b) -> (b[1] - a[1]));

        for(int i = 0; i < n; i++){
            System.out.println(items[arrs[i][0]]);
        }
    }
}