import java.util.Comparator;
import java.util.TreeMap;
public class Test {
public static void main(String[] args)
{
System.out.println("使用自然排序:(鍵類實作了compareTo接口)");
TreeMap<Id, String> treeMap=new TreeMap<Id,String>();
treeMap.put(new Id(1000), "demo1");
treeMap.put(new Id(1001), "demo2");
treeMap.put(new Id(1002), "demo3");
treeMap.put(new Id(1003), "demo4");
treeMap.put(new Id(1004), "demo5");
System.out.println("treeMap="+treeMap);
//使用釘子排序,指明比較器
System.out.println("使用定制排序:(從小到大排序)");
TreeMap<Id, String> treeMap2=
new TreeMap<Id,String>(new MyComparatorBigToSmall());
treeMap2.put(new Id(1000), "demo1");
treeMap2.put(new Id(1001), "demo2");
treeMap2.put(new Id(1002), "demo3");
treeMap2.put(new Id(1003), "demo4");
treeMap2.put(new Id(1004), "demo5");
System.out.println("treeMap2="+treeMap2);
System.out.println("使用定制排序:(從大到小排序)");
TreeMap<Id, String> treeMap3=
new TreeMap<Id,String>(new MyComparatorSmallToBig());
treeMap3.put(new Id(1000), "demo1");
treeMap3.put(new Id(1001), "demo2");
treeMap3.put(new Id(1002), "demo3");
treeMap3.put(new Id(1003), "demo4");
treeMap3.put(new Id(1004), "demo5");
System.out.println("treeMap3="+treeMap3);
}
}
//自定義比較器:實作comparator接口
class MyComparatorBigToSmall implements Comparator<Id> //使用泛型
{
@Override
public int compare(Id o1, Id o2) {
return o1.getIdNumber()-o2.getIdNumber();//從小到大排序
}
}
class MyComparatorSmallToBig implements Comparator<Id>
{
@Override
public int compare(Id o1, Id o2) {
return o2.getIdNumber()-o1.getIdNumber();//從大到小排序
}
}
//Id作為鍵的類實作Comparable接口
class Id implements Comparable<Id>
{
//成員屬性
private int IdNumber;
public int getIdNumber()
{
return IdNumber;
}
//構造函數
public Id() {
// TODO Auto-generated constructor stub
}
//構造函數
public Id(int IdNumber)
{
this.IdNumber=IdNumber;
}
//重寫toString()方法
@Override
public String toString() {
return "["+this.IdNumber+"]";
}
//重寫equals()方法
@Override
public boolean equals(Object obj)
{
if(obj==this)//自己和自己比較
return true;//那就是同一個相等了
//傳入的引用不為空,而且兩個的類型一樣
if(obj!=null&&obj.getClass()==this.getClass())
{
Id idNumber2=(Id) obj;//類型轉換
return this.IdNumber==idNumber2.IdNumber;
//如果兩個的編号一樣,那就認為相等
}
return false;
}
//重寫hashCode()方法
@Override
public int hashCode() {
return this.IdNumber*100;
}
//重寫compareTo()方法實作自然排序
@Override
public int compareTo(Id o)
{
//return this.IdNumber-o.IdNumber;//從小到大排序:
return o.IdNumber-this.IdNumber;//從大到小排序
}
}
java之TreeMap :自然排序和定制排序