【Groovy】集合遍历 ( 集合中有集合元素时调用 flatten 函数拉平集合元素 | 代码示例 )

    /**
     * 拉平集合。此集合和任何嵌套数组或集合的内容（递归地）添加到新集合中。
     * <pre class="groovyTestCase">assert [1,2,3,4,5] == [1,[2,3],[[4]],[],5].flatten()</pre>
     *
     * @param self 要拉平的集合
     * @return 返回一个新集合 , 该集合中的元素都被拉平了
     * @since 1.6.0
     */
    public static Collection<?> flatten(Collection<?> self) {
        return flatten(self, createSimilarCollection(self));
    }

class Test {
    static void main(args) {

        // 为 ArrayList 设置初始值
        def list = ["1", "2", "3"]


        // I. 集合的 " << " 操作符重载 , 添加元素


        // 操作符重载 " << " 操作符相当于调用 leftShift 方法
        def list2 = list << "4"

        // 打印 [1, 2, 3, 4]
        println list
        // 打印 [1, 2, 3, 4]
        println list2


        // II. 集合的 " << " 操作符重载 , 添加集合


        // 操作符重载 " << " 操作符相当于调用 leftShift 方法
        list2 = list << ["5", "6"]

        // 打印 [1, 2, 3, 4, [5, 6]]
        println list
        // 打印 [1, 2, 3, 4, [5, 6]]
        println list2


        // III. 调用集合的 flatten 方法拉平集合


        // 拉平集合
        list2 = list.flatten();
        // 打印 [1, 2, 3, 4, [5, 6]]
        println list
        // 打印 [1, 2, 3, 4, 5, 6]
        println list2
    }
}

[1, 2, 3, 4]
[1, 2, 3, 4]
[1, 2, 3, 4, [5, 6]]
[1, 2, 3, 4, [5, 6]]
[1, 2, 3, 4, [5, 6]]
flatten : [1, 2, 3, 4, 5, 6]