[Algorithm] Swift 뒤에 있는 큰 수 찾기 문제 풀이 [프로그래머스 Level2]

💬 문제

코딩테스트 연습 - 뒤에 있는 큰 수 찾기

---

💬 Idea

1. 첫번째 아이디어

기준 인덱스 + 1 부터 끝까지 완전탐색을 하여 기준 인덱스보다 큰 수를 발견하면 answer에 append하고 break하여 이중 포문을 탈출하자.

→ 하지만, 시간초과가 발생했다.

 

2. 두번째 아이디어

stack을 사용하여 저장하고 stack의 마지막 원소를 pop해가면서 기준 인덱스와 비교하자.

→ 인덱스를 같이 저장하기 위해 stack을 튜플 형태로 저장할 수 있도록 하였다.

---

💬 풀이

실패한 풀이 - 완전탐색 진행

→ 테스트케이스 20, 21, 22, 23 시간초과 발생

import Foundation

func solution(_ numbers:[Int]) -> [Int] {
    var ans: [Int] = []
    
    for i in 0..<numbers.count - 1 {
        for j in i + 1..<numbers.count {
            if numbers[i] < numbers[j] {
                ans.append(numbers[j])
                break
            }
        }
        
        if ans.count == i {
            ans.append(-1)
        }
    }
    
    ans.append(-1)
    return ans 
}

 

 

성공한 풀이 - stack 사용

import Foundation

func solution2(numbers:[Int]) -> [Int] {
    var ans = Array(repeating: -1, count: numbers.count)
    var stack: [(Int, Int)] = [(0, numbers[0])] // 인덱스를 같이 저장하기 위해 튜플 형태로 저장
    
    for i in 1..<numbers.count {
        // stack이 비어있지 않을때까지 반복
        while !stack.isEmpty {
            // stack의 마지막 value가 기준값보다 크다면 break 
            if stack.last!.1 >= numbers[i] { break }
            // stack의 마지막원소를 지우고 해당 인덱스에 큰 수 할당
            ans[stack.removeLast().0] = numbers[i]
        }
        // stack에 현재 기준값 저장
        stack.append((i, numbers[i]))
    }
    
    return ans
}

 

 

 

https://github.com/hwangJi-dev/swiftAlgorithm/blob/main/Programmers/Programmers/level2/%EB%92%A4%EC%97%90%20%EC%9E%88%EB%8A%94%20%ED%81%B0%20%EC%88%98%20%EC%B0%BE%EA%B8%B0.swift

GitHub - hwangJi-dev/swiftAlgorithm: Algorithm Practice with Swift