글 읽기 - swift 시간초과 질문

ipkjw2 1년 전 0

시간초과가 나는데 왜 나는지 도와주세요...

class MinHeap {
    var heap: [Int]
    
    init() {
        heap = [0]
    }
    
    func pop() -> Int {
        if heap.count == 1 {
            return 0
        }
        
        let result = heap[1]
        var location = 1
        var next: Int
        var tmp: Int
        heap[1] = heap[heap.count - 1]
        heap.removeLast()
        
        while location * 2 < heap.count {
            
            if location * 2 == heap.count - 1 {
                next = location * 2
            } else {
                next = heap[location * 2] > heap[location * 2 + 1] ? location * 2 + 1 : location * 2
            }
            
            if heap[location] > heap[next] {
                tmp = heap[location]
                heap[location] = heap[next]
                heap[next] = tmp
                location = next
            } else {
                break
            }
        }
        
        return result
    }
    
    func push(num: Int) {
        var location = heap.count
        heap.append(num)
        
        while location > 1 {
            if heap[location / 2] > heap[location] {
                heap[location] = heap[location / 2]
                location /= 2
            } else {
                break
            }
        }
        heap[location] = num
    }
}

let heap = MinHeap()
var result = ""
    
let N: Int = Int(readLine()!)!
var x: Int
    
for _ in 0..<N {
    x = Int(readLine()!)!
        
    if x == 0 {
        result.write("\(heap.pop())\n")
    } else {
        heap.push(num: x)
    }
}
    
print(result)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

swift 시간초과 질문

ipkjw2 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말