Table of Contents
In this post, we will see TreeSet in java.
Java TreeSet have following properties:
2) create TreeSetMain.java as below:
When you run above program, you get below out:
Java TreeSet have following properties:
- It can contain only unique element.
- It stores objects in ascending order by default,
- It implements NavigableSet interface which extends SortedSet.
- When you put objects in TreeSet, it must implement Comparable interface.
Lets understand it more with the help of example.
Example:
1) create a class named Country.java.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
package com.org.arpit.java2blog; public class Country implements Comparable{ String countryName; public Country(String countryName) { super(); this.countryName = countryName; } @Override public int compareTo(Object arg0) { Country country=(Country) arg0; return (this.countryName.compareTo(country.countryName) ) ; } public String getCountryName() { return countryName; } public void setCountryName(String countryName) { this.countryName = countryName; } } |
2) create TreeSetMain.java as below:
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
package com.org.arpit.java2blog; import java.util.Iterator; import java.util.TreeSet; public class TreeSetMain { /** * @author Arpit Mandliya */ public static void main(String[] args) { Country indiaCountry=new Country("India"); Country chinaCountry=new Country("China"); Country nepalCountry=new Country("Nepal"); Country bhutanCountry=new Country("Bhutan"); Country indiaCountry2=new Country("India"); Country nepalCountry2=new Country("Nepal"); TreeSet countryTreeSet = new TreeSet(); countryTreeSet.add(indiaCountry); countryTreeSet.add(chinaCountry); countryTreeSet.add(nepalCountry); countryTreeSet.add(bhutanCountry); countryTreeSet.add(indiaCountry2); countryTreeSet.add(nepalCountry2); Iterator counIter=countryTreeSet.iterator(); // put debug point here while(counIter.hasNext()) { System.out.println(counIter.next().countryName); } } } |
|
1 2 3 4 5 6 |
Bhutan China India Nepal |
As you can see, we have added 6 country objects in Treeset but we have only 4!!. As TreeSet can only have unique elements,indiaCountry2 and nepalCountry2 did not add to the TreeSet. Explanation:
When you will put indiaCountry2 in treeSet it will compare this object with indiaCountry as indiaCountry2.compareTo(indiaCountry) and compareTo method will return 0. Both object will treated as equal. As you can not add duplicate element in TreeSet, indiaCountry2 will not be added to above TreeSet.
Now put debug point at line 27 and right click on project->debug as-> java application. Program will stop execution at line 27 then right click on countryTreeSet then select watch.You will be able to see structure as below.
Tricky question on TreeSet:Â
Guess output of below program:
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
package com.org.arpit.java2blog; import java.util.Iterator; import java.util.TreeSet; public class Employee implements Comparable { public String name; public int compareTo(Object o) { return 0; } public static void main(String args []) { Employee employeeOne = new Employee(); Employee employeeTwo = new Employee(); employeeOne.name= "John"; employeeTwo.name= "Martin"; TreeSet employeeSet = new TreeSet(); employeeSet.add(employeeOne); employeeSet.add(employeeTwo); Iterator empIt=employeeSet.iterator(); while(empIt.hasNext()) { System.out.println(empIt.next().name); } } } |
What will be output of above program:
A. Martin
B. John
C. John
Martin
D. Compilation fails.
E. The code runs with no output.
F. An exception is thrown at runtime.
Correct Answer is B here.
Explanation:
As you can see we have overridden compareTo method in Employee class and always return 0.
Following steps will take place:
Following steps will take place:
- First element with “John” will be added to employeeSet.
- When we will added second element with martin, compareTo method will get called with employeeOne.compareTo(employeeTwo) and it will return 0.
- As compareTo method returns 0, employeeOne is equals to employeeTwo, so employeeTwo will not be added to treeSet.
- So output of above program is “John”
Was this post helpful?
Let us know if this post was helpful. Feedbacks are monitored on daily basis. Please do provide feedback as that\'s the only way to improve.
