二刷355

Author: diguage

docs/0355-design-twitter.adoc

@@ -1,16 +1,15 @@
 [#0355-design-twitter]
 = 355. 设计推特
 
-https://leetcode.cn/problems/design-twitter/[LeetCode - 355. 设计推特 ^]
+https://leetcode.cn/problems/design-twitter/[LeetCode - 355. 设计推特^]
 
 设计一个简化版的推特(Twitter)，可以让用户实现发送推文，关注/取消关注其他用户，能够看见关注人（包括自己）的最近 `10` 条推文。
 
 实现 `Twitter` 类：
 
 * `Twitter()` 初始化简易版推特对象
 * `void postTweet(int userId, int tweetId)` 根据给定的 `tweetId` 和 `userId` 创建一条新推文。每次调用此函数都会使用一个不同的 `tweetId`。
-* `List<Integer> getNewsFeed(int userId)` 检索当前用户新闻推送中最近 `10` 条推文的 ID。新闻推送中的每一项都必须是由用户关注的人或者是用户自己发布的推文。推文必须
-*按照时间顺序由最近到最远排序* 。
+* `List<Integer> 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` 的用户。
 
@@ -64,14 +63,14 @@ include::{sourcedir}/_0355_DesignTwitter.java[tag=answer]
 ----
 --
 
-// 二刷::
-// +
-// --
-// [{java_src_attr}]
-// ----
-// include::{sourcedir}/_0355_DesignTwitter_2.java[tag=answer]
-// ----
-// --
+二刷::
++
+--
+[{java_src_attr}]
+----
+include::{sourcedir}/_0355_DesignTwitter_2.java[tag=answer]
+----
+--
 ====
 
 

logbook/202503.adoc

@@ -1544,6 +1544,11 @@ endif::[]
 |{doc_base_url}/0480-sliding-window-median.adoc[题解]
 |⭕️ 滑动窗口+双堆。单纯利用双堆，只能通过 43 / 44 个测试用例。延迟删除的实现有些复杂，还要再研究一下。
 
+|{counter:codes2503}
+|{leetcode_base_url}/design-twitter/[355. 设计推特^]
+|{doc_base_url}/0355-design-twitter.adoc[题解]
+|✅ 多路归并的解法。
+
 |===
 
 截止目前，本轮练习一共完成 {codes2503} 道题。

src/main/java/com/diguage/algo/leetcode/_0355_DesignTwitter_2.java

@@ -0,0 +1,91 @@
+package com.diguage.algo.leetcode;
+
+import java.util.*;
+
+public class _0355_DesignTwitter_2 {
+  // tag::answer[]
+
+  /**
+   * @author D瓜哥 · https://www.diguage.com
+   * @since 2025-09-16 21:05:24
+   */
+  static class Twitter {
+
+    private Map<Integer, Set<Integer>> follower;
+    private Map<Integer, Tweet> tweets;
+
+    private int timestamp;
+
+
+    public Twitter() {
+      follower = new HashMap<>();
+      tweets = new HashMap<>();
+      timestamp = 0;
+    }
+
+    public void postTweet(int userId, int tweetId) {
+      timestamp++;
+      Tweet tweet = new Tweet(tweetId, timestamp);
+      tweet.next = tweets.get(userId);
+      tweets.put(userId, tweet);
+    }
+
+    public List<Integer> getNewsFeed(int userId) {
+      PriorityQueue<Tweet> queue = new PriorityQueue<>((a, b) ->
+        Integer.compare(b.timestamp, a.timestamp));
+      if (tweets.containsKey(userId)) {
+        queue.offer(tweets.get(userId));
+      }
+      Set<Integer> followers = follower.getOrDefault(userId, new HashSet<>());
+      for (Integer uid : followers) {
+        if (tweets.containsKey(uid)) {
+          queue.offer(tweets.get(uid));
+        }
+      }
+      List<Integer> result = new ArrayList<>();
+      while (!queue.isEmpty() && result.size() < 10) {
+        Tweet ts = queue.poll();
+        result.add(ts.tweetId);
+        if (ts.next != null) {
+          queue.offer(ts.next);
+        }
+      }
+      return result;
+    }
+
+    public void follow(int followerId, int followeeId) {
+      if (followeeId == followerId) {
+        return;
+      }
+      follower.computeIfAbsent(followerId, k -> new HashSet<>()).add(followeeId);
+    }
+
+    public void unfollow(int followerId, int followeeId) {
+      if (followeeId == followerId) {
+        return;
+      }
+      follower.computeIfAbsent(followerId, k -> new HashSet<>()).remove(followeeId);
+    }
+
+    private static class Tweet {
+      int tweetId;
+      int timestamp;
+      Tweet next;
+
+      public Tweet(int tweetId,int timestamp) {
+        this.tweetId = tweetId;
+        this.timestamp = timestamp;
+      }
+    }
+  }
+  // end::answer[]
+
+  public static void main(String[] args) {
+    Twitter twitter = new Twitter();
+    twitter.postTweet(1, 4);
+    twitter.postTweet(2, 5);
+    twitter.unfollow(1, 2);
+    twitter.follow(1, 2);
+    twitter.getNewsFeed(1);
+  }
+}