【leetcode】奖励最顶尖的 K 名学生

107 阅读3分钟

题目

给你两个字符串数组 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);
  }

再提交,通过测试了,但是耗时有点久

image.png 再来分析下,函数里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

image.png