list转tree

Class Test {

        private int id;

        private int pid;

        private String name;

        private List<Test> chlidren;

        // 省略 getter setter 构造方法， toString

}

class Demo {

    public static void main(String [] args) {

           Test  t1 = new Test(1, 0, "一级");
           Test  t2 = new Test(2, 1, "二级");
           Test  t3 = new Test(3, 2, "三级");
           Test  t4 = new Test(4, 2, "四级");
           Test  t5 = new Test(5, 3, "五级");
           
           List<Test> list = new ArrayList<Test>();
            
           list.add(t1);
           list.add(t2);
           list.add(t3);
           list.add(t4);
           list.add(t5);
           
           List<Test> resultList = new ArrayList<Test>();
           listToTree(list, resultList, 0);
            
           Object o = JSONObject.toJSON(resultList);
           System.out.println(o);
    }
    
    public static void listToTree(List<Test> list, List<Test> result, int pid) {
        
        for (int i = 0; i < list.size(); i ++) {
            Test item = list.get(i);
            if (item.getPid() == pid) {
                Test itemList = new Test(item.getId(), item.getPid(), item.getName(), new ArrayList<Test>());
                
                result.push(itemList);
                listToTree(list, itemList.getChlidren(), item.getId());
            }
        }        

    }


}