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

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

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	361 ms; 6,820 KB
testcase_01	AC	360 ms; 6,816 KB
testcase_02	AC	356 ms; 6,816 KB
testcase_03	AC	355 ms; 6,816 KB
testcase_04	AC	360 ms; 6,820 KB
testcase_05	AC	347 ms; 6,816 KB
testcase_06	AC	359 ms; 6,820 KB
testcase_07	AC	342 ms; 6,820 KB
testcase_08	AC	350 ms; 6,816 KB
testcase_09	AC	338 ms; 6,820 KB
testcase_10	AC	332 ms; 6,820 KB
testcase_11	AC	357 ms; 6,816 KB
testcase_12	AC	316 ms; 6,820 KB
testcase_13	AC	334 ms; 6,816 KB
testcase_14	AC	13 ms; 6,820 KB
testcase_15	AC	14 ms; 6,820 KB
testcase_16	AC	7 ms; 6,816 KB
testcase_17	AC	6 ms; 6,820 KB
testcase_18	AC	4 ms; 6,820 KB
testcase_19	AC	16 ms; 6,816 KB
testcase_20	AC	9 ms; 6,820 KB
testcase_21	AC	2 ms; 6,820 KB
testcase_22	AC	2 ms; 6,816 KB
testcase_23	AC	2 ms; 6,816 KB
testcase_24	AC	2 ms; 6,816 KB
testcase_25	AC	2 ms; 6,816 KB
testcase_26	AC	2 ms; 6,820 KB
testcase_27	AC	2 ms; 6,816 KB
testcase_28	AC	368 ms; 6,816 KB
testcase_29	AC	248 ms; 6,820 KB
testcase_30	AC	258 ms; 6,816 KB
testcase_31	AC	401 ms; 6,820 KB
testcase_32	AC	377 ms; 6,820 KB
testcase_33	AC	131 ms; 6,820 KB
testcase_34	AC	298 ms; 6,816 KB
testcase_35	AC	2 ms; 6,820 KB
testcase_36	AC	2 ms; 6,816 KB
testcase_37	AC	2 ms; 6,816 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();
}