[leetcode/lintcode 题解] Amazon 面试题：迷你推特

时间：2020-12-23 12:31:01 阅读：0 评论：0 收藏：0 [点我收藏+]

标签：des solution zhang news col nod ++ selected container

实现一个迷你的推特，支持下列几种方法

?postTweet(user_id, tweet_text).? 发布一条推特.
?getTimeline(user_id).? 获得给定用户最新发布的十条推特，按照发布时间从最近的到之前排序
?getNewsFeed(user_id).? 获得给定用户的朋友或者他自己发布的最新十条推特，从发布时间最近到之前排序
?follow(from_user_id, to_user_id).? from_user_id 关注 to_user_id.
?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

[leetcode/lintcode 题解] Amazon 面试题：迷你推特

标签：des solution zhang news col nod ++ selected container

原文地址：https://www.cnblogs.com/lintcode/p/14157027.html

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行