結果

問題 No.1843 Tree ANDistance
コンテスト
ユーザー kenta255
提出日時 2022-02-23 22:06:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 3,949 bytes
コンパイル時間 3,103 ms
コンパイル使用メモリ 224,764 KB
最終ジャッジ日時 2025-01-28 01:32:42
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 12 RE * 22 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef priority_queue<ll, vector<ll>, greater<ll> > PQ;
//#define P pair<ll,ll>
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)  // int範囲に注意 llより早い
#define FORR(I,A,B) for(int I = int((B)-1); I >= int(A); --I)
#define TO(x,t,f) ((x)?(t):(f))
#define SORT(x) (sort(x.begin(),x.end())) // 0 2 2 3 4 5 8 9
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define NUM(x,v) (POSU(x,v)-POSL(x,v))  //x is sorted
#define REV(x) (reverse(x.begin(),x.end())) //reverse
ll gcd_(ll a,ll b){if(a%b==0)return b;return gcd_(b,a%b);}
ll lcm_(ll a,ll b){ll c=gcd_(a,b);return ((a/c)*b);}
#define NEXTP(x) next_permutation(x.begin(),x.end())
const ll INF=ll(1e16)+ll(7);
const ll MOD=1000000007LL;
#define out(a) cout<<fixed<<setprecision((a))
//tie(a,b,c) = make_tuple(10,9,87); pairもいける
#define pop_(a) __builtin_popcount((a))
ll keta(ll a){ll r=0;while(a){a/=10;r++;}return r;}
ll ketawa(ll a){ll r=0;while(a){r+=a%10;a/=10;}return r;}
#define num_l(a,v) POSL(a,v) //v未満の要素数
#define num_eql(a,v) POSU(a,v) //v以下の要素数
#define num_h(a,v) (a.size() - POSU(a,v)) //vより大きい要素数
#define num_eqh(a,v) (a.size() - POSL(a,v)) //v以上の要素数


vector<int> Centroid(const vector< vector<int> > &G){
	queue<int> leaf;
	const int root = 0;
	const int N = G.size();
	vector<int> cnt(N,0),pa(N,-1);
	FOR(i,0,N){
		cnt[i] = G[i].size();
		if(G[i].size() == 1 && i != root){
			leaf.push(i);
		}
	}
	vector<int> size_sub_tree(N,1);
	while(leaf.size()){
		int u = leaf.front();
		leaf.pop();
		for(int v:G[u]){
			if(cnt[v] >= 2){
				size_sub_tree[v] += size_sub_tree[u];
				cnt[v]--;
				if(cnt[v] == 1 && v!=root){
					leaf.push(v);
					pa[u] = v;
				}
			}
		}
	}
	vector<int> res;
	FOR(i,0,N){
		bool ok = (N-size_sub_tree[i]) <= N/2;
		for(auto u:G[i]){
			if(u == pa[i]) continue;
			ok &= size_sub_tree[u] <= N/2;
		}
		if(ok) res.push_back(i);
	}
	return res;
}

int main(){
	int N;
	cin >> N;
	vector<vector<int>> G(N),Gc(N);
	FOR(i,0,N-1){
		int u,v,c;
		cin >> u >> v >> c;
		G[u-1].push_back(v-1);
		G[v-1].push_back(u-1);
		Gc[u-1].push_back(c);
		Gc[v-1].push_back(c);
	}
	queue< vector<vector<int>> > QG;
	queue< vector<vector<int>> > QGc;
	QG.push(G); G.clear();
	QGc.push(Gc); Gc.clear();

	ll ans = 0;
	while(QG.size()){
		int center = Centroid(QG.front())[0];
		auto g = QG.front();
		QG.pop();
		auto gc = QGc.front();
		QGc.pop();
		
		vector<ll> cnt(64,0); // ibit目が1のやつの個数
		for(int i=0;i<g[center].size();i++){
			map<int,int> number;
			int k = 1;
			// g[center][i]を根とする部分木で計算
			vector<ll> and_(g.size(),-1);
			and_[ g[center][i] ] = gc[center][i];
			and_[ center ] = 0;
			queue<int> v;
			v.push(g[center][i]);
			vector<vector<int>> makeG,makeGc;
			vector<int> use;
			while(v.size()){
				int from = v.front(); v.pop();
				for(int j=0;j<(int)g[from].size();j++){
					int to = g[from][j];
					if(and_[to] >= 0)continue;
					and_[to] = and_[from] & gc[from][j];
					v.push(to);
					if(number[from] == 0){ number[from] = k++; use.push_back(from); }
					if(number[ to ] == 0){ number[ to ] = k++; use.push_back(to); }
					makeG.resize(k);
					makeGc.resize(k);
					makeG[ number[from]-1 ].push_back( number[ to ]-1 );
					makeG[ number[ to ]-1 ].push_back( number[from]-1 );
					makeGc[ number[from]-1 ].push_back( gc[center][j] );
					makeGc[ number[ to ]-1 ].push_back( gc[center][j] );
				}
			}
			if(k>1)QG.push(makeG);
			if(k>1)QGc.push(makeGc);
			for(auto u:use){
				for(int j=0;j<33;j++){
					if(and_[u] & (1LL<<j)){
						(ans += (1LL<<j) * cnt[j]%MOD) %= MOD;
					}
				}
			}
			for(auto u:use){
				for(int j=0;j<33;j++){
					if(and_[u] & (1LL<<j)){
						cnt[j]++;
					}
				}
			}
		}
	}
	cout << ans << endl;
}
0