黑马程序员技术交流社区

标题: TreeSet小练习 [打印本页]

---

作者: 小p    时间: 2015-7-25 21:38
标题: TreeSet小练习
import java.util.*;

/*
需求：按字符串长度进行排序

分析：字符串本身具备比较性，但不是功能所需要的，所以需要用比较器

*/

class  TreeSetTest
{
        public static void main(String[] args)
        {
                TreeSet ts = new TreeSet(new StrLenComparator());

                ts.add("abcd");
                ts.add("cc");
                ts.add("cba");
                ts.add("aaa");
                ts.add("z");
                ts.add("hahaa");

                Iterator it = ts.iterator();

                while(it.hasNext())
                {
                        System.out.println(it.next());

                }

        }

}

class StrLenComparator implements Comparator
{
        public int compare(Object o1,Object o2)
        {
                String s1= (String)o1;
                String s2= (String)o2;

                /*
                if(s1.length()>s2.length())
                        return 1;
                if(s1.length()==s2.length())
                        return 0;
                        */
                int num= new Integer(s1.length()).compareTo(new Integer (s2.length()));
                if(num==0)
                        return s1.compareTo(s2);
                return -1;
        }
}

---

欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2