[프로그래머스] n^2 배열 자르기 - Java

https://programmers.co.kr/learn/courses/30/lessons/87390

코딩테스트 연습 - n^2 배열 자르기

---

문제

문제 설명

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.

1. n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
2. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
   • 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.
3. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
4. 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.

정수 n, left, right가 매개변수로 주어집니다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• 1 ≤ n ≤ 107
• 0 ≤ left ≤ right < n2
• right - left < 105

---

입출력 예

n	left	right	result
3	2	5	[3,2,2,3]
4	7	14	[4,3,3,3,4,4,4,4]

---

풀이

풀이 과정

구해야 하는 1차원 배열의 크기는 결국 right값 - left값 + 1만큼이므로 해당 크기만큼의 배열을 선언하고 그 부분의 값만 탐색하여 시간복잡도를 줄이는 방향으로 풀이

int[] answer = new int[(int)(right - left) + 1];

 

i = 0부터 answer 배열의 크기만큼 반복하며 실제 2차원 배열을 만들지 않아도 2차원 배열처럼 탐색이 가능하다

n X n 배열일 때, ( i / n ) + 1 은 행의 값이 되고 ( i % n ) + 1 은 열의 값이 된다

문제에서는 left와 right사이의 값만 필요로 하므로 i 대신 i + left값을 이용하고 left가 long 타입이기 때문에 계산 결과를 int형으로 변환해준다

for(int i = 0; i < answer.length; i++){
    int row = (int)((i + left) / n) + 1; // 행
    int col = (int)((i + left) % n) + 1; // 열
}

 

이제 정답배열의 원소마다 값을 입력해 주어야 하는데, 3 X 3 배열을 예로 각 원소의 값을 살펴보면

( 행, 열 ) = 값 일때

( 1, 1 ) = 1    ( 1, 2 ) = 2    ( 1, 3 ) = 3
( 2, 1 ) = 2    ( 2, 2 ) = 2    ( 2, 3 ) = 3
( 3, 1 ) = 2    ( 3, 2 ) = 3    ( 3, 3 ) = 3

원소의 값은 해당 원소의 행과 열 값중 큰 값을 따라가는 것을 볼 수 있다

따라서 정답 배열의 i 인덱스에는 현재 행과 열 중 큰 값을 넣어주면 된다

answer[i] = Math.max(row, col);

 

최종 코드

import java.util.*;

class Solution {
    public int[] solution(int n, long left, long right) {
        int[] answer = new int[(int)(right - left) + 1];
        
        for(int i = 0; i < answer.length; i++){
            int row = (int)((i + left) / n) + 1; // 행
            int col = (int)((i + left) % n) + 1; // 열
            answer[i] = Math.max(row, col);
        }
        
        return answer;
    }
}

---