TreeSet trong Java
Trong bài này, tôi sẽ hướng dẫn đến các bạn tìm hiểu loại Class Collection tiếp theo - đó là TreeSet trong Java. Nội dung của bài này sẽ mô tả đặc điểm, các phương thức thường dùng của Collection này. Với mỗi phương thức được liệt kê, tôi sẽ đưa ra ví dụ đơn giản để cho các bạn nắm bắt được. Cuối bài này, tôi sẽ đưa ra một số bài tập để các bạn luyện tập!
1. Đặc điểm
TreeSet
là Class triển khai phổ biến nhất của SortedSet
nên nó sẽ có một vài đặc điểm và phương thức tương đồng với SortedSet
. Như đã nói trong bài Tổng quan về Collection trong Java, thứ tự các phần tử trong TreeSet
mặc định được sắp xếp tăng dần và giá trị của các phần tử này là duy nhất.
2. Các phương thức phổ biến
Tạo mới một TreeSet
Để khai báo một TreeSet
, chúng ta cần phải import gói thư viện java.util.TreeSet
của Java. Cú pháp import như sau:
// Khai báo TreeSet // thì import gói thư viện java.util.TreeSet import java.util.TreeSet; public class TênClass { // ... }
Sau đây là ví dụ cách tạo mới một TreeSet
trong Java:
Bài viết này được đăng tại [free tuts .net]
public static void main(String[] args) { // khai báo 1 TreeSet có tên là treeSetInteger // có kiểu là Integer TreeSet<Integer> treeSetInteger = new TreeSet<>(); // khai báo 1 TreeSet có tên là treeSetCharacter // được tạo thành từ 1 Class Collection khác TreeSet<Character> treeSetCharacter = new TreeSet<>(new HashSet<>()); }
Hiển thị các phần tử có trong TreeSet
Để hiển thị các phần tử có trong TreeSet
, chúng ta có các cách như sau:
Hiển thị theo tên TreeSet.
public static void main(String[] args) { // khai báo 1 TreeSet có tên là treeSetCharacter // được tạo thành từ 1 Class Collection khác TreeSet<Character> treeSetCharacter = new TreeSet<>(new HashSet<>()); // Thêm các phần tử vào trong treeSetCharacter // sử dụng phương thức add() treeSetCharacter.add('R'); treeSetCharacter.add('A'); treeSetCharacter.add('B'); treeSetCharacter.add('R'); // hiển thị các phần tử có trong treeSetCharacter // trong treeSetCharacter có 2 phần tử là 'R' // mà các phần tử trong treeSetCharacter là không trùng nhau // nên sẽ chỉ có 1 phần tử 'R' được hiển thị // các phần tử trong treeSetCharacter // mặc định được sắp xếp theo thứ tự tăng dần. System.out.println("Các phần tử có trong treeSetCharacter là: " + treeSetCharacter); }
Kết quả sau khi biên dịch chương trình:
Sử dụng vòng lặp for
cải tiến duyệt theo đối tượng trong TreeSet.
public static void main(String[] args) { TreeSet<Integer> treeSetInteger = new TreeSet<>(); treeSetInteger.add(5); treeSetInteger.add(9); treeSetInteger.add(4); treeSetInteger.add(2); // hiển thị các phần tử có trong treeSetInteger // bằng cách sử dụng vòng lặp for duyệt theo đối tượng // trong đó kiểu dữ liệu của biến number // phải trùng với kiểu dữ liệu của treeSetInteger System.out.println("Các phần tử của treeSetInteger là: "); for (int number : treeSetInteger) { System.out.print(number + "\t"); } }
Kết quả sau khi biên dịch chương trình:
Sử dụng Iterator.
Để sử dụng được Iterator
chúng ta cần phải import gói thư viện java.util.Iterator
của Java:
public static void main(String[] args) { // khai báo TreeSet có kiểu là String TreeSet<String> treeSet = new TreeSet<>(); // thêm vào phần tử cho treeSet treeSet.add("C#"); treeSet.add("Java"); treeSet.add("PHP"); treeSet.add("SQL"); treeSet.add("HTML"); treeSet.add("CSS"); // khai báo một Iterator có kiểu là String Iterator<String> iterator = treeSet.iterator(); // hiển thị các phần tử có trong treeSet // bằng cách sử dụng Iterator System.out.println("Các phần tử có trong treeSet là: "); while (iterator.hasNext()) { System.out.print(iterator.next() + "\t"); } }
Kết quả sau khi biên dịch chương trình:
Thêm phần tử vào trong TreeSet
Thêm phần tử sử dụng phương thức add().
public static void main(String[] args) {
int number;
TreeSet<Integer> treeSetInteger = new TreeSet<>();
Scanner scanner = new Scanner(System.in);
// thêm các phần tử vào treeSetInteger
treeSetInteger.add(0);
treeSetInteger.add(3);
treeSetInteger.add(1);
treeSetInteger.add(4);
treeSetInteger.add(2);
treeSetInteger.add(8);
// hiển thị các phần tử trong treeSetInteger
System.out.println("Các phần tử trong treeSetInteger: ");
System.out.println(treeSetInteger);
System.out.println("Nhập phần tử cần thêm: ");
number = scanner.nextInt();
// thêm một phần tử mới vào treeSetInteger từ bàn phím
// nếu phần tử đó đã tồn tại thì thông báo đã tồn tại
// ngược lại thì thêm vào
if (!treeSetInteger.contains(number)) {
treeSetInteger.add(number);
System.out.println("Thêm thành công!");
System.out.println("Các phần tử trong treeSetInteger sau khi thêm: ");
System.out.println(treeSetInteger);
} else {
System.out.println("Phần tử " + number + " đã tồn tại!");
}
}
Kết quả sau khi biên dịch chương trình:
Thêm phần tử sử dụng phương thức addAll().
Java cung cấp cho chúng ta phương thức addAll()
để thêm tất cả các phần tử của một Collection
khác vào 1 TreeSet
đã tồn tại. Lưu ý: kiểu dữ liệu của Collection
cần thêm vào và TreeSet
này phải giống nhau. Ví dụ dưới đây sẽ khởi tạo 2 TreeSet
đó là treeSet
và treeSetExists
có cùng kiểu dữ liệu là Float
. Phương thức addAll()
sẽ thêm các phần tử có trong treeSet
vào treeSetExists
.
public static void main(String[] args) { // khai báo 2 TreeSet có kiểu là Float TreeSet<Float> treeSet = new TreeSet<>(); TreeSet<Float> treeSetExists = new TreeSet<>(); // thêm vào phần tử cho treeSet và treeSetExists treeSet.add(5f); treeSet.add(8.04f); treeSet.add(0.2f); treeSetExists.add(4.4f); treeSetExists.add(1.9f); // thêm các phần tử của treeSet // vào treeSetExists treeSetExists.addAll(treeSet); // hiển thị treeSetExists sau khi thêm // các phần tử được sắp xếp tăng dần System.out.println("Các phần tử có trong treeSetExists là: "); System.out.print(treeSetExists + "\t"); }
Kết quả sau khi biên dịch chương trình:
Xóa phần tử
Để xóa một phần tử khỏi TreeSet
, chúng ta sẽ sử dụng phương thức remove()
. Ví dụ dưới đây sẽ sử dụng phương thức remove()
để xóa một phần tử bất kỳ trong treeSetDouble
:
public static void main(String[] args) { double number; TreeSet<Double> treeSetDouble = new TreeSet<>(); Scanner scanner = new Scanner(System.in); // thêm các phần tử vào treeSetDouble treeSetDouble.add(0.5d); treeSetDouble.add(8.33d); treeSetDouble.add(1.2d); // hiển thị các phần tử trong treeSetDouble System.out.println("Các phần tử trong treeSetDouble: "); System.out.println(treeSetDouble); System.out.println("Nhập phần tử cần xóa: "); number = scanner.nextDouble(); // nếu phần tử cần xóa // có tồn tại treeSetDouble thì sẽ thông báo xóa thành công // và hiển thị các phần tử còn lại // ngược lại thông báo xóa không thành công if (treeSetDouble.contains(number)) { treeSetDouble.remove(number); System.out.println("Xóa thành công!"); System.out.println("Các phần tử còn lại trong treeSetDouble:"); System.out.println(treeSetDouble); } else { System.out.println("Xóa không thành công!"); } }
Kết quả sau khi biên dịch chương trình:
Trường hợp 1: Xóa phần tử thành công:
Trường hợp 2: Xóa phần tử thất bại:
Để xóa tất cả các phần tử có trong TreeSet
, chúng ta sẽ sử dụng phương thức clear()
có sẵn của Java. Ví dụ:
public static void main(String[] args) { int Number; TreeSet<Integer> treeSetInteger = new TreeSet<>(); Scanner scanner = new Scanner(System.in); // thêm các phần tử vào treeSetInteger treeSetInteger.add(7); treeSetInteger.add(19); treeSetInteger.add(1); treeSetInteger.add(20); // xóa toàn bộ các phần tử trong treeSetInteger // sử dụng phương thức clear() treeSetInteger.clear(); // sau khi xóa thì trong treeSetInteger // sẽ không có phần tử nào // phương thức isEmpty() dưới đây sẽ kiểm tra // nếu treeSetInteger không có giá trị // thì sẽ hiển thị thông báo "Không có phần tử" if (treeSetInteger.isEmpty()) { System.out.println("Không có phần tử."); } }
Kết quả sau khi biên dịch chương trình:
Trích xuất một phần trong TreeSet
Đối với TreeSet
, Java cung cấp cho chúng ta các phương thức để trích xuất các phần tử trong TreeSet
đó là subset()
, headset()
và tailset()
. Sau đây chúng ta sẽ cùng nhau tìm hiểu về 3 phương thức này.
Phương thức subSet().
SortedSet subSet(E fromElement, E toElement)
Công dụng: Phương thức subSet()
sẽ trả về một SortedSet
được trích xuất từ phần tử fromElement
đến phần tử đứng trước phần tử toElement
của một TreeSet
cho trước.
public static void main(String[] args) { TreeSet<Integer> treeSetInteger = new TreeSet<>(); // thêm các phần tử vào trong treeSetInteger treeSetInteger.add(2); treeSetInteger.add(1); treeSetInteger.add(4); treeSetInteger.add(3); treeSetInteger.add(6); treeSetInteger.add(5); treeSetInteger.add(8); treeSetInteger.add(7); treeSetInteger.add(0); treeSetInteger.add(9); System.out.println("Các phần tử có trong treeSetInteger: "); System.out.println(treeSetInteger); // khai báo 1 TreeSet có tên là subset // có các phần tử được trích xuất // trong đoạn [3,10) của treeSetInteger SortedSet<Integer> subset = treeSetInteger.subSet(3, 7); System.out.println("Các phần tử có trong subset: "); System.out.println(subset); // nếu phần tử đầu và phần tử cuối bằng nhau // thì kết quả của phương thức subSet() // sẽ trả về subset không có phần tử nào subset = treeSetInteger.subSet(3, 3); System.out.println("Các phần tử có trong subset: "); System.out.println(subset); }
Kết quả sau khi biên dịch chương trình:
Phương thức headSet().
SortedSet headSet(E toElement)
Công dụng: Phương thức headSet(
) sẽ trả về một SortedSet
được trích xuất từ phần tử đầu tiên đến phần tử đứng trước phần tử toElement
của một TreeSet
cho trước.
public static void main(String[] args) { TreeSet<Character> treeSetChar = new TreeSet<>(); // thêm các phần tử vào trong treeSetChar treeSetChar.add('V'); treeSetChar.add('A'); treeSetChar.add('B'); treeSetChar.add('C'); treeSetChar.add('G'); treeSetChar.add('E'); System.out.println("Các phần tử có trong treeSetChar: "); System.out.println(treeSetChar); // khai báo 1 SortedSet có tên là headset // có các phần tử được trích xuất // từ phần tử đầu tiên đến // phần tử đứng trước phần tử 'E' trong treeSetChar SortedSet<Character> headset = treeSetChar.headSet('E'); System.out.println("Các phần tử có trong headset: "); System.out.println(headset); }
Kết quả sau khi biên dịch chương trình:
Phương thức tailSet().
SortedSet tailSet(E fromElement)
Công dụng: Phương thức tailSet()
sẽ trả về một SortedSet
được trích xuất từ phần tử lớn hơn hoặc bằng phần tử fromElement
đến phần tử cuối cùng của một TreeSet
cho trước.
public static void main(String[] args) { TreeSet<Float> treeSetFloat = new TreeSet<>(); // thêm các phần tử vào trong TreeSet treeSetFloat.add(2.45f); treeSetFloat.add(1.9f); treeSetFloat.add(4.6f); treeSetFloat.add(3.20f); treeSetFloat.add(6.3f); treeSetFloat.add(5.17f); System.out.println("Các phần tử có trong treeSetFloat: "); System.out.println(treeSetFloat); // khai báo 1 SortedSet có tên là tailset // có các phần tử được trích xuất // từ phần tử lớn hơn hoặc bằng // phần tử 4.6f đến phần tử cuối cùng của treeSetFloat SortedSet<Float> tailset = treeSetFloat.tailSet(4.6f); System.out.println("Các phần tử có trong tailset: "); System.out.println(tailset); }
Kết quả sau khi biên dịch chương trình:
Tìm phần tử nhỏ nhất và lớn nhất trong SortedSet
Để tìm phần tử nhỏ nhất và lớn nhất trong SortedSet
, Java cung cấp cho chúng ta 2 phương thức đó là first()
(tìm phần tử nhỏ nhất) và last()
(tìm phần tử lớn nhất).
public static void main(String[] args) { TreeSet<String> treeSetString = new TreeSet<>(); // thêm các phần tử vào trong treeSetString treeSetString.add("JAVA"); treeSetString.add("PHP"); treeSetString.add("Spring"); treeSetString.add("Struts"); treeSetString.add("Python"); treeSetString.add("HTML"); System.out.println("Các phần tử có trong treeSetString: "); System.out.println(treeSetString); // tìm phần tử lớn nhất (phần tử cuối cùng) // và phần tử nhỏ nhất (phần tử đầu tiên) trong treeSetString String phanTuLonNhat = treeSetString.last(); String phanTuNhoNhat = treeSetString.first(); System.out.println("Phần tử lớn nhất và nhỏ nhất trong" + " treeSetString là " + phanTuLonNhat + " và " + phanTuNhoNhat); }
Kết quả sau khi biên dịch chương trình:
3. Lời kết
Trong bài này, tôi đã giới thiệu cho các bạn đặc điểm, các phương thức thường dùng đối với TreeSet
. Sang bài sau tôi sẽ giới thiệu đến các bạn một loại Interface Collection tiếp theo, đó là HashMap trong Java. Các bạn theo dõi nhé!
Câu hỏi thường gặp liên quan:
- Java - Viết chương trình tìm và hiển thị các phần tử nhỏ hơn 7 trong TreeSet.
- Java - Viết chương trình đảo ngược các phần tử của TreeSet.
- Java - Viết chương trình tìm phần tử nhỏ nhất trong TreeSet và xóa phần tử đó ra khỏi TreeSet.