用TreeSet存储以下数据

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

public class Ex2 {
public static void main(String[] args) {
TreeSet set = new TreeSet<>(new Comparator() {
@Override
public int compare(String o1, String o2) {
if (o1.length() > o2.length()) {
return 1;
} else if (o1.length() < o2.length()) {
return -1;
} else {
return o1.compareTo(o2);
}
}
});
set.add(“xiaoqiang”);
set.add(“zhangsan”);
set.add(“lisi”);
set.add(“xiaohua”);
set.add(“ruhua”);
set.add(“wangcai”);
for (String string : set) {
System.out.println(string);
}
}
}