Programski jezici → Java → Kolekcije u Javi: HashMap, HashSet i TreeMap

Kolekcije u Javi: HashMap, HashSet i TreeMap

HashMap: ključ i vrednost

import java.util.HashMap;
import java.util.Map;
import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        Map<String, Integer> stanovnici = new HashMap<>();
        stanovnici.put("Beograd", 1_681_405);    // dodavanje para
        stanovnici.put("Novi Sad", 368_967);
        stanovnici.put("Nis", 249_501);
        stanovnici.put("Nis", 249_816);          // isti ključ: vrednost se zamenjuje

        System.out.println(stanovnici.get("Novi Sad"));
        System.out.println(stanovnici.get("Kragujevac"));          // nema ključa: null
        System.out.println(stanovnici.getOrDefault("Kragujevac", 0));
        System.out.println("Ima Nis? " + stanovnici.containsKey("Nis") + ", parova: " + stanovnici.size());

        stanovnici.remove("Beograd");
        // ispis abecedno po ključu: kopija u TreeMap (HashMap nema određen redosled)
        for (Map.Entry<String, Integer> e : new TreeMap<>(stanovnici).entrySet())
            System.out.println(e.getKey() + " -> " + e.getValue());
    }
}

HashMap ne čuva nikakav određen redosled ključeva. Ako treba sortiran redosled, koristi se TreeMap, a ako treba redosled dodavanja – LinkedHashMap.

Brojanje pojavljivanja

Najčešća upotreba mape: koliko se puta pojavio svaki element. Metoda merge(kljuc, 1, Integer::sum) dodaje 1 na postojeću vrednost ili postavlja 1 za nov ključ.

import java.util.Map;
import java.util.Scanner;
import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String tekst = sc.nextLine().toLowerCase();
        Map<String, Integer> broj = new TreeMap<>();      // ključevi abecedno
        for (String rec : tekst.split("[^a-z]+")) {
            if (rec.isEmpty()) continue;
            broj.merge(rec, 1, Integer::sum);             // isto: broj.put(rec, broj.getOrDefault(rec, 0) + 1)
        }
        System.out.println(broj);

        // najčešća reč
        String najcesca = null;
        for (String r : broj.keySet())
            if (najcesca == null || broj.get(r) > broj.get(najcesca))
                najcesca = r;
        System.out.println("Najcesca: " + najcesca + " (" + broj.get(najcesca) + ")");
    }
}

HashSet i TreeSet: skup bez ponavljanja

import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        Set<Integer> a = new TreeSet<>(List.of(5, 1, 3, 9, 7, 3, 1));   // duplikati nestaju, sortirano
        Set<Integer> b = new TreeSet<>(List.of(3, 4, 5, 6));
        System.out.println("a = " + a + ", b = " + b);

        Set<Integer> presek = new TreeSet<>(a);
        presek.retainAll(b);                        // zadrži samo zajedničke
        Set<Integer> unija = new TreeSet<>(a);
        unija.addAll(b);
        Set<Integer> razlika = new TreeSet<>(a);
        razlika.removeAll(b);
        System.out.println("presek " + presek + ", unija " + unija + ", a\\b " + razlika);

        // add vraća false ako element već postoji – prvo ponavljanje u nizu
        Set<String> videno = new HashSet<>();
        for (String s : new String[]{"crvena", "plava", "zuta", "plava", "crvena"})
            if (!videno.add(s)) {
                System.out.println("Prvo ponavljanje: " + s);
                break;
            }
    }
}

Provera contains je u HashSet-u i HashMap-u praktično trenutna, bez obzira na broj elemenata, dok lista mora da pregleda element po element.

TreeMap: sortirani ključevi i pretraga

import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        // granica bodova -> ocena
        TreeMap<Integer, Integer> skala = new TreeMap<>();
        skala.put(0, 1);
        skala.put(50, 2);
        skala.put(65, 3);
        skala.put(80, 4);
        skala.put(90, 5);

        int[] bodovi = {45, 50, 72, 89, 100};
        for (int b : bodovi) {
            // floorKey: najveći ključ koji je <= b
            int ocena = skala.get(skala.floorKey(b));
            System.out.println(b + " bodova -> ocena " + ocena);
        }
        System.out.println("prva granica " + skala.firstKey() + ", poslednja " + skala.lastKey());
    }
}

Mapa sa listom: grupisanje

Vrednost u mapi može biti i lista. Metoda computeIfAbsent pravi praznu listu kada se ključ pojavi prvi put:

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        Map<String, List<String>> poOdeljenju = new TreeMap<>();
        for (int i = 0; i < n; i++) {
            String ime = sc.next(), odeljenje = sc.next();
            poOdeljenju.computeIfAbsent(odeljenje, k -> new ArrayList<>()).add(ime);
        }
        for (var e : poOdeljenju.entrySet())          // var: tip zaključuje prevodilac
            System.out.println(e.getKey() + ": " + String.join(", ", e.getValue()));
    }
}

Zadaci za vežbu

Za svaki zadatak postoji primer, uputstvo i rešenje. Rešenja su provereno tačna na primerima iz teksta.

Proveri znanje