leetcode-4.13[355. 设计推特]（python解法）

题目

题解

from collections import defaultdict

class Twitter:
    def __init__(self):
        """
        Initialize your data structure here.
        先说初始化函数，self.tweets是一个字典结构，键是userId，值是tweetId的列表，保存每个用户发表的tweet信息。类似，following         保存每个用户的关注列表。idd是保持所有tweet时间序的整数，无论哪位用户发一条tweet前idd都会自增1。
        """
        self.tweets = defaultdict(list)
        self.following = defaultdict(set)
        self.idd = 0

    def postTweet(self, userId: int, tweetId: int) -> None:
        """
        Compose a new tweet.
        """
        self.idd += 1
        self.tweets[userId].append((tweetId, self.idd))

    def follow(self, followerId: int, followeeId: int) -> None:
        """
        Follower follows a followee. If the operation is invalid, it should be a no-op.
        """
        self.following[followerId].add(followeeId)  #这里采用的集合，存在值则不add，不存在则add

    def unfollow(self, followerId: int, followeeId: int) -> None:
        """
        Follower unfollows a followee. If the operation is invalid, it should be a no-op.
        """
        if followeeId in self.following[followerId]:
            self.following[followerId].remove(followeeId)

    def getNewsFeed(self, userId: int) -> List[int]:
        """
        Retrieve the 10 most recent tweet ids in the user's news feed. Each item in the news feed must be posted by            users who the user followed or by the user herself. Tweets must be ordered from most recent to least recent.
        """
        users = self.following[userId]
        users.add(userId)
        tmp_data = [self.tweets[item][-10:] for item in users] # 取出每个用户(包括自己)的最后发布的10条推文
        tmp_data = [tmp for item in tmp_data for tmp in item] # 将每个用户的前10条推文都整合一起
        tmp_data.sort(key= lambda a: a[1], reverse=True) # 根据idd从大到小排序
        return [item[0] for item in tmp_data[:10]]  # 取出推文


# Your Twitter object will be instantiated and called as such:
# obj = Twitter()
# obj.postTweet(userId,tweetId)
# param_2 = obj.getNewsFeed(userId)
# obj.follow(followerId,followeeId)
# obj.unfollow(followerId,followeeId)