프로그래머스 - 알고리즘 탑 (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으로 표시합니다.

입출력 예

heightsreturn

[6,9,5,7,4]	[0,0,2,2,4]
[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]

입출력 예 설명

입출력 예 #1
앞서 설명한 예와 같습니다.

입출력 예 #2

[1,2,3] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[4,5,6] 번째 탑이 쏜 신호는 3번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.

입출력 예 #3

[1,2,4,5] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[3] 번째 탑이 쏜 신호는 2번째 탑이 수신합니다.
[6] 번째 탑이 쏜 신호는 5번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.

---

이번 문제에는 따로 공식을 원하는 것 같지는 않아요?!

 

그래서 딱히 좋은 방법이 떠오르지않아 for문을 2번 사용해서 그냥 풀어봤습니다.

 

 

문제에서 요구하는 것은 Int타입 배열이 들어오면

맨 마지막 자리에서 부터 왼쪽으로 연산을 진행해서 자기보다 큰값을 찾는다!? 그런 내용입니다.

단! 1번 찾았다면 더이상 연산을진행하지 않고 다음 자리로 이동해서 동일한 방법을 진행하면 됩니다.

 

1. Int타입 배열을 맨 마지막 인덱스 부터 0인덱스 까지 연산 진행
2. 자신보다 큰 값을 찾았다면 연산 종료
3. 1번 2번 내용을 0인덱스 까지 진행

 

네 간단하조. 리턴할 값 또한 Int타입 배열을 원하니 초기화를 모두 0으로 선언할게요.

var resultArray = Array(repeating: 0, count: heights.count)

 

그리고 for문을 진행하는데 stride함수를 이용해서 맨 뒷자리부터 거꾸로 연산을 진행합니다.

 for indexA in stride(from: heights.count - 1, through: 0, by: -1) {
 	.........
 }

 

그리고 for문 내부에서 이제 비교할 대상을 찾을 for문을 하나더 만듭니다.

for indexB in stride(from: indexA, through: 0, by: -1) {
	........
}

 

그리고 이제 두값을 비교해서 자기보다 크다면 미리 선언해둔 Array에 추가해주고 break를 이용해서 다음 대상에 대한 연산을 진행 합니다.

 if heights[indexA] < heights[indexB] {
	resultArray[indexA] = indexB + 1
	break
}

indexB + 1을 해주는 이유는 Array배열은 0부터 시작이라 처음에 height.count -1을 해주었기 때문입니다.?

 

 

 

아래는 전체 코드입니다.

func solution(_ heights:[Int]) -> [Int] {
    var resultArray = Array(repeating: 0, count: heights.count)
    
    for indexA in stride(from: heights.count - 1, through: 0, by: -1) {
        for indexB in stride(from: indexA, through: 0, by: -1) {
            if heights[indexA] < heights[indexB] {
                resultArray[indexA] = indexB + 1
                break
            }
        }
    }
    
    return resultArray
}

index A(6)= 2, index B(5) = 7
index A(5)= 7, index B(2) = 9
index A(4)= 5, index B(2) = 9
index A(3)= 3, index B(2) = 9
[0, 0, 0, 3, 3, 3, 6]