結果
問題 | No.1843 Tree ANDistance |
ユーザー |
|
提出日時 | 2023-07-15 13:19:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 897 ms / 2,000 ms |
コード長 | 2,991 bytes |
コンパイル時間 | 1,177 ms |
コンパイル使用メモリ | 114,000 KB |
最終ジャッジ日時 | 2025-02-15 14:58:29 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 38 |
ソースコード
#include<iostream> #include<set> #include<algorithm> #include<vector> #include<string> #include<set> #include<map> #include<numeric> #include<queue> #include<cmath> using namespace std; typedef long long ll; const ll INF=1LL<<60; typedef pair<int,int> P; typedef pair<int,P> PP; const ll MOD=1e9+7; const double PI=acos(-1); struct edge{ int u; int v; int c; edge(int u_,int v_,int c_):u(u_),v(v_),c(c_){}; }; class UnionFind{ public: std::vector<int> par; std::vector<int> rank; std::vector<int> sz;//sz[i]で,頂点iが含まれているグループのサイズ std::vector<int> root; UnionFind(int size){ rank=std::vector<int>(size+1,0); par = std::vector<int>(size+1,0); std::iota(par.begin(),par.end(),0);//#include<numeric> sz = std::vector<int>(size+1,1); root=std::vector<int>(size+1); std::iota(root.begin(),root.end(),0); } ~UnionFind(){ std::vector<int>().swap(rank); std::vector<int>().swap(par); std::vector<int>().swap(sz); std::vector<int>().swap(root); } int find(int x){ if(par[x]==x)return x; else return par[x] = find(par[x]); } bool issame(int u,int v){ return find(u)==find(v); } void merge(int u,int v){ u = find(u); v = find(v); if(u==v) return; if(rank[u]<rank[v]) std::swap(u,v); //uの傘下へvが入る //rank[u]>=rank[v] par[v]=u; sz[u]+=sz[v]; if(rank[u]==rank[v])rank[u]++; } int size(int u){//頂点uが属すグループの大きさを表す. u=find(u); return sz[u]; } std::map<int,std::vector<int>> element(){ std::map<int,std::vector<int>> mp; for(int i=0;i<par.size();i++){ root[i]=find(i); mp[root[i]].push_back(i); } return mp; } }; int main(){ int N; cin>>N; //vector<vector<edge>> G(N); vector<ll> A(N-1),B(N-1),W(N-1); for(int i=0;i<N-1;i++){ int a,b,c; cin>>a>>b>>c; a--;b--; A[i]=a; B[i]=b; W[i]=c; // G[a].emplace_back(a,b,c); // G[b].emplace_back(b,a,c); } ll ans=0; for(int k=0;k<31;k++){ UnionFind uf(N); for(int i=0;i<N-1;i++){ if( (W[i]>>k)&1){ uf.merge(A[i],B[i]); } } map<int,int> mp; for(int i=0;i<N;i++){ mp[uf.find(i)]++; } /* ll cnt=0; for(auto [_,num]:mp){ ll cnt+=(1LL*num*(num-1)/2)%MOD; cnt%=MOD; } ans+=(cnt*((1LL<<k)%MOD))%MOD; ans%=MOD; */ for(auto [_,num]:mp){ ll tmp=(1LL*num*(num-1)/2)%MOD; ans+=(tmp*((1LL<<k)%MOD))%MOD; ans%=MOD; } } cout<<ans<<endl; }