Java - 2차원 리스트 평탄화 (flatten)

리스트를 요소로 갖고 있는 리스트, 2차원 리스트를 1차원 리스트로 변경하는 방법을 소개합니다. 평평하게 한다는 의미로 평탄화(flatten)라고 하기도 합니다.

1. for문으로 2차원 리스트 평탄화

아래 예제에서는 2차원 리스트를 평탄화하는 flatten() 함수를 구현하였습니다. 함수 안에서는 for문과 addAll()로 2차원 리스트를 1차원 리스트로 변환하고 있습니다.

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

public class Example {

    public static void main(String[] args) {

        List<List<String>> myList = Arrays.asList(Arrays.asList("a", "b", "c"),
                Arrays.asList("d", "e", "f"),
                Arrays.asList("h", "i", "j"));

        List<String> flattenList = flatten(myList);

        System.out.println(flattenList);
    }

    public static List<String> flatten(List<List<String>> listOfLists) {
        List<String> result = new ArrayList<>();
        for (List<String> list : listOfLists) {
            result.addAll(list);
        }
        return result;
    }
}

Output:

[a, b, c, d, e, f, h, i, j]

2. forEach()로 2차원 리스트 평탄화

forEach()를 사용하여 모든 리스트 요소를 순회하면서 addAll()로 1차원 리스트에 모든 요소를 추가하면서 평탄화할 수 있습니다.

아래 예제에서 forEach(flattenList::addAll)forEach(list -> flattenList.addAll(list))와 동일한 의미가 됩니다.

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

public class Example1 {

    public static void main(String[] args) {

        List<List<String>> myList = Arrays.asList(Arrays.asList("a", "b", "c"),
                Arrays.asList("d", "e", "f"),
                Arrays.asList("h", "i", "j"));

        List<String> flattenList = new ArrayList<>();
        myList.forEach(flattenList::addAll);

        System.out.println(flattenList);
    }
}

Output:

[a, b, c, d, e, f, h, i, j]

3. Stream과 reduce()로 2차원 리스트 평탄화

Stream과 reduce()로 아래와 같이 2차원 리스트를 1차원 리스트를 평탄화할 수 있습니다.

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

public class Example2 {

    public static void main(String[] args) {

        List<List<String>> myList = Arrays.asList(Arrays.asList("a", "b", "c"),
                Arrays.asList("d", "e", "f"),
                Arrays.asList("h", "i", "j"));

        List<String> flattenList = myList.stream().reduce(new ArrayList<>(), (x, y) -> {
            x.addAll(y);
            return x;
        });

        System.out.println(flattenList);
    }
}

Output:

[a, b, c, d, e, f, h, i, j]
Loading script...
codechachaCopyright ©2019 codechacha