[Swift] 프로그래머스 월간 코드 챌린지 3 n^2 배열 자르기

 

Problem

 

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

---

Solution

 

1. 배열에 숫자가 채워지는 규칙을 파악한다.

 

배열에 숫자가 채워지는 규칙은 i행에 i개의 i가 채워지고 그 다음 i+1부터 n까지 채워집니다.

 

예를 들어 n이 10이고 i가 5라면 5번째 행은 5가 5개가 먼저 채워집니다. -> [5,5,5,5,5]

 

그리고 6(5+1)부터 10까지 채워집니다. -> [5,5,5,5,5,6,7,8,9,10]

 

2. 시작되는 행과 열, 끝나는 행과 열을 구한다.

 

주어진 left와 right를 통해서 숫자가 몇 번째 행,몇 번째 열에서 시작되고 끝나는지 구합니다.

 

행은 n으로 나누면 되고 열을 n으로 나눈 나머지를 구하면 됩니다.

 

    let startColumn:Int = Int(left)/n
    let endColumn:Int = Int(right)/n
    let startRow:Int = Int(left)%n
    let endRow:Int = Int(right)%n

 

3. 시작되는 행부터 열까지 1에서 구한 규칙을 적용한다.

 

먼저 numbers에 1에서 적용한 규칙대로 숫자를 채워줍니다.

 

그리고 시작 행이라면 시작 열부터 끝까지 넣어주고 끝나는 행이라면 0부터 끝나는 열까지 넣어줍니다.

 

그게 아니라면 numbers를 넣어줍니다.

 

 var answer:[Int] = []

    for i in startColumn...endColumn {
        var numbers:[Int] = Array(repeating: i+1, count:i+1)
        if i+1 < n {
            let tail:[Int] = Array(i+2...n)
            numbers.append(contentsOf:tail)
        }

        if startColumn == endColumn {
            return Array(numbers[startRow...endRow])
        }

        if i == startColumn {
            answer.append(contentsOf: Array(numbers[startRow...]))
            continue
        }

        if i == endColumn {
            answer.append(contentsOf: Array(numbers[0...endRow]))
            continue
        }
        answer.append(contentsOf: numbers)
    }

 

4. 규칙대로 숫자를 넣은 answer를 반환한다.

 

return answer

---

Source Code

 

---

P.S

 

다른 사람 풀이를 보니 그냥 left부터 right까지 행과 열을 구해서 차례로 넣으면 됐는데... 좀 어렵게 생각했다.

 

func solution(_ n:Int, _ left:Int64, _ right:Int64) -> [Int] {
    return (Int(left)...Int(right)).map { max($0 / n, $0 % n) + 1 }
}