{"componentChunkName":"component---src-templates-blog-js","path":"/blog/algorithm/algo_20200303_1/","result":{"data":{"markdownRemark":{"html":"

Tree에서 한 node를 제거했을 때, 남은 leaf node의 개수를 구하는 프로그램.

\n

백준 1068 : 트리

\n

\"img\"\n\"img\"

\n
    \n
  1. \n

    node 1개 제거 후, 같이 제거되는 모든 node를 표시한다.

    \n\n
    2. \n
  2. \n

    남은 node의 개수를 res에 저장.

    \n\n
    3. \n
  3. 남은 node들의 parent를 Set에 저장한다.
    4. \n
  4. \n

    남은 node의 개수 (res)-parent Set의 size = leaf node의 개수를 구한다.\n

    \n

    Java Code

    \n
    5. \n
\n
import java.util.HashSet;\nimport java.util.Scanner;\nimport java.util.Set;\n\npublic class Main {\n    static Scanner in = new Scanner(System.in);\n    static int n;\n    static int[] parent = new int[50];\n    static Set<Integer> parentSet = new HashSet<>();\n    static int target;\n    public static void main(String[] args) {\n        n = in.nextInt();\n        for(int i=0;i<n;i++) {\n            parent[i] = in.nextInt();\n        }\n        target = in.nextInt();\n//        제거되는 node의 parent = -2로 저장\n        parent[target] = -2;\n        int res = n;\n        for(int i=0;i<n;i++) {\n            if(Cut(i)) {\n//                같이 제거되는 node의 parent = -2로 저장\n                parent[i] = -2;\n                res--;\n            }\n        }\n        \n        for(int i=0;i<n;i++) {\n            if(parent[i]!=-2&&parent[i]!=-1) {\n//                제거되지 않은 남은 node & root가 아닌 node 중에서,\n//                그 node의 parent를 Set에 저장.\n                parentSet.add(parent[i]);\n            }\n        }\n        int leaf = res-parentSet.size();\n        System.out.println(leaf);\n\n    }\n    public static boolean Cut(int n) {\n//        node n의 조상 중, 제거되는 node가 있을 경우 Cut=true.\n        while (parent[n]!=-1) {\n            if(parent[n]==-2) {\n                return true;\n            }\n            n = parent[n];\n        }\n        return false;\n    }\n}
","id":"a0318139-67b2-5070-afe7-6aae61223e6c","frontmatter":{"title":"[Algorithm] 백준 1068 : 트리 ","date":"2020-03-03T00:00:00.000Z","description":null},"fields":{"slug":"/blog/algorithm/algo_20200303_1/"}}},"pageContext":{"slug":"/blog/algorithm/algo_20200303_1/"}}}