#904435 (C++23) No.1843 Tree ANDistance

• 問題: No.1843 Tree ANDistance
• ユーザー: たたき@競プロ
• 提出日時: 2023-08-17 00:46:37
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 374 ms / 2,000 ms
• コード長: 746 bytes
• コンパイル時間: 6,001 ms
• コンパイル使用メモリ: 310,116 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-04 08:05:51
• 合計ジャッジ時間: 15,255 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	340 ms; 5,248 KB
testcase_01	AC	362 ms; 5,248 KB
testcase_02	AC	334 ms; 5,376 KB
testcase_03	AC	323 ms; 5,376 KB
testcase_04	AC	334 ms; 5,376 KB
testcase_05	AC	322 ms; 5,376 KB
testcase_06	AC	340 ms; 5,376 KB
testcase_07	AC	325 ms; 5,376 KB
testcase_08	AC	324 ms; 5,376 KB
testcase_09	AC	309 ms; 5,376 KB
testcase_10	AC	307 ms; 5,376 KB
testcase_11	AC	326 ms; 5,376 KB
testcase_12	AC	292 ms; 5,376 KB
testcase_13	AC	308 ms; 5,376 KB
testcase_14	AC	14 ms; 5,376 KB
testcase_15	AC	14 ms; 5,376 KB
testcase_16	AC	6 ms; 5,376 KB
testcase_17	AC	6 ms; 5,376 KB
testcase_18	AC	4 ms; 5,376 KB
testcase_19	AC	16 ms; 5,376 KB
testcase_20	AC	8 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	2 ms; 5,376 KB
testcase_25	AC	2 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	344 ms; 5,376 KB
testcase_29	AC	228 ms; 5,376 KB
testcase_30	AC	238 ms; 5,376 KB
testcase_31	AC	374 ms; 5,376 KB
testcase_32	AC	350 ms; 5,376 KB
testcase_33	AC	124 ms; 5,376 KB
testcase_34	AC	280 ms; 5,376 KB
testcase_35	AC	2 ms; 5,376 KB
testcase_36	AC	2 ms; 5,376 KB
testcase_37	AC	2 ms; 5,376 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint=modint1000000007;
using ll=long long;
using pp=pair<int,int>;
#define sr string 
#define vc vector
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define pb push_back
#define all(v) v.begin(),v.end()
#define pque priority_queue
#define bpc(a) __builtin_popcount(a)
struct tup{int a,b,c;};
int main(){
  int n;cin>>n;
  vc<tup>v(n-1);
  rep(i,n-1)cin>>v[i].a>>v[i].b>>v[i].c;
  mint ans=0;
  for(int b=0;b<30;b++){
    dsu d(n);
    mint t=1<<b;
    rep(i,n-1)if(v[i].c>>b&1)d.merge(v[i].a-1,v[i].b-1);
    rep(i,n)if(d.leader(i)==i){
      int x=d.size(i);
      ans+=t*x*(x-1)/2;
    }
  }
  cout<<ans.val();
}