题目
给你两个字符串数组 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] 。
我的思路
定义一个Map,先根据report计算出总分,总分作为key,总分对应的学生id数组(升序)作为value,然后根据总分降序将对应的value push进结果里,最后根据k切割数组返回
题解
var topStudents = function(positive_feedback, negative_feedback, report, student_id, k) {
const scoreMap = new Map();
const getTotal = feedback => {
return feedback.reduce((pre, cur) => {
if (positive_feedback.includes(cur)) pre += 3;
if (negative_feedback.includes(cur)) pre -= 1;
return pre;
}, 0);
}
const getSortedArr = (arr,order) => {
return arr.sort((a, b) => {
return order === 'desc' ? b - a : a - b;
});
}
for(let i = 0; i < report.length; i++) {
const comment = report[i].split(' ');
const total = getTotal(comment);
if(scoreMap.has(total)) {
scoreMap.set(total, getSortedArr([...(scoreMap.get(total) || []), student_id[i]],'asc'));
}else {
scoreMap.set(total, [student_id[i]]);
}
}
const keys = getSortedArr(Array.from(scoreMap.keys()),'desc');
let result = [];
keys.forEach(item=> {
if(result.length >= k) {
return;
}
result.push(...scoreMap.get(item));
})
return result.slice(0,k);
}
运行,通过测试,然后提交,发现跑测试用例到最后几个大数据的时候超时了,应该是复杂度过高,再来分析下这个函数,发现在获取总分数时,都需要在正面数组和负面数组里查找,我们都知道Map查找适用于已经知道键的情况下,可以提供更快的查找速度,因此可以把正面和负面都塞到Map里
const scoreFeedMap = new Map();
positive_feedback.forEach((f) => scoreFeedMap.set(f,3));
negative_feedback.forEach((f) => scoreFeedMap.set(f,-1));
const getTotal = feedback => {
return feedback.reduce((pre, cur) => {
if(scoreFeedMap.get(cur)) pre+=scoreFeedMap.get(cur);
return pre;
}, 0);
}
再提交,通过测试了,但是耗时有点久
再来分析下,函数里sort次数有些多,在遍历report时和获取keys时均进行了sort,有没有办法降低下?让我们来看下如下代码:
const entries = new Map([
[10, 10],
[5, 5],
[-2,-2]
]);
const obj = Object.fromEntries(entries);
console.log(obj);
// Expected output: Object { 5: 5, 10: 10, -2: -2 }
对象自带天然排序,这样可以减少sort次数,但需要注意负数在对象自带排序里是最大的,所以初始化总分数时定为100
var topStudents = function(positive_feedback, negative_feedback, report, student_id, k) {
const scoreMap = new Map();
const scoreFeedMap = new Map();
positive_feedback.forEach((f) => scoreFeedMap.set(f,3));
negative_feedback.forEach((f) => scoreFeedMap.set(f,-1));
const getTotal = feedback => {
return feedback.reduce((pre, cur) => {
if(scoreFeedMap.has(cur)) pre+=scoreFeedMap.get(cur);
return pre;
}, 100);
}
const getSortedArr = (arr,order) => {
return arr.sort((a, b) => {
return order === 'desc' ? b - a : a - b;
});
}
for(let i = 0; i < report.length; i++) {
const comment = report[i].split(' ');
const total = getTotal(comment);
let list = scoreMap.get(total);
if(list) {
list.push(student_id[i])
}else {
list = [student_id[i]]
}
scoreMap.set(total,list)
}
const scoreObj = Object.fromEntries(scoreMap);
const keys = Object.keys(scoreObj);
let result = [];
for(let i = keys.length - 1; i >= 0; i--) {
if(result.length >= k) break;
const key = keys[i];
const studentScoreList = getSortedArr(scoreObj[key],'asc');
result.push(...studentScoreList)
}
return result.slice(0,k);
}
利用对象减少sort,同时将分数对应的学生id数组排序调整到最后面的keys遍历,利用k可提前跳出遍历,这样又能减少sort次数,最终提交耗时降低到196