2512.奖励最顶尖的 K 名学生

【LetMeFly】2512.奖励最顶尖的 K 名学生:哈希表设计

力扣题目链接:https://leetcode.cn/problems/reward-top-k-students/

给你两个字符串数组 positive_feedback 和 negative_feedback ,分别包含表示正面的和负面的词汇。不会 有单词同时是正面的和负面的。

一开始,每位学生分数为 0 。每个正面的单词会给学生的分数 加 3 分,每个负面的词会给学生的分数 减  1 分。

给你 n 个学生的评语,用一个下标从 0 开始的字符串数组 report 和一个下标从 0 开始的整数数组 student_id 表示,其中 student_id[i] 表示这名学生的 ID ,这名学生的评语是 report[i] 。每名学生的 ID 互不相同

给你一个整数 k ,请你返回按照得分 从高到低 最顶尖的 k 名学生。如果有多名学生分数相同,ID 越小排名越前。

 

示例 1:

输入:positive_feedback = ["smart","brilliant","studious"], negative_feedback = ["not"], report = ["this student is studious","the student is smart"], student_id = [1,2], k = 2
输出:[1,2]
解释:
两名学生都有 1 个正面词汇,都得到 3 分,学生 1 的 ID 更小所以排名更前。

示例 2:

输入:positive_feedback = ["smart","brilliant","studious"], negative_feedback = ["not"], report = ["this student is not studious","the student is smart"], student_id = [1,2], k = 2
输出:[2,1]
解释:
- ID 为 1 的学生有 1 个正面词汇和 1 个负面词汇,所以得分为 3-1=2 分。
- ID 为 2 的学生有 1 个正面词汇,得分为 3 分。
学生 2 分数更高,所以返回 [2,1] 。

 

提示:

  • 1 <= positive_feedback.length, negative_feedback.length <= 104
  • 1 <= positive_feedback[i].length, negative_feedback[j].length <= 100
  • positive_feedback[i] 和 negative_feedback[j] 都只包含小写英文字母。
  • positive_feedback 和 negative_feedback 中不会有相同单词。
  • n == report.length == student_id.length
  • 1 <= n <= 104
  • report[i] 只包含小写英文字母和空格 ' ' 。
  • report[i] 中连续单词之间有单个空格隔开。
  • 1 <= report[i].length <= 100
  • 1 <= student_id[i] <= 109
  • student_id[i] 的值 互不相同 。
  • 1 <= k <= n

方法一:哈希表设计

对于一句评语,我们可以以空格未依据将其分为不同的单词。

如何快速判断某个单词得多少分呢?只需要预处理一下,将正负单词预先放入哈希表中,这样就能快速判断某个单词是否为“正负性词语”了。

怎么选出得分最高的$k$个$id$呢?题目中说$student_id[i]$的值互不相同,也就是说每个学生最多收到一次评语,因此我们直接将[负的得分, 学生id]放入一个数组中,最后排个序选前$k$个的$id$即可。

  • 时间复杂度$O((len(positive_feedback) + len(negative_feedback))\times L + len(student_id)\log len(student_id))$,其中$L$为一个单词的平均长度
  • 空间复杂度$O((len(positive_feedback) + len(negative_feedback))\times L + len(student_id))$

AC代码

C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution {
public:
vector<int> topStudents(vector<string>& positive_feedback, vector<string>& negative_feedback, vector<string>& report, vector<int>& student_id, int k) {
unordered_set<string> postive, negative;
for (string& s : positive_feedback) {
postive.insert(s);
}
for (string& s : negative_feedback) {
negative.insert(s);
}
vector<pair<int, int>> v;
for (int i = 0; i < report.size(); i++) {
int lastSpace = -1;
int thisScore = 0;
for (int j = 0; j <= report[i].size(); j++) {
if (j == report[i].size() || report[i][j] == ' ') { // (lastSpace, j)
string thisWord = report[i].substr(lastSpace + 1, j - lastSpace - 1);
lastSpace = j;
if (postive.count(thisWord)) {
thisScore += 3;
}
else if (negative.count(thisWord)) {
thisScore -= 1;
}
}
}
v.push_back({-thisScore, student_id[i]});
}
sort(v.begin(), v.end());
vector<int> ans;
for (int i = 0; i < k; i++) {
ans.push_back(v[i].second);
}
return ans;
}
};

Python

1
2
3
4
5
6
7
8
9
10
11
12
13
14
# from typing import List

class Solution:
def topStudents(self, positive_feedback: List[str], negative_feedback: List[str], report: List[str], student_id: List[int], k: int) -> List[int]:
mp = {}
for s in positive_feedback:
mp[s] = -3
for s in negative_feedback:
mp[s] = 1
v = []
for i in range(len(report)):
v.append([sum(mp.get(word, 0) for word in report[i].split()), student_id[i]])
v.sort()
return [t[1] for t in v[:k]]

同步发文于CSDN,原创不易,转载经作者同意后请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/133762019


2512.奖励最顶尖的 K 名学生
https://blog.letmefly.xyz/2023/10/11/LeetCode 2512.奖励最顶尖的K名学生/
作者
Tisfy
发布于
2023年10月11日
许可协议