[Java] 2차원 배열을 1차원 배열로 반환하는 함수

static int[] arrayTwoOfArrayOne(int[][] arr) {
    // return 할 일차원 배열 선언
    int[] nArray = null;
    
    // 순환을 돌면서 파라미터 배열의 길이(전체 요소 수)를 구하고,
    int total_length = 0;
    for (int i=0; i<arr.length; i++) { // 행의 개수만큼 반복
        total_length += arr[i].length; // 각 행의 길이를 합산
    }
    
    // return 할 일차원 배열 크기 초기화
    nArray = new int[total_length];

    // 다시 순환을 돌면서 배열을 넣는다.
    int w = 0; // 인덱스 번호 추적을 위한 변수
    for (int i=0; i<arr.length; i++) { // 행의 개수만큼
        for (int j=0; j<arr[i].length; j++) { // 열의 개수만큼
            int temp = arr[i][j];
            nArray[w] = temp;
            w++;
        }
    }
    return nArray;
}

 

함수 호출부

 

 

실행결과