宝宝也能看懂的 leetcode 周赛 - 170 - 3

1311. 获取你好友已观看的视频

Hi 你们好，我是张小猪。欢迎来到『宝宝也能看懂』系列之 leetcode 周赛题解。

这里是第 170 期的第 3 题，也是题目列表中的第 1311 题 -- 『获取你好友已观看的视频』

题目描述

有 n 我的，每一个人都有一个 0 到 n-1 的惟一 id。

给你数组 watchedVideos 和 friends，其中 watchedVideos[i] 和 friends[i] 分别表示 id = i 的人观看过的视频列表和他的好友列表。

Level 1 的视频包含全部你好友观看过的视频，level 2 的视频包含全部你好友的好友观看过的视频，以此类推。通常的，Level 为 k 的视频包含全部从你出发，最短距离为 k 的好友观看过的视频。

给定你的 id 和一个 level 值，请你找出全部指定 level 的视频，并将它们按观看频率升序返回。若是有频率相同的视频，请将它们按名字字典序从小到大排列。

示例 1：

输入：watchedVideos = [["A","B"],["C"],["B","C"],["D"]], friends = [[1,2],[0,3],[0,3],[1,2]], id = 0, level = 1
输出：["B","C"]
解释：
你的 id 为 0 ，你的朋友包括：
id 为 1 -> watchedVideos = ["C"]
id 为 2 -> watchedVideos = ["B","C"]
你朋友观看过视频的频率为：
B -> 1
C -> 2

示例 2：

输入：watchedVideos = [["A","B"],["C"],["B","C"],["D"]], friends = [[1,2],[0,3],[0,3],[1,2]], id = 0, level = 2
输出：["D"]
解释：
你的 id 为 0 ，你朋友的朋友只有一我的，他的 id 为 3 。

提示：

• n == watchedVideos.length == friends.length
• 2 <= n <= 100
• 1 <= watchedVideos[i].length <= 100
• 1 <= watchedVideos[i][j].length <= 8
• 0 <= friends[i].length < n
• 0 <= friends[i][j] < n
• 0 <= id < n
• 1 <= level < n
• 若是 friends[i] 包含 j ，那么 friends[j] 包含 i

官方难度

MEDIUM

解决思路

这道题一眼看上去还挺吓人，一堆参数，很长的限制条件，连例子都好长。然而仔细想一下就会发现，其实都是纸脑抚，哼，小猪才不会被你骗到呢，略略略 >.<

题目的意思就是根据给定的初始 id，去寻找他第 level 轮的朋友们，从而获得朋友们看过的视频列表。最终把这个列表进行排序返回。那么总的来讲就分为两部分，第一部分就是找到最终的目标朋友列表，第二部分就是把视频列表按要求进行排序返回。

找到目标朋友列表

咱们先看第一部分，这里的流程其实很是清晰：

1. 根据初始 id 寻找到朋友列表。
2. 根据当前的朋友列表继续扩展开寻找下一轮的朋友列表。
3. 直到知足第 level 轮便可终止。

看到这个流程，结合咱们前几期的题解，相信小伙们应该很容易能想到一个思路，那就是广度优先遍历。要是没想到的话，哼，罚你关注个人公众号，略略略 >.<

关于广度优先遍历，我后续关于算法和数据结构的新坑会详细说明。这里能够先结合思路和下面的代码，或者翻翻我以前的题解，应该就能明白。不过这里有个小坑小伙伴们须要注意，那就是第 level 轮的好友指的是最短距离为 level 的好友。例如你的好友是 A、B、C，而 A 的好友是 C、D，B 的好友是 E，C 的好友是 A、F。那么这里获得的你的第一轮好友列表是 A、B、C，而第二轮好友列表是 D、E、F，并无 A、C，由于他们已经在第一轮中出现了，在第二轮中并非最短距离了。

这里给出一个比较容易理解的模板化的代码，固然咱们为了处理最短距离问题，加入一个 visited 用以进行标识。

const visited = new Uint8Array(friends.length);
let queue = [id];
visited[id] = 1;
while (level--) {
  const next = [];
  for (const id of queue) {
    for (const newId of friends[id]) {
      if (visited[newId] === 0) { next.push(newId); visited[newId] = 1; }
    }
  }
  queue = next;
}

排序视频列表

题目中对于这个排序规则作了几点要求：

• 根据视频被观看次数，进行升序排列。
• 对于观看次数相同的视频，按照名称字母序进行排列。

那么咱们这里的处理流程就比较明晰了：

1. 基于目标好友列表统计出各个视频观看的次数。
2. 基于次数和字母序进行排序。

前者能够基于 Map 很容易的实现，后者因为 JS 的数组 sort 能够支持自定义的排序函数，因此也很容易实现。具体代码以下：

// 统计观看次数
const freq = new Map();
for (const id of queue) {
  for (const video of watchedVideos[id]) {
    freq.set(video, freq.has(video) ? freq.get(video) + 1 : 1);
  }
}

// 自定义排序函数
(a, b) => a[1] === b[1] ? (a[0] > b[0] ? 1 : -1) : a[1] - b[1];

最终代码

结合上述两部份内容，咱们很容易能获得相似下面的最终代码：

const watchedVideosByFriends = (watchedVideos, friends, id, level) => {
  const visited = new Uint8Array(friends.length);
  let queue = [id];
  visited[id] = 1;
  while (level--) {
    const next = [];
    for (const id of queue) {
      for (const newId of friends[id]) {
        if (visited[newId] === 0) { next.push(newId); visited[newId] = 1; }
      }
    }
    queue = next;
  }

  const freq = new Map();
  for (const id of queue) {
    for (const video of watchedVideos[id]) {
      freq.set(video, freq.has(video) ? freq.get(video) + 1 : 1);
    }
  }

  return Array.from(freq.entries()).sort((a, b) => {
    return a[1] === b[1] ? (a[0] > b[0] ? 1 : -1) : a[1] - b[1]
  }).map(item => item[0]);
};

这段代码跑出了 128ms，暂时 beats 100%。

总结

这道题目总体比较直白，没有太多能够分析的部分。不过因为它的流程其实比较明显的能够拆分红独立的两部分分别进行实现，这样每一部分的逻辑和实现都会更加清晰和容易。对应的，咱们在实际工做开发中，其实把 feature 和 function 进行合理的拆分也是会颇有利于整理和维护的。

相关连接

• Weekly Contest 170 题目列表
• 系列题解 repo
• 个人 segmentfault 专栏
• 个人知乎专栏