TreeSet 클래스

👀 특징

• 객체의 정렬에 사용되는 클래스
• 중복을 허용하지 않으며(set) 오름차순이나 내림차순으로 객체를 정렬한다.
• 이진 검색 트리(binary search tree)로 구현되어 있다.
• 객체 비교를 위해 Comparable 이나 Comparator 인터페이스를 구현해야 한다.

---

🔨 생성과 주요 메소드

생성

Treeset<String> treeSet = new TreeSet<String>();

요소 추가

treeSet.add(str); // 추가

요소 제거

treeSet.remove(0);

요소 정렬

Comparable

• 주의!! 클래스로 새로 정의된 객체의 경우 어떻게 비교해서 넣을지 구현을 같이 해주어야 한다.
• 기본 자료형에는 Comparable 가 이미 구현되어 있다.

public class Member implements Comparable<Member> {

  @Override
  public int compareTo(Member member){
      // 기존 것과 비교할 것을 구현해주면 된다.
      return (this.memberId - member.memberId);
  }
 
 }

• return 값이
  - 양수 = memberId를 기준으로 오름차순 정렬
  - 음수 = memberId를 기준으로 내림차순 정렬

Comparator

• 보통은 Comparable을 사용하지만... 그래도 정리해두기!
• 주의! 사용 전에 생성자에 Default Construcktor 만들어두어야 한다.

Treeset<Member> treeSet = new TreeSet<Member>(new Member());
// 혹은 다른 compare가 있는 클래스도 가능

public class Member implements Comparator<Member> {

  @Override
  public int compare(Member member1, Member member2){
      // member1 == 기존 것
      // member2 == 새로 들어온 것
      // 기존 것과 비교할 것을 구현해주면 된다.
      return (member1.memberId - member2.memberId);
  }
 
 }

• return 값이
  - 양수 = memberId를 기준으로 오름차순 정렬
  - 음수 = memberId를 기준으로 내림차순 정렬

---

👿 관련 문제

• 백준1991: 트리 순회
  

---

🖇️ 참고 자료

• Do it! 자바 프로그래밍 입문