[C#] 프로그래머스 Lv.2 n^2 배열 자르기

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 함수를 완성해주세요.

 

 

문제 풀이

처음에는 단순히 2차원 배열을 생성하고, 이를 배열에 옮겨준 후, left와 right사이의 값을 추출하여 해결했지만, 런타임 에러로 인해 문제가 생겼다.

 

런타임 에러 코드

 

using System;

public class Solution {
 public int[] solution(int n, long left, long right)
 {
     int length = (int)(right - left + 1);

     int[] answer = new int[length];

     int[,] board = new int[n, n];

     for(int i = 0; i < n; i++)
     {
         for(int j = 0; j < n; j++)
         {
             board[i, j] = Math.Max(i, j) + 1;
         }
     }

     int[] arr = new int[n * n];

     for(int i = 0; i < n; i++)
     {
         for(int j = 0; j < n; j++)
         {
             arr[i * n + j] = board[i, j];
         }
     }

     Array.Copy(arr, (int)left, answer, 0, length);

     return answer;
 }
}

 

 

다른 풀이 방법

길이가 10^7까지 가는거보면 반복하는데 있어 엄청난 시간이 걸리고, 심지어 두 번 반복해서 생성하고 옮겨주고 하기 때문에 런타임 에러가 발생하는 것 같다.

 

문제는 2차원 배열로 풀어라고 했지만, 사실 2차원 배열을 만들지 않고도 풀 수 있는 방법이 있다.

 

반복문을 사용한다면 i인덱스를 n으로 나누면 해당 수는 행(row)이 되며, i인덱스를 n으로 나눈 나머지 수는 열(col)이 된다.

 

 

정답

   public int[] solution(int n, long left, long right)
   {
      List<int> answer = new List<int>();

       for(long i = left; i <= right; i++)
       {
           long row = i / n;
           long col = i % n;
           answer.Add((int)Math.Max(row, col) + 1);
       }


       return answer.ToArray();
   }

 

 

회고

2차원 배열을 생성하지 않고도 해당 수를 나눔과 나머지로 계산할 수 있다는 것을 알게 되었다.