TreeSet（不可重復，自動排序）實現自定義排序

當把一個對象加入TreeSet集合中時，TreeSet調用該對象的compareTo(Object obj)方法與容器中的其他對象比較大小，返回-1表示該元素在前，1表示該元素在后。

方法一：讓類實現Comparable接口，並覆蓋compareTo（）方法，將自定義的類加入TreeSet即可

 

import java.util.Scanner;
import java.util.TreeSet;
//TreeSet應用
class Main {
    public static void main(String[] args) {
        Scanner s = new Scanner(System.in);
        TreeSet<StringDemo> set = new TreeSet<>();
        while(s.hasNextLine()) {
            int n = Integer.parseInt(s.nextLine());
            for (int i = 0; i < n; i++) {
                String string = s.nextLine();
                set.add(new StringDemo(string));
            }
            for(StringDemo s1: set) {
                System.out.println(s1.getString());
            }
        }
        s.close();
    }
}
class StringDemo implements Comparable{
    private String string;
    StringDemo(String s) {
        this.string = s;
    }
    public int compareTo(Object obj) {
        StringDemo s = (StringDemo)obj;
        if(this.string.compareTo(s.string) <= 0)
            return -1;
        return 1;
    }
    public String getString() {
        return this.string;
    }
}

 

方法二：自定義一個實現Comparator的排序器，實現compare（）方法（兩個一起比較），在創建TreeSet的時候將此排序器加入即可

import java.util.Comparator;
import java.util.Scanner;
import java.util.TreeSet;
//TreeSet應用
class Main {
    public static void main(String[] args) {
        Scanner s = new Scanner(System.in);
        TreeSet<String> set = new TreeSet<>(new StrComparator());
        while(s.hasNextLine()) {
            int n = Integer.parseInt(s.nextLine());
            for (int i = 0; i < n; i++) {
                String string = s.nextLine();
                set.add(string);
            }
            for(String s1: set) {
                System.out.println(s1);
            }
            set.clear();
        }
        s.close();
    }
}
class StrComparator implements Comparator{
    public int compare(Object obj1, Object obj2) {
        String s1 = (String) obj1;
        String s2 = (String) obj2;
        if(s1.compareTo(s2) <= 0)
            return -1;
        return 1;
    }
}

 compare()方法，第一個參數為后加入的，第二個參數為已存在的（不按順序，使用的數據結構應該為樹），返回1表示參數一在后面，-1在前面，返回0參數一不加入Set