#904436 (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
• 実行時間: 351 ms / 2,000 ms
• コード長: 746 bytes
• コンパイル時間: 5,621 ms
• コンパイル使用メモリ: 310,112 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-04 08:06:06
• 合計ジャッジ時間: 12,887 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	316 ms; 5,248 KB
testcase_01	AC	316 ms; 5,248 KB
testcase_02	AC	313 ms; 5,376 KB
testcase_03	AC	313 ms; 5,376 KB
testcase_04	AC	318 ms; 5,376 KB
testcase_05	AC	304 ms; 5,376 KB
testcase_06	AC	313 ms; 5,376 KB
testcase_07	AC	299 ms; 5,376 KB
testcase_08	AC	308 ms; 5,376 KB
testcase_09	AC	297 ms; 5,376 KB
testcase_10	AC	292 ms; 5,376 KB
testcase_11	AC	315 ms; 5,376 KB
testcase_12	AC	276 ms; 5,376 KB
testcase_13	AC	289 ms; 5,376 KB
testcase_14	AC	11 ms; 5,376 KB
testcase_15	AC	13 ms; 5,376 KB
testcase_16	AC	7 ms; 5,376 KB
testcase_17	AC	6 ms; 5,376 KB
testcase_18	AC	4 ms; 5,376 KB
testcase_19	AC	14 ms; 5,376 KB
testcase_20	AC	7 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	322 ms; 5,376 KB
testcase_29	AC	216 ms; 5,376 KB
testcase_30	AC	226 ms; 5,376 KB
testcase_31	AC	351 ms; 5,376 KB
testcase_32	AC	328 ms; 5,376 KB
testcase_33	AC	113 ms; 5,376 KB
testcase_34	AC	258 ms; 5,376 KB
testcase_35	AC	1 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();
}