結果

問題 No.1817 Reversed Edges
ユーザー merlin
提出日時 2022-01-21 22:20:00
言語 Java
(openjdk 23)
結果
AC  
実行時間 643 ms / 2,000 ms
コード長 1,293 bytes
コンパイル時間 2,812 ms
コンパイル使用メモリ 80,556 KB
実行使用メモリ 81,648 KB
最終ジャッジ日時 2024-11-26 00:49:02
合計ジャッジ時間 14,394 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.*;
import java.util.*;

class Main
{
    public static void main(String args[])throws Exception
    {
        BufferedReader bu=new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb=new StringBuilder();
        int n=Integer.parseInt(bu.readLine());
        int i,c[]=new int[n],e[]=new int[n];
        ArrayList<Integer> g[]=new ArrayList[n];

        for(i=0;i<n;i++) g[i]=new ArrayList<>();
        for(i=0;i<n-1;i++)
        {
            String s[]=bu.readLine().split(" ");
            int u=Integer.parseInt(s[0])-1,v=Integer.parseInt(s[1])-1;
            g[u].add(v);
            g[v].add(u);
        }

        dfs(g,c,0,-1);
        reroot(g,c,e,0,-1);
        for(i=0;i<n;i++) sb.append(c[i]+e[i]+"\n");
        System.out.print(sb);
    }

    static void dfs(ArrayList<Integer> g[],int c[],int n,int p)
    {
        for(int x:g[n])
        if(x!=p)
        {
            dfs(g,c,x,n);
            c[n]+=c[x];
            if(x<n) c[n]++;
        }
    }

    static void reroot(ArrayList<Integer> g[],int c[],int e[],int n,int p)
    {
        for(int x:g[n])
        if(x!=p)
        {
            e[x]=e[n];
            e[x]+=c[n]-c[x];
            if(n>x) e[x]--;
            else e[x]++;
            reroot(g,c,e,x,n);
        }
    }
}
0