[Java][НП] - Audition

Да се имплементира класа за аудиција Audition со следните методи:

    void addParticpant(String city, String code, String name, int age) додава нов кандидат со код code, име и возраст за аудиција во даден град city. Во ист град не се дозволува додавање на кандидат со ист код како некој претходно додаден кандидат (додавањето се игнорира, а комплексноста на овој метод треба да биде O(1))
    void listByCity(String city) ги печати сите кандидати од даден град подредени според името, а ако е исто според возраста (комплексноста на овој метод не треба да надминува O(n∗log2(n)), каде n е бројот на кандидати во дадениот град).



import java.util.*;

class Participant implements Comparable<Participant> {
    private String city;
    private String code;
    private String name;
    private int age;

    public Participant(String city, String code, String name, int age) {
        this.city = city;
        this.code = code;
        this.name = name;
        this.age = age;
    }

    @Override
    public int compareTo(Participant o) {
        int res = name.toLowerCase().compareTo(o.name.toLowerCase());
        if(res == 0)
            return Integer.compare(age,o.age);
        return res;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Participant that = (Participant) o;
        return Objects.equals(code, that.code);
    }


    @Override
    public int hashCode() {
        return Objects.hash(code);
    }

    public String getCity() {
        return city;
    }

    public int getAge() {
        return age;
    }

    @Override
    public String toString() {
        return String.format("%s %s %d",code,name,age);
    }
}

class Audition {
    private Map<String,Set<Participant>> participants_by_city;

    public Audition() {
        participants_by_city = new HashMap<>();
    }
    public void addParticpant(String city, String code, String name, int age){
        participants_by_city.putIfAbsent(city,new HashSet<>());
        Participant participant = new Participant(city, code, name, age);
        participants_by_city.computeIfPresent(city,(k,v) -> {
            v.add(participant);//koristi equals, ako kodovite im se isti ne dodava takov
            return v;
        });


    }

    public void listByCity(String city){
        Set<Participant> participants_of_city = participants_by_city.get(city);
        participants_of_city.stream()
                .sorted(Participant::compareTo)
                .forEach(System.out::println);
    }
}

public class AuditionTest {
    public static void main(String[] args) {
        Audition audition = new Audition();
        List<String> cities = new ArrayList<String>();
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNextLine()) {
            String line = scanner.nextLine();
            String[] parts = line.split(";");
            if (parts.length > 1) {
                audition.addParticpant(parts[0], parts[1], parts[2],
                        Integer.parseInt(parts[3]));
            } else {
                cities.add(line);
            }
        }
        for (String city : cities) {
            System.out.printf("+++++ %s +++++\n", city);
            audition.listByCity(city);
        }
        scanner.close();
    }
}




Sample input

Скопје;001;Ана;17
Скопје;002;Борис;19
Скопје;002;Иван;15
Скопје;003;Јована;23
Скопје;004;Михаела;18
Битола;002;Николина;17
Битола;001;Стефанија;16
Битола;001;Елена;19
Битола;005;Анастасија;21
Битола;004;Игор;20
Гевгелија;003;Аце;17
Гевгелија;001;Иван;25
Гевгелија;002;Мартина;15
Гевгелија;005;Наташа;19
Гевгелија
Битола
Скопје

Sample output

+++++ Гевгелија +++++
003 Аце 17
001 Иван 25
002 Мартина 15
005 Наташа 19
+++++ Битола +++++
005 Анастасија 21
004 Игор 20
002 Николина 17
001 Стефанија 16
+++++ Скопје +++++
003 Јована 23
001 Ана 17
002 Борис 19
004 Михаела 18