프로그래머스 탑 Swift

문제 설명

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.

예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

송신 탑(높이)수신 탑(높이)

5(4)	4(7)
4(7)	2(9)
3(5)	2(9)
2(9)	-
1(6)	-

맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.

제한 사항

• heights는 길이 2 이상 100 이하인 정수 배열입니다.
• 모든 탑의 높이는 1 이상 100 이하입니다.
• 신호를 수신하는 탑이 없으면 0으로 표시합니다.

입출력 예

 

heights	return
[3,9,9,3,5,7,2]	[0,0,0,3,3,3,6]
[1,5,3,6,7,6,5]	[0,0,2,0,0,5,6]
[6,9,5,7,4]	[0,0,2,2,4]

풀이:height의 오른쪽부터 왼쪽으로 차례대로 가야하기 때문에 index를 역순으로 돌려야했다. 

for in stirde문으로 heights의 길이의 -1부터 -1까지 돌렸는데 0이 들어가야하기 때문에 -1까지 돌렸다.

그리고 for문을 한 개 더써서 j를 i부터 -1까지 돌리는데 heights[j]가 heights[i]보다 크다면 바로 answer배열에 추가하고

바깥 포문을 바로 다음거를 하도록 continue outer를 써주었다.

그리고 마지막으로 배열이 차례대로 추가되므로 answer를 역순으로 나타내게 reversed를 써주었다.

import Foundation
 
func solution(_ heights:[Int]) -> [Int] {
    var answer = [Int]()
    outer:for i in stride(from: heights.count-1, to: -1, by: -1){
        for j in stride(from: i, to: -1, by: -1){
            if heights[i] < heights[j]{
                answer.append(j+1)
                continue outer
            }
        }
            answer.append(0)
    }
    return answer.reversed()
}

다른사람 풀이 중 가장 좋다고 생각하는 것

for문을 역순으로 돌리려면 reversed를 저렇게 쓰면 됨.

import Foundation
 
func solution(_ heights:[Int]) -> [Int] {
    var answer = [Int](repeating: 0, count: heights.count)
    var size = heights.count
 
    for i in (1..<size).reversed() {
        for j in (0..<i).reversed() {
            if heights[i] < heights[j] {
                answer[i] = j + 1
                break
            }
        }
    }
    return answer
}