TreeSet用法

/*
TreeSet的元素要具有可比性,如果沒有具備,可以通過兩種方式添加:
1.實現Comparable接口,然後重寫其中的compareTo()
2.創建一個類,實現Comparator接口,然後重寫裏面的compare(),
然後用此類創建對象作爲TreeSet的參數傳入;
注意:如果兩個方法都存在,方法二有效;
*/
package treeSetDemo3;
import java.util.*;
class TreeSetDemo3 
{
	public static void main(String[] args) 
	{
		TreeSet ts=new TreeSet(new MyCompare());
		ts.add(new Person("007",21));
		ts.add(new Person("005",23));
		ts.add(new Person("004",45));
		ts.add(new Person("009",12));
		Iterator it=ts.iterator();
		while (it.hasNext())
		{
			Person p=(Person)it.next();
			System.out.println(p.getName()+"------"+p.getAge());
		}
	}
}
//Person類,實現Comparable接口,重寫compareTo();
class Person implements Comparable
{
   private String name;
   private int age;
   Person(String name,int age)
   {
	   this.name=name;
	   this.age=age;
   }
   //獲取姓名
   public String getName()
   {
	   return name;
   }
   //獲取年齡
   public int getAge()
   {
	   return age;
   }
   //重寫compareTo()
   public int compareTo(Object obj)
   {
	   if(!(obj instanceof Person))
		   throw new RuntimeException("不是Person的對象");
	   Person p=(Person)obj;
	   if(this.age>p.age)
        return 1;
	   else if(this.age==p.age)
	   {
		   //Stirng 類中已經實現Comparable接口
		   return this.name.compareTo(p.name);
	   }
	   return -1;
   }
}
//Comparator類
class MyCompare implements Comparator
{
	public int compare(Object obj1,Object obj2)
	{
		Person p=(Person)obj1;
		Person p1=(Person)obj2;
		return p.getName().compareTo(p1.getName());
	}
}

發佈了69 篇原創文章 · 獲贊 1 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章