結果
問題 | No.1843 Tree ANDistance |
ユーザー |
![]() |
提出日時 | 2022-02-18 21:24:09 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 149 ms / 2,000 ms |
コード長 | 1,772 bytes |
コンパイル時間 | 2,315 ms |
コンパイル使用メモリ | 199,640 KB |
最終ジャッジ日時 | 2025-01-27 23:39:39 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 38 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; } template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; } #define all(x) (x).begin(),(x).end() #define fi first #define se second #define mp make_pair #define si(x) int(x.size()) const int mod=1000000007,MAX=200005; const int INF=1<<30; //Union Find struct UF{ int n; vector<int> par,size,edge; void init(int n_){ n=n_; par.assign(n,-1); size.assign(n,1); edge.assign(n,0); for(int i=0;i<n;i++){ par[i]=i; } } int root(int a){ if(par[a]==a) return a; else return par[a]=root(par[a]); } void unite(int a,int b){ edge[root(a)]++; if(root(a)!=root(b)){ size[root(a)]+=size[root(b)]; edge[root(a)]+=edge[root(b)]; par[root(b)]=root(a); } } bool check(int a,int b){ return root(a)==root(b); } }; int main(){ std::ifstream in("text.txt"); std::cin.rdbuf(in.rdbuf()); cin.tie(0); ios::sync_with_stdio(false); int N;cin>>N; vector<int> A(N),B(N),C(N); for(int i=0;i<N-1;i++){ cin>>A[i]>>B[i]>>C[i]; A[i]--;B[i]--; } ll ans=0; for(int t=0;t<30;t++){ UF uf;uf.init(N); for(int i=0;i<N-1;i++){ if(C[i]&(1<<t)) uf.unite(A[i],B[i]); } for(int i=0;i<N;i++){ if(uf.root(i)==i){ ll x=uf.size[i]; ans+=x*(x-1)/2%mod*(1<<t); ans%=mod; } } } cout<<ans<<endl; }