Home » Java » Java Programs

Java program to get the subset of a TreeSet collection based on a specified range

Given a TreeSet collection, we have to get the subset based on a specified range.
Submitted by Nidhi, on June 08, 2022

Problem statement

In this program, we will create a TreeSet collection with integer elements. Then we will get the subset of a TreeSet collection using the subSet() method.

Source Code

The source code to get the subset of a TreeSet collection based on a specified range is given below. The given program is compiled and executed successfully.

Example of TreeSet.subSet() Method in Java

// Java program to get the subset of a TreeSet collection 
// based on a specified range

import java.io.*;
import java.util.*;

public class Main {
  public static void main(String args[]) {
    TreeSet < Integer > tree = new TreeSet < Integer > ();

    tree.add(10);
    tree.add(20);
    tree.add(25);
    tree.add(30);
    tree.add(35);
    tree.add(40);

    TreeSet < Integer > subSet = (TreeSet < Integer > ) tree.subSet(20, 35);

    System.out.println("TreeSet: " + tree);
    System.out.println("SubSet: " + subSet);
  }
}

Output

TreeSet: [10, 20, 25, 30, 35, 40]
SubSet: [20, 25, 30]

Explanation

In the above program, we imported the "java.util.*", "java.io.*" packages to use the TreeSet class. Here, we created a public class Main.

The Main class contains a main() method. The main() method is the entry point for the program. And, created a TreeSet collection tree and added integer elements using add() method. Then we get the subset of the TreeSet collection using the subSet() method based on the specified range and printed the result.

Java TreeSet Programs »

---