递归2.0

import java.util.ArrayList;
import java.util.List;

public class Tree {

    /**
     *
     * @param userList 需要递归的参数
     * @return
     */
    public static List<User> first(List<User> userList){
        List<User> users = new ArrayList<>();
        userList.stream().forEach(str ->{
            //当前为父节点
            if (str.getPid() == null){
                then(str,userList);
                users.add(str);
            }
        });
        return users;
    }

    /**
     * 递归方法
     * @param user 父节点
     * @param list 集合参数
     */
    public static void then(User user,List<User> list){
        List<User> users = new ArrayList<>();
        list.stream().forEach(str ->{
            if (str.getPid() == user.getId()){
                users.add(str);
                then(str,list);
            }
        });
        user.setList(users);
    }
}