首页 > 试题广场 >

小红书推荐算法

[编程题]小红书推荐算法

热度指数：1018 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

我们假定一个用户搜索的关键词是他感兴趣的，现在请你基于这个前提设计一个小红书购物的推荐算法。
该算法的核心思想如下：首先给定一个商品清单，其中有每个商品所包含的关键词属性，然后给出用户最近搜索过的一些关键词，请你将包含用户搜索过的更多关键词的商品排在用户目录的前面。
对于包含关键词数量相同的商品，我们按默认顺序排序，也就是说按输入给定的顺序优先级。

输入描述:

第一行输入一个正整数，代表商品数量、用户搜索的关键词数量。
第二行输入个互不相同的、仅由小写字母组成的字符串，代表用户搜索过的关键词。
接下来的行，每两行描述一个商品。
第一行输入一个仅由小写字母组成的字符串和一个正整数，代表商品的名称和商品包含的关键词属性数量。
第二行输入个互不相同的、仅由小写字母组成的字符串，代表每个商品的属性。

所有的之和不超过
保证所有字符串长度不超过 。所有商品的名称互不相同。

输出描述:

输出行，每行一个字符串，代表用户主页中显示的商品名称。

示例1

输入

2 5
red book game music sigma
mozart 3
book classic music
arcaea 4
red music game hard

输出

arcaea
mozart

说明

arcaea 这个商品包含了用户搜索的 3 个关键词，而 mozart 只包含了 2 个。

算法知识视频讲解

Java

牛客199397335号

import java.util.*; 

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 商品数量
        int itemCount = in.nextInt();
        // 关键词数量
        int wordCount = in.nextInt();
        // 用哈希集方便判断是否未关键词
        Set<String> set = new HashSet<>();
        for (int i = 0; i < wordCount; i++) {
            set.add(in.next());
        }
        // 保存商品名称
        String[] items = new String[itemCount];
        // 0存储商品下标，1存储商品关键词数量
        int[][] arr = new int[itemCount][2];
        // 统计商品对应的关键词数量
        for (int i = 0; i < itemCount; i++) {
            items[i] = in.next();
            arr[i][0] = i;
            int count = in.nextInt();
            for (int j = 0; j < count; j++) {
                String word = in.next();
                if (set.contains(word)) {
                    arr[i][1]++;
                }
            }
        }
        // 按照商品关键词数量排序
        Arrays.sort(arr, (a, b) -> b[1] - a[1]);
        for (int i = 0; i < itemCount; i++) {
            System.out.println(items[arr[i][0]]);
        }
    }
}

编辑于 2025-05-25 15:41:49 回复(0)

Java

能干的铁锤在度假

import java.util.Scanner;
import java.util.*;

public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int productNum = in.nextInt();
int wordsNum = in.nextInt();
in.nextLine();
Set<String> wordsDict = new HashSet<>();
for (int i = 0; i < wordsNum; i++) {
String temp = in.next();
wordsDict.add(temp);
}
in.nextLine();
List<Product> productList = new ArrayList<>();
for (int i = 0; i < productNum; i++) {
String name = in.next();
int num = in.nextInt();
in.nextLine();
List<String> words = new ArrayList<>();
Collections.addAll(words, in.nextLine().split(" "));
productList.add(new Product(name, words, i));
}
for (Product product: productList){
for (String word: product.words){
if (wordsDict.contains(word)) {
product.sameWordsNum++;
}
}

}

//排序实现，先比较product包含的关键词数量，再比较输入顺序

Collections.sort(productList, (p1, p2) -> {
if (p1.sameWordsNum != p2.sameWordsNum) {
return p2.sameWordsNum - p1.sameWordsNum;
}
else{
return p1.priority - p2.priority;
}
});
for (Product product: productList){
System.out.println(product.name);
}
}
}
class Product {
String name;
List<String> words;
int sameWordsNum = 0;
int priority;
Product(String name, List<String> words, int priority) {
this.name = name;
this.words = words;
this.priority = priority;
}
}

发表于 2025-03-27 20:59:09 回复(1)

Java

爱熬夜的奶酪

刚开始用PriorityQueue写超时，原因是每次加入新节点都会排序，太耗时了。后面换List统一add最后排序，一次O(NlogN)即可，运行时间500ms。

import java.io.*;
import java.util.*;

public class Main {
    static class Product {
        String name;
        int count;
        int order;
        
        public Product(String name, int count, int order) {
            this.name = name;
            this.count = count;
            this.order = order;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] nq = br.readLine().split(" ");
        int n = Integer.parseInt(nq[0]);
        int q = Integer.parseInt(nq[1]);
        
        Set<String> keywordSet = new HashSet<>(Arrays.asList(br.readLine().split(" ")));
        
        List<Product> products = new ArrayList<>(n);
        for (int i = 0; i < n; i++) {
            String[] parts = br.readLine().split(" ");
            String name = parts[0];
            int m = Integer.parseInt(parts[1]);
            
            String[] keywords = br.readLine().split(" ");
            int count = 0;
            for (String kw : keywords) {
                if (keywordSet.contains(kw)) count++;
            }
            
            products.add(new Product(name, count, i));
        }
        
        products.sort((a, b) -> {
            if (a.count != b.count) {
                return Integer.compare(b.count, a.count);
            } else {
                return Integer.compare(a.order, b.order);
            }
        });
        
        StringBuilder sb = new StringBuilder();
        for (Product p : products) {
            sb.append(p.name).append('\n');
        }
        System.out.print(sb);
    }
}

发表于 2025-03-19 22:54:23 回复(1)

提交观点

问题信息

哈希排序

难度：

3条回答 42收藏 490浏览

小红书推荐算法

输入描述:

输出描述:

输入

输出

说明

问题信息

热门推荐

相关试题