题目介绍

设计一个简化版的推特(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", "postTweet", "getNewsFeed", "follow", "postTweet", "getNewsFeed", "unfollow", "getNewsFeed"] [[], [1, 5], [1], [1, 2], [2, 6], [1], [1, 2], [1]] 输出 [null, null, [5], null, null, [6, 5], null, [5]]

解释 Twitter twitter = new Twitter(); twitter.postTweet(1, 5); // 用户 1 发送了一条新推文 (用户 id = 1, 推文 id = 5) twitter.getNewsFeed(1); // 用户 1 的获取推文应当返回一个列表，其中包含一个 id 为 5 的推文 twitter.follow(1, 2); // 用户 1 关注了用户 2 twitter.postTweet(2, 6); // 用户 2 发送了一个新推文 (推文 id = 6) twitter.getNewsFeed(1); // 用户 1 的获取推文应当返回一个列表，其中包含两个推文，id 分别为 -> [6, 5] 。推文 id 6 应当在推文 id 5 之前，因为它是在 5 之后发送的 twitter.unfollow(1, 2); // 用户 1 取消关注了用户 2 twitter.getNewsFeed(1); // 用户 1 获取推文应当返回一个列表，其中包含一个 id 为 5 的推文。因为用户 1 已经不再关注用户 2

提示：

1 <= userId, followerId, followeeId <= 500 0 <= tweetId <= 104 所有推特的 ID 都互不相同 postTweet、getNewsFeed、follow 和 unfollow 方法最多调用 3 * 104 次

来源：力扣（LeetCode）链接：leetcode-cn.com/problems/de… 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

需要维护一个推文对象和一个用户关注对象
用户新增推文的时候，该用户没用推文，先初始化为数组，有就直接放到对应用户的推文数组中，用户关注也是同理
获取推文先获取用户本身的推文在和获取关注用户的推文，合并并按时间排序，取出最新的10条数组

var Twitter = function() {
    this.date = 0
    this.tweetIdsMap = {}
    this.followMap = {}
};

/** 
 * @param {number} userId 
 * @param {number} tweetId
 * @return {void}
 */
Twitter.prototype.postTweet = function(userId, tweetId) {
    !this.tweetIdsMap[userId] && (this.tweetIdsMap[userId] = [])
    this.tweetIdsMap[userId].push({
        date: this.date++,
        tweetId
    })
};

/** 
 * @param {number} userId
 * @return {number[]}
 */
Twitter.prototype.getNewsFeed = function(userId) {
    let followArr = []
    let followUser = this.followMap[userId] || []
    for (let i = 0; i < followUser.length; i++) {
       followArr = [...followArr, ...(this.tweetIdsMap[followUser[i]] || [])]
    }
    let dataArr = [...(this.tweetIdsMap[userId] || []), ...followArr].sort((a, b) => b.date - a.date)
    let arr = []
    let i = 0
    while(i !== 10 && dataArr[i]) {
        arr.push(dataArr[i].tweetId)
        i++
    }
   return arr
};

/** 
 * @param {number} followerId 
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.follow = function(followerId, followeeId) {
    !this.followMap[followerId] && (this.followMap[followerId] = [])
    if(this.followMap[followerId].find(item => item === followeeId)) return
    this.followMap[followerId].push(followeeId)
};

/** 
 * @param {number} followerId 
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.unfollow = function(followerId, followeeId) {
    // [].findIndex
    !this.followMap[followerId] && (this.followMap[followerId] = [])
    // console.log(this.followMap[followerId], '=====')
    this.followMap[followerId].splice(this.followMap[followerId].findIndex(item => followeeId === item), 1)
};

/**
 * Your Twitter object will be instantiated and called as such:
 * var obj = new Twitter()
 * obj.postTweet(userId,tweetId)
 * var param_2 = obj.getNewsFeed(userId)
 * obj.follow(followerId,followeeId)
 * obj.unfollow(followerId,followeeId)
 */

[路飞]_355. 设计推特

题目介绍

解题思路