출처: https://school.programmers.co.kr/learn/courses/30/lessons/87390
문제 설명
정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.
n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.
1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
새로운 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
다음 애니메이션은 주어진 과정대로 1차원 배열을 만드는 과정을 나타낸 것입니다.
ex1
입출력 예 #2
다음 애니메이션은 주어진 과정대로 1차원 배열을 만드는 과정을 나타낸 것입니다.
약 20분간 짠 코드문이다.
class Solution {
public int[] solution(int n, long left, long right) {
int size = (int)(right - left + 1);
int[] answer = new int[size];
// n행 n열 크기의 비어있는 2차원 배열
// arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.
int arr[] = new int[n*n - 1];
// index left~ right 까지 잘라서 answer 배열에 저장
for(long i = left; i <= right; i++){
int index = (int)(i - left);
answer[index] = (int)(Math.max(i / n, i % n) + 1);
}
return answer;
}}
코드 실행결과 두 테스트 케이스가 통과 되었다.
테스트 1
입력값 〉 3, 2, 5
기댓값 〉 [3, 2, 2, 3]
실행 결과 〉 테스트를 통과하였습니다.
테스트 2
입력값 〉 4, 7, 14
기댓값 〉 [4, 3, 3, 3, 4, 4, 4, 4]
실행 결과 〉 테스트를 통과하였습니다.
하지만, 채점시 55점 밖에 안 맞았다. 왜그럴까.
실질적으로
int arr[] = new int[n*n - 1]; 배열은 필요가 없고,
직접 만들어줘야 하는것이다.
그외에 long 타입을 int 타입으로 형변환 해줘야 반복문이 돌아간다는 점도 고려해야 했다.
최종적으로 바꾼 코드문
class Solution {
public int[] solution(int n, long left, long right) {
int size = (int)(right - left + 1);
int[] answer = new int[size];
// n행 n열 크기의 비어있는 2차원 배열
// arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.
int arr[] = new int[n*n - 1];
// index left~ right 까지 잘라서 answer 배열에 저장
for(long i = left; i <= right; i++){
int index = (int)(i - left);
answer[index] = (int)(Math.max(i / n, i % n) + 1);
}
return answer;
}}
그외에 다른 사람의 코드문
import java.util.*;
public class Solution {
public static int[] solution(int n, long left, long right) {
int length = (int)(right - left) + 1;
int[] result = new int[length];
int idx = 0;
for (long i = left; i <= right; i++) {
long row = i / n + 1;
long col = i % n + 1;
result[idx++] = (int) Math.max(row, col);
}
return result;
}}
