设计Twitter

    至于 Twitter 的什么功能跟算法有关系，等我们描述一下题目要求就知道了。

    Twitter 和微博功能差不多，我们主要要实现这样几个 API：

    举个具体的例子，方便大家理解 API 的具体用法：

    3. twitter.postTweet(1, 5);
    4. // 用户 1 发送了一条新推文 5
    6. twitter.getNewsFeed(1);
    7. // return [5]，因为自己是关注自己的
    9. twitter.follow(1, 2);
    10. // 用户 1 关注了用户 2
    12. twitter.postTweet(2, 6);
    13. // 用户2发送了一个新推文 (id = 6)
    15. twitter.getNewsFeed(1);
    16. // return [6, 5]
    17. // 解释：用户 1 关注了自己和用户 2，所以返回他们的最近推文
    18. // 而且 6 必须在 5 之前，因为 6 是最近发送的
    20. twitter.unfollow(1, 2);
    21. // 用户 1 取消关注了用户 2
    23. twitter.getNewsFeed(1);
    24. // return [5]

    这个场景在我们的现实生活中非常常见。拿朋友圈举例，比如我刚加到女神的微信，然后我去刷新一下我的朋友圈动态，那么女神的动态就会出现在我的动态列表，而且会和其他动态按时间排好序。只不过 Twitter 是单向关注，微信好友相当于双向关注。除非，被屏蔽…

    这几个 API 中大部分都很好实现，最核心的功能难点应该是 getNewsFeed，因为返回的结果必须在时间上有序，但问题是用户的关注是动态变化的，怎么办？

    这里就涉及到算法了：如果我们把每个用户各自的推文存储在链表里，每个链表节点存储文章 id 和一个时间戳 time（记录发帖时间以便比较），而且这个链表是按 time 有序的，那么如果某个用户关注了 k 个用户，我们就可以用合并 k 个有序链表的算法合并出有序的推文列表，正确地 getNewsFeed 了！

    具体的算法等会讲解。不过，就算我们掌握了算法，应该如何编程表示用户 user 和推文动态 tweet 才能把算法流畅地用出来呢？这就涉及简单的面向对象设计了，下面我们来由浅入深，一步一步进行设计。

    根据刚才的分析，我们需要一个 User 类，储存 user 信息，还需要一个 Tweet 类，储存推文信息，并且要作为链表的节点。所以我们先搭建一下整体的框架：

    之所以要把 Tweet 和 User 类放到 Twitter 类里面，是因为 Tweet 类必须要用到一个全局时间戳 timestamp，而 User 类又需要用到 Tweet 类记录用户发送的推文，所以它们都作为内部类。不过为了清晰和简洁，下文会把每个内部类和 API 方法单独拿出来实现。

    1、Tweet 类的实现

    1. class Tweet {
    2.     private int id;
    3.     private int time;
    4.     private Tweet next;
    6.     // 需要传入推文内容（id）和发文时间
    7.     public Tweet(int id, int time) {
    8.         this.id = id;
    9.         this.time = time;
    10.     }

    2、User 类的实现

    我们根据实际场景想一想，一个用户需要存储的信息有 userId，关注列表，以及该用户发过的推文列表。其中关注列表应该用集合（Hash Set）这种数据结构来存，因为不能重复，而且需要快速查找；推文列表应该由链表这种数据结构储存，以便于进行有序合并的操作。画个图理解一下：

    User

    除此之外，根据面向对象的设计原则，「关注」「取关」和「发文」应该是 User 的行为，况且关注列表和推文列表也存储在 User 类中，所以我们也应该给 User 添加 follow，unfollow 和 post 这几个方法：

    3、几个 API 方法的实现

    1. class Twitter {
    2.     private static int timestamp = 0;
    3.     private static class Tweet {...}
    4.     private static class User {...}
    6.     // 我们需要一个映射将 userId 和 User 对象对应起来
    7.     private HashMap<Integer, User> userMap = new HashMap<>();
    9.     /** user 发表一条 tweet 动态 */
    10.     public void postTweet(int userId, int tweetId) {
    11.         // 若 userId 不存在，则新建
    12.         if (!userMap.containsKey(userId))
    13.             userMap.put(userId, new User(userId));
    14.         User u = userMap.get(userId);
    15.         u.post(tweetId);
    16.     }
    18.     /** follower 关注 followee */
    19.     public void follow(int followerId, int followeeId) {
    20.         // 若 follower 不存在，则新建
    21.         if(!userMap.containsKey(followerId)){
    22.             User u = new User(followerId);
    23.             userMap.put(followerId, u);
    24.         }
    25.         // 若 followee 不存在，则新建
    26.         if(!userMap.containsKey(followeeId)){
    27.             User u = new User(followeeId);
    28.             userMap.put(followeeId, u);
    29.         }
    30.         userMap.get(followerId).follow(followeeId);
    31.     }
    33.     /** follower 取关 followee，如果 Id 不存在则什么都不做 */
    34.     public void unfollow(int followerId, int followeeId) {
    35.         if (userMap.containsKey(followerId)) {
    37.         }
    38.     }
    40.     /** 返回该 user 关注的人（包括他自己）最近的动态 id，
    41.     最多 10 条，而且这些动态必须按从新到旧的时间线顺序排列。*/
    42.     public List<Integer> getNewsFeed(int userId) {
    43.         // 需要理解算法，见下文
    44.     }
    45. }

    实现合并 k 个有序链表的算法需要用到优先级队列（Priority Queue），这种数据结构是「二叉堆」最重要的应用，你可以理解为它可以对插入的元素自动排序。乱序的元素插入其中就被放到了正确的位置，可以按照从小到大（或从大到小）有序地取出元素。

    借助这种牛逼的数据结构支持，我们就很容易实现这个核心功能了。注意我们把优先级队列设为按 time 属性从大到小降序排列，因为 time 越大意味着时间越近，应该排在前面：

    1. public List<Integer> getNewsFeed(int userId) {
    2.     List<Integer> res = new ArrayList<>();
    3.     if (!userMap.containsKey(userId)) return res;
    4.     // 关注列表的用户 Id
    5.     Set<Integer> users = userMap.get(userId).followed;
    6.     // 自动通过 time 属性从大到小排序，容量为 users 的大小
    7.     PriorityQueue<Tweet> pq = 
    8.         new PriorityQueue<>(users.size(), (a, b)->(b.time - a.time));
    10.     // 先将所有链表头节点插入优先级队列
    11.     for (int id : users) {
    12.         Tweet twt = userMap.get(id).head;
    13.         if (twt == null) continue;
    14.         pq.add(twt);
    15.     }
    17.     while (!pq.isEmpty()) {
    18.         // 最多返回 10 条就够了
    19.         if (res.size() == 10) break;
    20.         // 弹出 time 值最大的（最近发表的）
    21.         Tweet twt = pq.poll();
    22.         res.add(twt.id);
    23.         // 将下一篇 Tweet 插入进行排序
    24.         if (twt.next != null) 
    25.             pq.add(twt.next);
    26.     }
    27.     return res;

    这个过程是这样的，下面是我制作的一个 GIF 图描述合并链表的过程。假设有三个 Tweet 链表按 time 属性降序排列，我们把他们降序合并添加到 res 中。注意图中链表节点中的数字是 time 属性，不是 id 属性：

    本文运用简单的面向对象技巧和合并 k 个有序链表的算法设计了一套简化的时间线功能，这个功能其实广泛地运用在许多社交应用中。

    我们先合理地设计出 User 和 Tweet 两个类，然后基于这个设计之上运用算法解决了最重要的一个功能。可见实际应用中的算法并不是孤立存在的，需要和其他知识混合运用，才能发挥实际价值。

    当然，实际应用中的社交 App 数据量是巨大的，考虑到数据库的读写性能，我们的设计可能承受不住流量压力，还是有些太简化了。而且实际的应用都是一个极其庞大的工程，比如下图，是 Twitter 这样的社交网站大致的系统结构：

    design

    我们解决的问题应该只能算 Timeline Service 模块的一小部分，功能越多，系统的复杂性可能是指数级增长的。所以说合理的顶层设计十分重要，其作用是远超某一个算法的。

    最后，Github 上有一个优秀的开源项目叫 system-design-primer，专门收集了很多大型系统设计的案例和解析，而且有中文版本，上面这个图也出自该项目。对系统设计感兴趣的读者可以点击链接查看。

    PS：本文前两张图片和 GIF 是我第一次尝试用平板的绘图软件制作的，花了很多时间，尤其是 GIF 图，需要一帧一帧制作。如果本文内容对你有帮助，点个赞分个享，鼓励一下我呗！

    坚持原创高质量文章，致力于把算法问题讲清楚，欢迎关注我的公众号 labuladong 获取最新文章：

    目录