题目描述

设计一个简化版的推特(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… 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

创建列表list储存消息tweetId,创建map，以tweetId为key保存消息体
再创建一个map，在发生关注时候，以followerId为key保存这个用户关注的列表，取消关注时从关注列表中移除对应id
返回前十条推文时，遍历推文列表，推文的userId 是自己时，或者userId在自己关注列表中时，就是我们要返回的推文
收集齐十条，提前跳出循环，否则循环结束，返回所有符合条件的id集合

代码



var Twitter = function() {
    this.idList=[];//id列表
    this.idMap=new Map();//新闻体
    this.userFollow=new Map();//关注列表

};

/** 创建一条新推文
 * @param {number} userId 用户id
 * @param {number} tweetId 推特id
 * @return {void}
 */
Twitter.prototype.postTweet = function(userId, tweetId) {
    this.idList.unshift(tweetId);//新发布的消息在上面，往前插入
    this.idMap.set(tweetId,userId);
};

/** 检索当前用户新闻推送中最近  10 条推文的 ID 
 * @param {number} userId
 * @return {number[]}
 */
Twitter.prototype.getNewsFeed = function(userId) {
    let result=[];
    for(let i=0;i<this.idList.length;i++){
        let id=this.idMap.get(this.idList[i]);//遍历新闻列表，获取每个新闻id对应的发布者userId
        let followList=this.userFollow.get(userId);
        console.log(this.userFollow)
        if(id===userId||(followList&&followList.has(id))){//新闻id的发布者userId，等于当前查询的userId，或者等于在当前查询的userId的关注列表中
          result.push(this.idList[i]);
        }
        if(result.length>=10) break;//收集满10个，跳出循环
    }
    return result;
};

/** 关注
 * @param {number} followerId 
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.follow = function(followerId, followeeId) {
    let list=this.userFollow.get(followerId)?this.userFollow.get(followerId):new Set();//用set保存id避免重复
    list.add(followeeId);
    this.userFollow.set(followerId,list);
};

/** 
 * @param {number} followerId 
 * @param {number} followeeId
 * @return {void}
 */
Twitter.prototype.unfollow = function(followerId, followeeId) {
   let list=this.userFollow.get(followerId);//获取用户关注的id列表
   if(!list) return;//没有关注列表直接返回
    list.delete(followeeId);//删除这个关注
   //再存回去
   this.userFollow.set(followerId,list);
};

/**
 * 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)
 */

[路飞]_算法_设计推特

题目描述

解题思路

代码