Android에서 TreeSet(트리 셋) 사용하는 방법 – 단계별 예제 가이드

이 예제는 Android 앱에서 TreeSet(트리 셋)을 사용하는 방법을 단계별로 보여줍니다. TreeSet은 요소가 자동으로 오름차순 정렬되고 중복 값을 허용하지 않는 자료구조이므로, 정렬된 상태로 고유한 문자열 목록을 관리할 때 특히 유용합니다.

1단계 — 새 프로젝트 만들기

Android Studio에서 File ⇒ New Project로 이동한 후, 프로젝트 생성에 필요한 모든 정보를 입력하여 새 프로젝트를 만듭니다.

2단계 — 레이아웃 작성 (activity_main.xml)

res/layout/activity_main.xml 파일에 다음 코드를 추가합니다.

<?xml version="1.0" encoding="utf-8"?>
<LinearLayout xmlns:android="https://schemas.android.com/apk/res/android"
    xmlns:tools="https://schemas.android.com/tools"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    tools:context=".MainActivity"
    android:orientation="vertical">
    <EditText
        android:id="@+id/name"
        android:layout_width="match_parent"
        android:hint="Enter Name"
        android:layout_height="wrap_content" />
    <LinearLayout
        android:layout_width="wrap_content"
        android:layout_height="wrap_content">
        <Button
            android:id="@+id/save"
            android:text="Save"
            android:layout_width="wrap_content"
            android:layout_height="wrap_content" />
        <Button
            android:id="@+id/refresh"
            android:text="Refresh"
            android:layout_width="wrap_content"
            android:layout_height="wrap_content" />
    </LinearLayout>
    <ListView
        android:id="@+id/listView"
        android:layout_width="match_parent"
        android:layout_height="wrap_content">
    </ListView>
</LinearLayout>

위 레이아웃은 이름을 입력받는 EditText, 저장(Save)·새로고침(Refresh) 두 개의 버튼, 그리고 데이터를 표시하는 ListView로 구성되어 있습니다. 사용자가 저장 버튼을 누르면 입력한 이름이 TreeSet에 추가되고, 그 내용이 ArrayList로 복사되어 ListView에 반영됩니다. 새로고침 버튼을 누르면 ListView의 변경 사항이 즉시 갱신됩니다.

3단계 — MainActivity 코드 작성

src/MainActivity.java 파일에 다음 코드를 추가합니다.

package com.example.andy.myapplication;

import android.os.Bundle;
import android.support.v7.app.AppCompatActivity;
import android.view.View;
import android.widget.ArrayAdapter;
import android.widget.EditText;
import android.widget.ListView;
import android.widget.Toast;
import java.util.ArrayList;
import java.util.Set;
import java.util.TreeSet;

public class MainActivity extends AppCompatActivity {
    EditText name;
    ArrayAdapter arrayAdapter;
    ArrayList<String> array_list;
    Set set = new TreeSet();
    private ListView listView;

    @Override
    protected void onCreate(Bundle readdInstanceState) {
        super.onCreate(readdInstanceState);
        setContentView(R.layout.activity_main);
        array_list = new ArrayList<>();
        name = findViewById(R.id.name);
        listView = findViewById(R.id.listView);
        findViewById(R.id.refresh).setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                arrayAdapter.notifyDataSetChanged();
                listView.invalidateViews();
                listView.refreshDrawableState();
            }
        });

        findViewById(R.id.save).setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                if (!name.getText().toString().isEmpty()) {
                    set.add(name.getText().toString());
                    array_list.clear();
                    array_list.addAll(set);
                    arrayAdapter = new ArrayAdapter(MainActivity.this, android.R.layout.simple_list_item_1, array_list);
                    listView.setAdapter(arrayAdapter);
                    Toast.makeText(MainActivity.this, "Inserted", Toast.LENGTH_LONG).show();
                } else {
                    name.setError("Enter NAME");
                }
            }
        });
    }
}

주요 코드 설명

• Set set = new TreeSet(); — 이름을 저장할 TreeSet 인스턴스를 생성합니다. TreeSet은 자동 정렬과 중복 제거를 함께 처리해 줍니다.
• 저장 버튼 클릭 시 — 입력값이 비어 있지 않으면 TreeSet에 이름을 추가한 뒤, TreeSet의 전체 내용을 ArrayList로 옮겨 ArrayAdapter를 통해 ListView에 표시합니다. 값이 비어 있으면 "Enter NAME" 오류 메시지를 표시합니다.
• 새로고침 버튼 클릭 시 — notifyDataSetChanged(), invalidateViews(), refreshDrawableState()를 차례로 호출해 ListView를 최신 상태로 갱신합니다.

앱 실행 및 결과 확인

이제 애플리케이션을 실행해 보겠습니다. 실제 Android 기기가 컴퓨터에 연결되어 있다고 가정합니다. Android Studio에서 프로젝트의 액티비티 파일 중 하나를 열고 툴바의 Run 아이콘을 클릭합니다. 목록에서 본인의 모바일 기기를 선택하면 기기 화면에 앱의 기본 화면이 표시됩니다.

위 실행 결과에서 입력한 이름이 TreeSet을 거쳐 정렬된 상태로 ArrayList에 삽입되고, ListView에 알파벳 순서대로 표시되는 것을 확인할 수 있습니다. 같은 이름을 여러 번 입력해도 TreeSet이 중복을 걸러 주므로 목록에는 항상 고유한 값만 남게 됩니다.