[leetcode/lintcode 題解] Amazon 面試題：迷你推特

實現一個迷你的推特，支持下列幾種方法

1. ​postTweet(user_id, tweet_text).​ 發佈一條推特.
2. ​getTimeline(user_id).​ 得到給定用戶最新發布的十條推特，按照發布時間從最近的到以前排序
3. ​getNewsFeed(user_id).​ 得到給定用戶的朋友或者他本身發佈的最新十條推特，從發佈時間最近到以前排序
4. ​follow(from_user_id, to_user_id).​ from_user_id 關注 to_user_id.
5. ​unfollow(from_user_id, to_user_id).​ from_user_id 取消關注 to_user_id.

 

在線評測地址：領釦題庫官網

 

 

樣例 1:

自動換行

輸入: 

 postTweet(1, "LintCode is Good!!!")

 getNewsFeed(1)

 getTimeline(1)

 follow(2, 1)

 postTweet(1, "LintCode is best!!!")

 getNewsFeed(2)

 unfollow(2, 1)

 getNewsFeed(2)

輸出: 

 1

 [1]

 [1]

 2

 [2,1]

 []

題解

由於題目裏涉及到對推文按照時間排序, 同時 Tweet 類自己不含時間信息, 因此咱們須要額外地記錄每條推文發出的時間.

能夠定義一個類的靜態變量做爲計數器來實現.

而後分析咱們須要的數據結構:

• ​class Node {Tweet, int};​ 對原有的Tweet類的擴展, 使其能夠記錄時間 (固然, 也能夠用類的繼承來實現)
• ​map<int, vector<Node>>​ 用戶id到這個用戶發送了的推文的映射
• ​map<int, set<int>>​ 用戶id到這個用戶關注的人的id的映射

而後對應每種方法的實現:

• ​postTweet()​ 直接添加到​map<int, vector<Node>>​中便可
• ​getTimeline()​ 根據​map<int, vector<Node>>​得到該用戶的最新推文, 返回便可
• ​getNewsFeed()​ 同時用到上面定義的兩個映射, 比較暴力的作法是獲取這些用戶的全部推文, 排序, 拿出前十個; 或者能夠利用堆進行 "多路歸併"
• ​follow()​ 在 ​map<int, set<int>>​ 中添加便可
• ​unfollow()​ 在 ​map<int, set<int>>​ 中刪除便可

(本題解使用C++相關數據結構描述, 不過映射和集合在其餘語言中也有對應的實現)

 

自動換行

public class MiniTwitter {

   class Node {

       public int order;

       public Tweet tweet;

       public Node(int o, Tweet t) {

           this.order = o;

           this.tweet = t;

       }

   }

   class SortByOrder implements Comparator {

       public int compare(Object obj1, Object obj2) {

           Node node1 = (Node) obj1;

           Node node2 = (Node) obj2;

           if (node1.order < node2.order)

               return 1;

           else

               return -1;

       }

   }

   private Map> friends;

   private Map> users_tweets;

   private int order;

   public List getLastTen(List tmp) {

       int last = 10;

       if (tmp.size() < 10)

           last = tmp.size();

       return tmp.subList(tmp.size() - last, tmp.size());

   }

   public List getFirstTen(List tmp) {

       int last = 10;

       if (tmp.size() < 10)

           last = tmp.size();

       return tmp.subList(0, last);

   }

   public MiniTwitter() {

       // initialize your data structure here.

       this.friends = new HashMap>();

       this.users_tweets = new HashMap>();

       this.order = 0;

   }

   // @param user_id an integer

   // @param tweet a string

   // return a tweet

   public Tweet postTweet(int user_id, String tweet_text) {

       // Write your code here

       Tweet tweet = Tweet.create(user_id, tweet_text);

       if (!users_tweets.containsKey(user_id))

           users_tweets.put(user_id, new ArrayList());

       order += 1;

       users_tweets.get(user_id).add(new Node(order, tweet));

       return tweet;

   }

   // @param user_id an integer

   // return a list of 10 new feeds recently

   // and sort by timeline

   public List getNewsFeed(int user_id) {

       // Write your code here

       List tmp = new ArrayList();

       if (users_tweets.containsKey(user_id))

           tmp.addAll(getLastTen(users_tweets.get(user_id)));

       if (friends.containsKey(user_id)) {

           for (Integer user : friends.get(user_id)) {

               if (users_tweets.containsKey(user))

                   tmp.addAll(getLastTen(users_tweets.get(user)));

           }

       }

       Collections.sort(tmp, new SortByOrder());

       List rt = new ArrayList();

       tmp = getFirstTen(tmp);

       for (Node node : tmp) {

           rt.add(node.tweet);

       }

       return rt;

   }

   // @param user_id an integer

   // return a list of 10 new posts recently

   // and sort by timeline

   public List getTimeline(int user_id) {

       // Write your code here

       List tmp = new ArrayList();

       if (users_tweets.containsKey(user_id))

           tmp.addAll(getLastTen(users_tweets.get(user_id)));

       Collections.sort(tmp, new SortByOrder());

       List rt = new ArrayList();

       tmp = getFirstTen(tmp);

       for (Node node : tmp)

           rt.add(node.tweet);

       return rt;

   }

   // @param from_user_id an integer

   // @param to_user_id an integer

   // from user_id follows to_user_id

   public void follow(int from_user_id, int to_user_id) {

       // Write your code here

       if (!friends.containsKey(from_user_id))

           friends.put(from_user_id, new HashSet());

       friends.get(from_user_id).add(to_user_id);

   }

   // @param from_user_id an integer

   // @param to_user_id an integer

   // from user_id unfollows to_user_id

   public void unfollow(int from_user_id, int to_user_id) {

       // Write your code here

       if (friends.containsKey(from_user_id))

           friends.get(from_user_id).remove(to_user_id);

   }

}

更多題解參考：九章官網solution