글 읽기 - 코드를 수정하였지만... 여전히 틀렸다고 나오는... 반례 부탁드립니다!

hans081545 3년 전 0

어제 같은 문제로 답변을 받고 수정 하였는데요

수정 이후 8%정도 가다가 다시 틀렸다고 나오네요... 혹시 아직도 잘못된 부분이 있을까요? 의견 부탁드립니다.

참고로 게시판에 나와있는 모든 반례들은 다 해보았지만 정상적으로 나와서 혼란 스럽습니다...

진행 상황 : 이진 트리가 아닐경우 - 해결/부모 노드가 생성되기전 자식노드 부터 생성하려는 경우 - 해결/루트 노드만 남을경우 - 해결

var removeNodeNum: Int?
var lefNum: Int = 0

class TreeNode {
    
    var value: Int?
    var parent: TreeNode?
    var child: [TreeNode] = []
    
    public convenience init(value: Int) {
        self.init(value: value, child: nil, parent:nil)
    }
    
    public init(value : Int, child: TreeNode?, parent: TreeNode?) {
        
        self.value = value
        self.parent = parent
        guard let child = child else { return }
        self.child.append(child)
        
    }
    
    func addNode(value: Int, parent: Int) -> Bool {
        
        if parent == -1 {
            return true
        }
        
        if let pNode = searchParent(parent: parent) {
            pNode.child.append(TreeNode.init(value: value, child: nil, parent: pNode))
            return true
        } else {
            return false
        }
        
    }
    
    func searchParent(parent: Int) -> TreeNode? {
        
        var pNode: TreeNode?
        
        if self.value == parent {
            
            pNode = self
            
        } else {
            
            for i in 0..<self.child.count {
                
                pNode = self.child[i].searchParent(parent: parent)
                if pNode != nil {
                    break
                }
                
            }
            
        }
        
        return pNode
        
    }
    
    func getLefCnt() {
        
        var childCnt = self.child.count
        
        for i in 0..<self.child.count {
            
            if child[i].value != removeNodeNum {
                self.child[i].getLefCnt()
            } else {
                childCnt -= 1
            }
            
        }
        
        if childCnt == 0 {
            
            lefNum += 1
            
        }
        
    }
    
}

func main() {
    
    guard let strN = readLine() else { return }
    guard let n = Int(strN) else { return }
    
    var tempArr = Array.init(repeating: 0, count: n)
    guard let strPArr = readLine()?.split(separator: " ") else { return }
    var tree: TreeNode?
    var addNode = Array.init(repeating: false, count: n)
    
    for i in 0..<n {
        
        tempArr[i] = Int(strPArr[i])!
        if -1 == Int(strPArr[i])! {
            tree = TreeNode.init(value: i)
        }
        
    }
    
    //부모 노드부터 생성되면 그 이후에 자식노드 삽입...(모든 노드가 삽입 될때까지 반복)
    var addNodeCnt = 0
    while addNodeCnt != n {
        
        for i in 0..<n {
            
            if addNode[i] == false {
                
                if tree!.addNode(value: i, parent: tempArr[i]) {
                    addNodeCnt += 1
                    addNode[i] = true
                }
                
            }
            
        }
        
    }
    
    guard let strRemoveNode = readLine() else { return }
    guard let removeNode = Int(strRemoveNode) else { return }
    removeNodeNum = removeNode
    
    tree?.getLefCnt()
    print(lefNum)
    
}

main()

jh05013 3년 전 2

1
-1
0

hans081545 3년 전 0

감사합니다! 루트 노드가 제거될 경우는 생각 못했었네요! 덕분에 완전히 통과할 수 있었습니다.

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

문제

문제

출처

ICPC

코드를 수정하였지만... 여전히 틀렸다고 나오는... 반례 부탁드립니다!

hans081545 3년 전 0

jh05013 3년 전 2

hans081545 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회 / 고등학교 대회

출처

대학교 대회

도움말