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

프로그래머스

---

풀이

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

 

전체 배열을 만들어 left와 right만큼 잘라 return 한다면 시간초과가 발생한다.

따라서 전체 배열을 만들지 않고, left,right만큼의 배열만 생성해 return 해야 하는데

2차원배열의 좌표를 활용해 원하는 값을 찾을 수 있었다.

 

(0,0) (0,1) (0,2)

(1,0) (1,1) (1,1)

(2,0) (2,1) (2,2)

 

이런 식으로 2차원 배열이 있다면

x, y 중 큰 값 + 1을 하면 해당 좌표에 문제에서 원하는 값을 찾을 수 있다.

 

1 2 3

2 2 3

3 3 3

---