본문 바로가기
카테고리 없음

자바 TreeMap 사용법 & 예제 총정리

by kangs' tong 2023. 10. 24.

자바 TreeMap 사용법

1. TreeMap이란?

TreeMap은 자바의 컬렉션 클래스 중 하나로, 키와 값으로 이루어진 데이터를 정렬된 순서로 저장한다. 이진 탐색 트리(binary search tree) 구조를 기반으로 하며, 키에 대한 오름차순으로 정렬된다.

2. TreeMap 선언하기

TreeMap은 java.util.TreeMap 패키지에 포함되어 있어 해당 패키지를 import 해주어야 한다. 아래는 TreeMap을 선언하는 예제이다.

import java.util.TreeMap;

public class TreeMapExample {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();
    }
}

위의 예제에서는 키를 Integer 타입으로, 값은 String 타입으로 선언된 TreeMap을 생성하였다.

3. TreeMap에 데이터 추가하기

TreeMap에 데이터를 추가하기 위해서는 put() 메서드를 사용한다. 아래는 데이터를 추가하는 예제이다.

import java.util.TreeMap;

public class TreeMapExample {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();

        treeMap.put(1, "Apple");
        treeMap.put(3, "Banana");
        treeMap.put(2, "Orange");
    }
}

위의 예제에서는 각각의 키와 값으로 "Apple", "Banana", "Orange"을 TreeMap에 추가하였다.

4. TreeMap 데이터 접근

TreeMap에서 데이터를 접근하는 방법으로는 get() 메서드와 entrySet() 메서드를 사용할 수 있다. 아래는 데이터를 접근하는 예제이다.

4.1. get() 메서드를 통한 데이터 접근하기

get() 메서드를 사용하여 해당 키에 대한 값에 접근할 수 있다. 아래는 get() 메서드를 사용하는 예제이다.

import java.util.TreeMap;

public class TreeMapExample {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();

        treeMap.put(1, "Apple");
        treeMap.put(3, "Banana");
        treeMap.put(2, "Orange");

        String value = treeMap.get(2);
        System.out.println(value); // Orange
    }
}

위의 예제에서는 get(2)를 통해 키 2에 대한 값을 가져와 출력하였다.

4.2. entrySet() 메서드를 통한 데이터 접근하기

entrySet() 메서드는 TreeMap에 저장된 모든 키와 값의 쌍을 Map.Entry 타입으로 반환한다. 이를 통해 모든 데이터에 접근할 수 있다. 아래는 entrySet() 메서드를 사용하는 예제이다.

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

public class TreeMapExample {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();

        treeMap.put(1, "Apple");
        treeMap.put(3, "Banana");
        treeMap.put(2, "Orange");

        for (Map.Entry<Integer, String> entry : treeMap.entrySet()) {
            System.out.println(entry.getKey() + " : " + entry.getValue());
        }
        // 1 : Apple
        // 2 : Orange
        // 3 : Banana
    }
}

위의 예제에서는 entrySet() 메서드를 통해 TreeMap의 모든 데이터에 접근하여 키와 값을 출력하였다.

5. TreeMap 데이터 삭제

TreeMap에서 데이터를 삭제하기 위해서는 remove() 메서드를 사용한다. 아래는 데이터를 삭제하는 예제이다.

import java.util.TreeMap;

public class TreeMapExample {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();

        treeMap.put(1, "Apple");
        treeMap.put(3, "Banana");
        treeMap.put(2, "Orange");

        treeMap.remove(2);

        System.out.println(treeMap);
        // {1=Apple, 3=Banana}
    }
}

위의 예제에서는 remove(2)를 통해 키 2에 해당하는 데이터를 삭제하였다.

정리

위의 포스팅에서는 자바 TreeMap의 사용법과 예제를 총정리하였습니다. TreeMap의 선언 방법, 데이터 추가 방법, 데이터 접근 방법, 데이터 삭제 방법에 대해 알아보았습니다. TreeMap은 정렬된 순서로 데이터를 저장하고, 키를 기준으로 정렬되기 때문에 특정 키에 대한 검색이나 범위 검색에 유용하게 사용될 수 있습니다.

댓글