FriendShip Service

Support follow & unfollow, getFollowers, getFollowings.
Note: the results of getfollow() are sorted.

Example

follow(1, 3)
getFollowers(1) // return [3]
getFollowings(3) // return [1]
follow(2, 3)
getFollowings(3) // return [1,2]
unfollow(1, 3)
getFollowings(3) // return [2]

思路：TreeSet是已经排好序的set，直接用treeset去收集就可以了；

public class FriendshipService {
    private HashMap<Integer, Set<Integer>> followers, followings;
    public FriendshipService() {
        this.followers = new HashMap<Integer, Set<Integer>>();
        this.followings = new HashMap<Integer, Set<Integer>>();
    }

    /*
     * @param user_id: An integer
     * @return: all followers and sort by user_id
     */
    public List<Integer> getFollowers(int user_id) {
        if(!followers.containsKey(user_id)){
            return new ArrayList<Integer>();
        }
        return new ArrayList<Integer>(followers.get(user_id));
    }

    /*
     * @param user_id: An integer
     * @return: all followings and sort by user_id
     */
    public List<Integer> getFollowings(int user_id) {
        if(!followings.containsKey(user_id)) {
            return new ArrayList<Integer>();
        }
        return new ArrayList<Integer>(followings.get(user_id));
    }

    /*
     * @param from_user_id: An integer
     * @param to_user_id: An integer
     * @return: nothing
     */
    public void follow(int to_user_id, int from_user_id) {
        if(!followers.containsKey(to_user_id)){
            followers.put(to_user_id, new TreeSet<Integer>());
        }
        followers.get(to_user_id).add(from_user_id);
        
        if(!followings.containsKey(from_user_id)){
            followings.put(from_user_id, new TreeSet<Integer>());
        }
        followings.get(from_user_id).add(to_user_id);
    }

    /*
     * @param from_user_id: An integer
     * @param to_user_id: An integer
     * @return: nothing
     */
    public void unfollow(int to_user_id, int from_user_id) {
        if(followers.containsKey(to_user_id)){
            followers.get(to_user_id).remove(from_user_id);
        }
        if(followings.containsKey(from_user_id)){
            followings.get(from_user_id).remove(to_user_id);
        }
    }
}