开启我的LeetCode刷题日记：355. 设计推特Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

编程世界总是离不了算法

最近在看框架源码时，会有很多算法的实现逻辑，有时候会感到吃力

于是决定蹭着假期，加强算法和数据结构相关的知识

那怎么提升呢？

其实我知道算法这东西没有捷径，多写多练才能提升，于是我开启我的LeetCode刷题之旅

第一阶段目标是：200道，每天1到2篇

为了不乱，本系列文章目录分为三部分：

今日题目：xxx
我的思路
代码实现

今天题目：355. 设计推特

设计一个简化版的推特(Twitter)，可以让用户实现发送推文，关注/取消关注其他用户，能够看见关注人（包括自己）的最近 10 条推文。

实现 Twitter 类：

Twitter() 初始化简易版推特对象
void postTweet(int userId, int tweetId) 根据给定的 tweetId 和 userId 创建一条新推文。每次调用此函数都会使用一个不同的 tweetId 。
List getNewsFeed(int userId) 检索当前用户新闻推送中最近 10 条推文的 ID 。新闻推送中的每一项都必须是由用户关注的人或者是用户自己发布的推文。推文必须按照时间顺序由最近到最远排序。
void follow(int followerId, int followeeId) ID 为 followerId 的用户开始关注 ID 为 followeeId 的用户。
void unfollow(int followerId, int followeeId) ID 为 followerId 的用户不再关注 ID 为 followeeId 的用户。

我的思路

在 Twitter 函数中定义好数据结构 followMap：用户关注列表，用 Set 数据类型不需要去处理重复数据，取消关注（从列表删除）也会更方便； postMap：用户推文列表； latestPostId：推文的自增id，用于后续获取推文列表时排序；

在 postTweet 函数中，将新增的推文 { tweetId, postTime } 放到列表的最前面，并确保 latestPostId 自增；

在 follow 函数中，先检查 followMap 是否已存在 followerId 数据，若已存在，直接 add(followeeId), 若不存在，新增 new Set([followeeId])；在 unfollow 函数中，直接检查是否存在 followMap[followerId] 列表，若存在直接delete(followeeId)；在 getNewsFeed 函数中，因为要取用户和用户关注的用户的最新 10 条推文，所以只需要把这些用户的前10条推文取出来，再根据postTime去排序，然后取最新10条推文。

代码实现

/**
 * Initialize your data structure here.
 */
var Twitter = function() {
  this.followMap = {}
  this.postMap = new Map()
  this.latestPostId = 0
}

/**
 * Compose a new tweet.
 * @param {number} userId
 * @param {number} tweetId
 * @return {void}
 */
Twitter.prototype.postTweet = function(userId, tweetId) {
  const postTime = this.latestPostId++
  let tweeList = [{ tweetId, postTime }]
  if (this.postMap.has(userId)) {
    tweeList = tweeList.concat(this.postMap.get(userId))
  }
  this.postMap.set(userId, tweeList)
}

/**
 * Retrieve the 10 most recent tweet ids in the user's news feed. Each item in the news feed must be posted by users who the user followed or by the user herself. Tweets must be ordered from most recent to least recent.
 * @param {number} userId
 * @return {number[]}
 */
Twitter.prototype.getNewsFeed = function(userId) {
  const followeeIdList = this.followMap[userId] ? [...this.followMap[userId]] : []
  const tweeList = []
  const userIds = [...new Set(followeeIdList.concat([userId]))]
  userIds.forEach(uid => {
    if (this.postMap.has(uid)) {
      tweeList.push(...this.postMap.get(uid).slice(0, 10))
    }
  })
  tweeList.sort((a, b) => b.postTime - a.postTime)

  return tweeList.slice(0, 10).map(item => item.tweetId)
}

/**
 * Follower follows a followee. If the operation is invalid, it should be a no-op.
 * @param {number} followerId
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.follow = function(followerId, followeeId) {
  if (this.followMap[followerId]) {
    this.followMap[followerId].add(followeeId)
  } else {
    this.followMap[followerId] = new Set([followeeId])
  }
}

/**
 * Follower unfollows a followee. If the operation is invalid, it should be a no-op.
 * @param {number} followerId
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.unfollow = function(followerId, followeeId) {
  if (this.followMap[followerId]) {
    this.followMap[followerId].delete(followeeId)
  }
}

总结

实现方式其实有很多，这里仅供参考~

由于刚开始刷题，也不知道从哪里刷好，如果前辈们有好的建议，希望不吝赐教，感谢🌹