結果

問題 No.1843 Tree ANDistance
ユーザー ripityripity
提出日時 2022-02-18 22:23:40
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 2,246 bytes
コンパイル時間 2,513 ms
コンパイル使用メモリ 79,488 KB
実行使用メモリ 333,448 KB
最終ジャッジ日時 2024-06-29 09:16:09
合計ジャッジ時間 9,839 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

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

public class Main {
    
    public static Scanner sc = new Scanner(System.in);
    public static PrintWriter pw = new PrintWriter(System.out);
    
    public static void main(String[] args) {
        
        int t = 1;
        while( t > 0 ) {
            solve();
            t--;
        }
        
        pw.flush();
        
    }
    
    static void solve() {
        
        int N = sc.nextInt();
        long ans = 0;
        long P = 1000000007;
        ArrayList<ArrayList<ArrayList<Integer>>> edge = new ArrayList<>();
        for( int i = 0; i < 30; i++ ) {
            edge.add(new ArrayList<>());
            for( int j = 0; j < N; j++ ) {
                edge.get(i).add(new ArrayList<>());
            }
        }
        
        for( int i = 0; i < N-1; i++ ) {
            int a = sc.nextInt()-1;
            int b = sc.nextInt()-1;
            int c = sc.nextInt();
            for( int j = 0; j < 30; j++ ) {
                if( (c>>j)%2 == 1 ) {
                    edge.get(j).get(a).add(b);
                    edge.get(j).get(b).add(a);
                }
            }
        }
        
        for( int i = 0; i < 30; i++ ) {
            long k = modpow(2,i,P);
            LinkedList<Integer> queue = new LinkedList<>();
            boolean[] visited = new boolean[N];
            for( int j = 0; j < N; j++ ) {
                if( visited[j] ) continue;
                queue.offer(j);
                long cnt = 0;
                while( !queue.isEmpty() ) {
                    int now = queue.poll();
                    if( visited[now] ) continue;
                    visited[now] = true;
                    cnt++;
                    for( int next : edge.get(i).get(now) ) {
                        if( !visited[next] ) queue.offer(next);
                    }
                }
                ans += cnt*(cnt-1)/2%P*k%P;
                ans %= P;
            }
        }
        
        pw.println(ans);
        
    }
    
    static long modpow( long x, long n, long mod ) {
		
		if( n == 0 ) {
			return 1;
		}
		
		long k = 1;
		while( n > 1 ) {
			if( n%2 == 1 ) k *= x;
			x *= x;
			n /= 2;
			x %= mod;
			k %= mod;
		}
		
		return (k*x)%mod;
		
	}
    
}
0