結果
問題 | No.1843 Tree ANDistance |
ユーザー |
![]() |
提出日時 | 2022-02-18 21:44:40 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 185 ms / 2,000 ms |
コード長 | 1,646 bytes |
コンパイル時間 | 3,931 ms |
コンパイル使用メモリ | 231,224 KB |
実行使用メモリ | 7,308 KB |
最終ジャッジ日時 | 2024-06-29 08:36:07 |
合計ジャッジ時間 | 8,404 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 38 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using namespace std; #define rep1(a) for(int64_t i = 0; i < a; i++) #define rep2(i, a) for(int64_t i = 0; i < a; i++) #define rep3(i, a, b) for(int64_t i = a; i < b; i++) #define rep4(i, a, b, c) for(int i = a; i < b; i += c) #define overload4(a, b, c, d, e, ...) e #define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__) const int64_t MOD=998244353; const long long INF = 1LL<<60; struct UnionFind { vector<int> par,siz; UnionFind(int N):par(N),siz(N,1){ for(int i=0;i<N;i++)par[i]=i; } void init(int N){ par.resize(N); siz.assign(N,1); for(int i=0;i<N;++i)par[i]=i; } int root(int x){ if(par[x]==x)return x; return par[x]=root(par[x]); } void unite(int x,int y){ int rx=root(x); int ry=root(y); if(rx==ry)return; if(siz[rx]<siz[ry])swap(rx, ry); siz[rx]+=siz[ry]; par[ry]=rx; } bool same(int x,int y){ int rx=root(x); int ry=root(y); return rx==ry; } int size(int x){ return siz[root(x)]; } }; int main(){ int N;cin>>N; vector<int64_t> A(N),B(N),C(N); rep(i,N){ cin>>A[i]>>B[i]>>C[i]; A[i]--,B[i]--; } UnionFind T(N); mint ans=0,d=1; int64_t D=1; rep(i,31){ //T.init(N); UnionFind T(N); rep(j,N){ if(C[j]&D){ T.unite(A[j],B[j]); //cout<<(1<<j)<<endl; } } rep(j,N){ if(T.root(j)==j){ int64_t s=T.size(j); ans+=d*(s*(s-1)/2); //cout<<ans.val()<<endl; } } d*=2; D*=2; } cout<<ans.val(); }