TreeSet练习

用TreeSet存储以下数据：
“xiaoqiang”、 “zhangsan”、“lisi”、“xiaohua”、 “ruhua”、 “wangcai”
要求按照字符串的长度进行排序后再存储（如果长度相同，则按字符串的自然顺序排序）

package test3_31.t1;

import java.util.Comparator;
import java.util.TreeSet;

/*2. 用TreeSet存储以下数据：
“xiaoqiang”、 “zhangsan”、“lisi”、“xiaohua”、 “ruhua”、 “wangcai”
要求按照字符串的长度进行排序后再存储（如果长度相同，则按字符串的自然顺序排序）
*/
public class test1 {
    public static void main(String[] args) {
        //创建TreeSet
        TreeSet<String> treeSet = new TreeSet<>(new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                //按照长度排序
                int i = o1.length() - o2.length();
                //长度相同，按照字符串原排序
                return i!=0?i:o1.compareTo(o2);
            }
        });
        treeSet.add("xiaoqiang");
        treeSet.add("zhangsan");
        treeSet.add("lisi");
        treeSet.add("xiaohua");
        treeSet.add("wangcai");
        treeSet.add("ruhua");
        System.out.println(treeSet);
    }
}