class Firma implements Iterable<Pracownik> {
    private List<Pracownik> pracownicy = new ArrayList<>();

    void addPracownik(Pracownik pracownik) {
        this.pracownicy.add(pracownik);
    }

    public int getLiczbaPracownikow() {
        return pracownicy.size();
    }

    public List<Pracownik> getByStanowisko(Stanowisko stanowisko) {
        return pracownicy.stream().filter(pracownik -> pracownik.getStanowisko() == stanowisko).collect(Collectors.toList());
    }

    public int getAveragePensja() {
        int suma = pracownicy.stream().mapToInt(Pracownik::getPensja).sum();
        return suma / pracownicy.size();
    }

    public int getAveragePensja(Stanowisko stanowisko) {
        int suma = getByStanowisko(stanowisko).stream().mapToInt(Pracownik::getPensja).sum();
        return suma / getByStanowisko(stanowisko).size();
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        pracownicy.forEach(p -> sb.append(p.toString()).append('\n'));
        return sb.toString();
    }

    @NotNull
    @Override
    public Iterator<Pracownik> iterator() {
        return pracownicy.iterator();
    }

    @Override
    public void forEach(Consumer<? super Pracownik> action) {

    }

    @Override
    public Spliterator<Pracownik> spliterator() {
        return null;
    }
}