글 읽기 - 알고리즘 어떤 부분을 수정해야 할까요?

citizen 7년 전 0

일단 이 문제가 이진트리에 관한 것이 아니라는 것을 가정하고

알고리즘을 구현해봤는데요. N-1개 줄에 각각의 노드의 상관관계를

입력으로 받아주니까 부모자식관계가 빠져있는 노드는 없을 것 같고,

각 줄마다 입력으로 주어지는 두개의 숫자중 parent[] 값이 0(default값)이

아니거나 루트인 노드를 부모로 나머지 노드를 자식으로 간주하였습니다.

parent[0]은 사용하지 않는 값이구요 parent[1]부터 해서 1번노드의 부모(물론 값은 0)

...로 지정하였습니다.

일단 예제는 모두 해결이 되는데요. 10%도 못넘기고 틀렸다고 나오네요

어디를 손 봐야할지 알려주시면 감사하겠습니다.

import java.util.*;
import java.awt.List;
import java.io.*;
import java.math.*;
public class Main{
	public static BufferedReader br = 
            new BufferedReader(new InputStreamReader(System.in));
    public static BufferedWriter bw = 
            new BufferedWriter(new OutputStreamWriter(System.out));
    
    public static void main(String args[]) throws IOException{
        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(br.readLine());
        int[] parent = new int[n+1];
        for(int i = 0; i < n-1; i++){
        	String[] nums = (br.readLine()).split(" ");
        	int n1 = Integer.parseInt(nums[0]);
        	int n2 = Integer.parseInt(nums[1]);
        	if(parent[n2] != 0 || n2 == 1){ // n2가 부모가 있거나 루트라면 n1은 n2의 자식으로
        		parent[n1] = n2;
        	} else{
        		parent[n2] = n1; // 해당사항이 없다면 전자(n1)이 후자(n2)의 부모로 설정
        	}
        }
        for(int i = 2; i <= n; i++){
        	bw.write(parent[i]+"\n");
        }
        bw.close();
    }
    
}

jseo 7년 전 0

3 2

1 2

답은

1
2

입니다

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

문제

문제

출처

ICPC

알고리즘 어떤 부분을 수정해야 할까요?

citizen 7년 전 0

jseo 7년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말