글 읽기 - swift 위상정렬 알고리즘 왜 틀렸을까요 ㅠ 7프로까지 맞습니다.

lgw9898 1년 전 0

간단한 위상정렬 알고리즘이라서 풀었는데 틀렸다고 나오네요 ㅠ

import Foundation
// 0: 학생수 1: 주어지는 정보 반복 수
        let input = readLine()!.split(separator: " ").map { Int($0)! }
        
        var graph = [Int: [Int]]() // 비교하는 학생에 대한 정보
        
        var indegree = [Int](repeating: 0, count: input[0]+1) // 진입차수 정보
        indegree[0] = -1 // index를 학생번호로 쓰기 위함.
        
        for _ in 0..<input[1] {
            let info = readLine()!.split(separator: " ").map { Int($0)! }
            
            if graph[info[0]] == nil {
                graph[info[0]] = [info[1]]
            } else {
                var temp = graph[info[0]]!
                temp.append(info[1])
                graph[info[0]] = temp
            }
            
            indegree[info[1]] += 1
        }
        
//        print(graph)
//        print(indegree)
        
        var queue = [Int]()
        indegree.enumerated().forEach {
            if $0.element == 0 {
                queue.append($0.offset)
            }
        }
//        print(queue)
        
        
        var result = [Int]()
        // 큐가 빌때까지
        while !queue.isEmpty {
            let now = queue.removeFirst()
            result.append(now)
            
            // 해당원소와 연결된 노드들의 진입차수에서 -1
            if let sequence = graph[now] {
                for i in sequence{
                    indegree[i] -= 1
                    
                    if indegree[i] == 0 {
                        queue.append(i)
                    }
                }
            }
            
        }
        
        print(result)

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

문제

문제

출처

ICPC

swift 위상정렬 알고리즘 왜 틀렸을까요 ㅠ 7프로까지 맞습니다.

lgw9898 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말