[图论]游戏

题目描述

B B B 经常与 A A A 一起玩游戏。今天,他们在一棵树上玩游戏。 A A A 有 m 1 m1 m1 块石子, B B B 有 m 2 m2 m2 块石子,游戏一开始,所有石头放在树的节点处,除了树根。

A A A 先移动石子。然后两人轮流移动,每次移动只能选择自己的一个石子,而且只能从当前位置移到父亲节点处,游戏过程中允许一个节点处放多个石子。

谁先把自己所有的石子移到树根处谁就失败了,假设两人都是非常聪明,游戏过程中都使用最优策略,给定石子起始位置,要你计算出谁是赢家。

为了简化题目,假设树根为 0 0 0。

输入格式

输入包含多组测试数据。

第一行输入 T T T 表示测试数据组数。

接下来每组测试数据第一行输入 3 3 3 个整数 n n n, m 1 m1 m1, m 2 m2 m2, n n n 为树的节点个数。

接下来 n − 1 n - 1 n−1 行,每行包含两个整数 x , y x,y x,y ( 0 ≤ x , y < n ) (0 \le x, y < n) (0≤x,y<n),表示树中有一条边连接 x x x 和 y y y。

接下来 m 1 m1 m1 行,每行一个整数,表示 A A A 的石子的位置。

接下来 m 2 m2 m2 行,每行一个整数,表示 B B B 的石子的位置。

输出格式

对于每组数据,输出 A A A 或 B B B 表示 A A A 或 B B B 赢了。

数据范围

对于 30 % 30\% 30% 的数据, 1 ≤ n ≤ 10 1 \le n \le 10 1≤n≤10, 1 ≤ m 1 , m 2 ≤ n 1 \le m1, m2 \le n 1≤m1,m2≤n。

对于 100 % 100\% 100% 的数据, 1 ≤ n ≤ 1 0 4 1 \le n \le 10^4 1≤n≤104, 1 ≤ m 1 , m 2 ≤ 1 0 4 1 \le m1,m2 \le 10^4 1≤m1,m2≤104。

样例

输入:

2
3 1 1
0 1
2 0
1
2
3 2 1
0 1
1 2
2 2
2

输出:

B
A

思路

首先一眼看上去,似乎是个博弈题,但其实只是简单的树。

石子只能移到其根节点,且根节点只有一个,因此每个石子移到根节点的距离就能求出来。

将 A A A 和 B B B 的每个石子离根节点的距离求出来相加,进行比较。

A A A 为先手,只有 s u m A > s u m B sum_A > sum_B sumA>sumB 时, A A A 才能赢,否则 B B B 赢。

比如 s u m A = s u m B = 1 sum_A = sum_B = 1 sumA=sumB=1, A A A 先走, B B B 再走, A A A 就不能走了, B B B 赢。

代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
int T;
int n, m1, m2;
int dep[100010];//深度
int a[100010], b[100010];
vector<int> v[100010];//图
//x:当前节点 y:父亲节点 z:当前节点深度
void dfs(int x, int y, int z){
	dep[x] = z;
	for(auto i : v[x]){
		if(i == y){//父亲
			continue;
		}
		dfs(i, x, z + 1);
	}
}
int main(){
	scanf("%d", &T);
	while(T --){
		//多组数据清空
		memset(a, 0, sizeof(a));
		memset(b, 0, sizeof(b));
		memset(dep, 0, sizeof(dep));
		scanf("%d %d %d", &n, &m1, &m2); 
		for(int i = 1; i < n; ++ i){
			int x, y;
			scanf("%d %d", &x, &y);
			v[x].push_back(y);
			v[y].push_back(x);
		}
		//将 a和b 每个位置的石子个数统计出来,方便以后计算
		for(int i = 1; i <= m1; ++ i){
			int x;
			scanf("%d", &x);
			a[x] ++;
		}
		for(int i = 1; i <= m2; ++ i){
			int x;
			scanf("%d", &x);
			b[x] ++;
		}
		//dfs 计算每个节点的深度
		dfs(0, -1, 0);
		int sum1 = 0, sum2 = 0;
		for(int i = 1; i <= n; ++ i){
			sum1 += a[i] * dep[i];
		}
		for(int i = 1; i <= n; ++ i){
			sum2 += b[i] * dep[i];
		}
		if(sum1 > sum2){
			printf("A\n");
		}
		else{
			printf("B\n");	
		}
		//v 在清空时注意要把 0 清了
		for(int i = 0; i <= n; ++ i){
			v[i].clear();
		}
	}
	return 0;
}
相关推荐
JSU_曾是此间年少15 分钟前
数据结构——线性表与链表
数据结构·c++·算法
此生只爱蛋1 小时前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
何曾参静谧1 小时前
「C/C++」C/C++ 指针篇 之 指针运算
c语言·开发语言·c++
lulu_gh_yu2 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法
ULTRA??3 小时前
C加加中的结构化绑定(解包,折叠展开)
开发语言·c++
凌云行者3 小时前
OpenGL入门005——使用Shader类管理着色器
c++·cmake·opengl
凌云行者4 小时前
OpenGL入门006——着色器在纹理混合中的应用
c++·cmake·opengl
~yY…s<#>4 小时前
【刷题17】最小栈、栈的压入弹出、逆波兰表达式
c语言·数据结构·c++·算法·leetcode
可均可可5 小时前
C++之OpenCV入门到提高004:Mat 对象的使用
c++·opencv·mat·imread·imwrite
白子寰5 小时前
【C++打怪之路Lv14】- “多态“篇
开发语言·c++