結果

問題 No.1843 Tree ANDistance
ユーザー publfl2publfl2
提出日時 2022-02-18 21:43:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 138 ms / 2,000 ms
コード長 982 bytes
コンパイル時間 512 ms
コンパイル使用メモリ 52,292 KB
実行使用メモリ 16,952 KB
最終ジャッジ日時 2024-06-29 08:33:14
合計ジャッジ時間 3,899 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <vector>
#define MOD 1000000007

std::vector< std::pair<int,int> > edge[110];
int next[100010],size[100010];
int find(int k)
{
	if(k==next[k]) return k;
	else return next[k] = find(next[k]);
}

int main()
{
	int a;
	scanf("%d",&a);
	for(int i=1;i<a;i++)
	{
		int b,c,d;
		scanf("%d%d%d",&b,&c,&d);
		for(int j=1;j<=30;j++)
		{
			if(d%2==1) edge[j].push_back(std::make_pair(b,c));
			d/=2;
		}
	}
	
	long long int ans = 0;
	for(int j=1;j<=30;j++)
	{
		for(int i=1;i<=a;i++) next[i] = i;
		for(int i=1;i<=a;i++) size[i] = 1;
		
		for(int i=0;i<edge[j].size();i++)
		{
			int s = edge[j][i].first;
			int t = edge[j][i].second;
			if(find(s)!=find(t))
			{
				size[find(t)] += size[find(s)];
				next[find(s)] = find(t);
			}
		}
		
		for(int i=1;i<=a;i++)
		{
			if(i==find(i))
			{
				long long int val = (long long int)size[i]*(size[i]-1)/2;
				val %= MOD, val *= (1<<(j-1)), val %= MOD;
				ans += val, ans %= MOD;
			}
		}
	}
	printf("%lld",ans);
}
0