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

코드

import Foundation

func solution(_ n:Int, _ left:Int64, _ right:Int64) -> [Int] {
    var answer = [Int]()
    
    // index 값을 i라고 하자
	// i/n + 1 = 그 행의 최소 값을 나타내 준다
	// i%n + 1 = 그 자리 값을 나타내 준다
    // 두개 중 max값을 구하면 우리가 구하고 싶은 숫자가 나오는 규칙이다
    for i in Int(left)...Int(right) {
        answer.append(max(i/n + 1, i%n + 1))
    }
    
    return answer
}

회고

• 처음에 중첩 for문으로 찾아서 시간 초과가 떴다
• 규칙을 찾는데 시간이 너무 오래 걸렸다
• 종이에 적으면서 풀어야 더 빨리 생각날 것 같다