[Java] 배열 평탄화

다차원 배열을 일차원 배열로 간단하게 평탄화

• flatMap 과 재귀호출을 이용하여 평탄화하는 메소드를 정의

private static Stream<Object> flatten(Object[] array) {
    return Arrays.stream(array)
                 .flatMap(o -> o instanceof Object[]? flatten((Object[])o): Stream.of(o));
}

• 사용례
  • 다차원 배열의 경우 Arrays.deepToString(..), 일차원 배열의 경우 Arrays.toString(..) 사용

Object[] array = { 1, 2, new Object[]{ 3, 4, new Object[]{ 5 }, 6, 7 }, 8, 9, 10 };
System.out.println("original: "+Arrays.deepToString(array));

Object[] flat = flatten(array).toArray();
System.out.println("flat:     "+Arrays.toString(flat));

• int 등 특정 타입 배열로 정의한다면 mapToInt 등을 이용해 cast 할 수 있다.

int[] flatInt = flatten(array).mapToInt(Integer.class::cast).toArray();
System.out.println("flat int: "+Arrays.toString(flatInt));