TreeSet的使用和底层逻辑

最新推荐文章于 2025-07-03 08:15:01 发布

原创最新推荐文章于 2025-07-03 08:15:01 发布 · 208 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

本文介绍了如何使用Java的TreeSet集合类，并重点讲解了如何通过实现Comparable接口来保证元素排序。通过实例演示了如何创建并操作包含整数和自定义Emp2类对象的TreeSet。

package TreeSet;
import java.util.Set;
import java.util.TreeSet;
//TreeSet如果加类的话 必须实现Comparable的接口
public class TestTreeSet {
    public static void main(String[] args) {
        Set<Integer>set=new TreeSet<>();
        set.add(300);
        set.add(200);
        set.add(600);
        //按照元素递增的方式排好序
        for(Integer m:set){
            System.out.println(m);
        }
        Set<Emp2>set2=new TreeSet<>();
        set2.add((new Emp2(100,"张三",3000)));
        set2.add((new Emp2(50,"李四",2000)));
        set2.add((new Emp2(150,"王五",8000)));
        for(Emp2 m:set2){
            System.out.println(m);
        }

    }
}
class Emp2 implements  Comparable<Emp2>{//实现Comparable接口
    int id;
    String name;
    double salary;
    public Emp2(int id,String name,double salary){
        this.id=id;
        this.name=name;
        this.salary=salary;

    }
    public String toString(){
        return "id:"+id+"name:"+name+",salary"+salary;
    }

    @Override
    public int compareTo(Emp2 o) {//复数小于 0等于 正数大于
        if(this.salary>o.salary){
            return 1;
        }else if(this.salary<o.salary){
            return -1;
        }else{
            if(this.id>o.id){
                return 1;
            }else if(this.id<o.id){
                return -1;
            }else{
                return 0;
            }
        }

    }
}