Design A Leaderboard | biweekly-contest-12 | leetcode | Java solution

Question Source

Java Solution:

class User implements Comparable<User> {
    int id;
    int score;
    
    public User(int id, int score) {
        this.id = id;
        this.score =score;
    }
    
    @Override
    public int compareTo(User u) {
        return u.score - this.score;
    }
    
    @Override
    public String toString(){
        return id +":"+score+" ";
    }
}

class Leaderboard {

    private Set<Integer> ids;
    private List<User> users;
    
    public Leaderboard() {
        this.ids = new HashSet<>();
        this.users = new ArrayList<>();
    }
    
    public void addScore(int playerId, int score) {
        if(!ids.contains(playerId)) {
            ids.add(playerId);
            users.add(new User(playerId, score));
        } else {
            for(User u : users) { 
            
            if(u.id == playerId) {
                u.score+= score;
                break;
            }
        }
        }
    }
    
    public int top(int K) {
        Collections.sort(users);
        System.out.println(K + " " + users);
        
        int sum = 0;
        for(int i = 0; i < K && i < users.size(); i++) {
            sum+= users.get(i).score;
        }
        return sum;
    }
    
    public void reset(int playerId) {
        ids.remove(playerId);
        User del = null;
        for(User u : users) { 
            
            if(u.id == playerId) {
                del = u;
                break;
            }
        }
        
        if(del != null) {
            users.remove(del);
        }
    }
}