Question

我正在尝试学习Java Stream API，并且正在编写一些示例。

所以我的示例如下：

我有一个列表列表，每个列表可以包含许多节点。

我想要一个程序来检查并返回满足某些条件的节点。

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Main {
        public static void main( String[] args ) {
        ArrayList<ArrayList<Node>> lists =  new ArrayList<>();
        /*here i'm creating a list of 10 list from 0 to 9
        * and each list will have one node, the nodes will have a random 
        degree
        */
        IntStream.range(0,10).forEach(  index -> {
                                                lists.add(new ArrayList<>());
                                                int random=new Random().nextInt(10) + 1;
                                                lists.get(index).add(new Node(random));
        });

        Node specificLsit = getaSpecificNode(lists);
    }

   /*we chould retun a new Node(1) if there is a node that have a degree=1
    *and a new Node(2) id there is a node with degree= 2 
    *or null if the above condition fails.
    *so what is the java stream operation to writre for that purpose.
    */
    private static Node getaSpecificNode( ArrayList<ArrayList<Node>> lists ) {
        Node nodeToReturn =null;
        //code go here to return the desired result
        return nodeToReturn;
    }
}

class Node{
    int degree;

    Node(int degree){
        this.degree = degree;
    }
    @Override
    public String toString() {
        return this.degree+"";
    }
}

2 for循环很容易解决问题，但是我想要一个使用流api的解决方案。

我尝试过的：

 private static Node getaSpecificNode( ArrayList<ArrayList<Node>> lists ) {
    Node nodeToReturn =null;
    lists.forEach((list)->{
        list.forEach((node)->{
            if (node.degree ==1 || node.degree ==2 )
                nodeToReturn = node;

        });

    });
    return nodeToReturn ;
}

不幸的是，我收到一个编译错误，变量nodeToReturn应该是最终的，但就我而言，我试图对其进行修改。

有更好的解决方案吗？

Answer 1

这应该可以解决问题：

lists.stream().flatMap(List::stream).filter(e -> e.getDegree() == 1 || e.getDegree() == 2)
              .findAny()
              .orElse(null);

在这里，我们将ArrayList<ArrayList<Node>>转换为flatMap，然后根据您的情况应用过滤器。如果找到匹配项，则返回Node，否则返回null。

java流操作检查并返回特定对象

1 个答案: