F1. 生活在树上(easy version)树,dfs

题目链接

F1. 生活在树上(easy version)

题目背景

本题是 B 组的最后一题,是 F2 题的简单版本,两道题目的解法略有不同。本题和 F2 题在题意上的区别在于本题给定树上的边权,而不是点权。

小智生活在「传智国」,这是一个有 n n n 个城市的国家,各个城市由 n − 1 n-1 n1 条道路相连。

每个道路都有长度 w i w_i wi ,我们定义,小智从城市 a a a 走到城市 b b b 的代价是 d i s a , b = ⨁ e ∈ p a t h ( a , b ) w e \mathrm{dis}_{a, b} = \bigoplus \limits_{e \in \mathrm{path}\left(a, b\right)} w_e disa,b=epath(a,b)we,其中 ⨁ \bigoplus 表示按位异或(如果你不知道什么是按位异或,请参见题目下方的提示/说明), p a t h ( a , b ) \mathrm{path}\left(a,b\right) path(a,b) 表示 a a a b b b 的简单路径上的边集。也即 d i s a , b \mathrm{dis}_{a, b} disa,b 表示将 a a a b b b 的简单路径上所有边写作 e 1 , e 2 , e 3 , … e_1, e_2, e_3, \dots e1,e2,e3, 后,求 w e 1 ⨁ w e 2 ⨁ w e 3 … w_{e_1} \bigoplus w_{e_2}\bigoplus w_{e_3} \dots we1we2we3 的结果。

有一天,小智获得了去参加传智杯的机会,他在前往比赛地的路上想到了一个问题,但他好像不会做,于是他把这个题告诉了你。聪明的同学,你可以帮帮他吗?

题目描述

小智说:「由于我们的国家只有 n n n 个城市和 n − 1 n-1 n1 条道路,那么我们的国家就相当于是一棵树。我在想,在我们的国家中,是否有城市满足『到城市 a a a 的代价和到城市 b b b 的代价的异或等于 k k k』。好难哦,我想不出来,你能帮帮我吗?」

形式化的,给定城市 a , b a, b a,b 和整数 k k k,请你计算有哪几个城市 t t t 满足 d i s t , a ⨁ d i s t , b = k \mathrm{dis}_{t, a} \bigoplus \mathrm{dis}_{t, b} = k dist,adist,b=k

输入格式

第一行有两个整数 n n n m m m,表示国家的城市数和询问的个数。

接下来 n − 1 n-1 n1 行,每行有两个整数 x , y , l x, y, l x,y,l,表示城市 x x x 与城市 y y y 有一条长度为 l l l 的边。

接下来 m m m 行,每行有三个整数 a , b , k a, b, k a,b,k,表示小智从城市 a a a 走到城市 b b b k k k 的含义与题目描述一致。

输出格式

m m m 行,每行一个整数。

对于第 i i i 个询问,如果存在至少一个城市 t t t 满足要求,则输出 Yes

如果不存在任何一个城市满足条件,则输出 No

输出字符串大小写不敏感,例如,YesyESYESyes 等都算作 Yes

样例 #1

样例输入 #1

5 3
1 2 2
1 3 6 
2 4 8
2 5 1
1 2 4
2 3 12
1 4 10

样例输出 #1

nO
No
YeS

样例 #2

样例输入 #2

5 10
2 1 63
3 1 57
4 2 2
5 2 84
5 2 84
4 1 9977404983223574764
2 5 84
2 1 15996060349666123522
5 4 86
3 1 8428615422876116375
5 1 107
2 3 6
2 3 6
4 2 2

样例输出 #2

yeS
nO
YEs
No
YEs
nO
YEs
yeS
yeS
YEs

提示

相关概念解释

「树」:树是一个有 n n n 个结点和 n − 1 n-1 n1 条边的无向简单连通图。

「按位异或」:按位异或即 C++、python、java 语言中的 「^」 运算。它是一个二元运算,步骤是将两个数的二进制位按位比较,相同为 0 0 0,不同为 1 1 1。例如: 3 ⨁ 5 = ( 011 ) 2 ⨁ ( 101 ) 2 = ( 110 ) 2 = 6 3 \bigoplus 5 = (011)_2 \bigoplus (101)_2 = (110)_2 = 6 35=(011)2(101)2=(110)2=6请注意,这是一个按位运算,不是一个逻辑运算

样例 1 解释

下图为传智国的地图。

∀ t ∈ { 1 , 2 , 3 , 4 , 5 } \forall t \in \{1, 2, 3, 4, 5\} t{1,2,3,4,5},都不可能有 d i s t , 1 ⨁ d i s t , 2 = 4 \mathrm{dis} _{t,1} \bigoplus \mathrm{dis}_{t, 2} = 4 dist,1dist,2=4 d i s t , 2 ⨁ d i s t , 3 = 12 \mathrm{dis}_{t, 2} \bigoplus \mathrm{dis}_{t, 3} = 12 dist,2dist,3=12,于是输出 No

而取 t = 5 t = 5 t=5,有 d i s t , 1 ⨁ d i s t , 4 = 10 \mathrm{dis}_{t, 1} \bigoplus \mathrm{dis}_{t, 4} = 10 dist,1dist,4=10,于是输出 Yes

数据规模与约定

对于所有测试点,保证 1 < n ≤ 5 × 1 0 5 1 < n \leq 5 \times 10^5 1<n5×105 1 ≤ m ≤ 5 × 1 0 5 1 \leq m \leq 5 \times 10^5 1m5×105 0 ≤ w i < 2 64 0 \leq w_i < 2^{64} 0wi<264

对于每次询问,保证 1 ≤ a , b ≤ n 1 \leq a,b \leq n 1a,bn a ≠ b a \neq b a=b 0 ≤ k < 2 64 0 \leq k < 2^{64} 0k<264
思路:
首先需要明确的是和异或的特点,两个相同的数异或的结果为0,任何数异或0之后的仍旧是它本身。所以这道题目,需要注意的就是树的存储方式。只要能选中任何一个点作为起点,计算完到其他n-1个点的异或的结果,就能知道是否满足给定的条件。
代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e5+11;
#define ll unsigned long long 
int n,m;
ll dis[maxn];
bool flag[maxn];
struct edge
{
	int to;
	ll v;
};
vector<edge> e[maxn];
void dfs(int x)
{
	
	for(int i=0;i<e[x].size();i++)
	{
		ll y=e[x][i].to,z=e[x][i].v;
		if(flag[y]) continue;
		dis[y]=dis[x]^z;
		flag[y]=1;
		dfs(y);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)
	{
		int x,y;
		ll z;
		scanf("%d%d%llu",&x,&y,&z);
		e[x].push_back({y,z});
		e[y].push_back({x,z});
	}
	flag[1]=1;
	dfs(1);
	//for(int i=1;i<=n;i++) cout<<"i="<<i<<" "<<"dis[i]="<<dis[i]<<endl;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		ll k;
		scanf("%d%d%llu",&a,&b,&k);
		//cout<<"dis[a]^dis[b="<<(dis[a]^dis[b])<<endl;
		if((dis[a]^dis[b])==k) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值