gesp(C++六级)（6）洛谷：P10109：[GESP202312 六级] 工作沟通

gesp(C++六级)（6）洛谷：P10109：[GESP202312 六级] 工作沟通

题目描述

某公司有 $N$ 名员工，编号从 $0$ 至 $N-1$。其中，除了 $0$ 号员工是老板，其余每名员工都有一个直接领导。我们假设编号为 $i$ 的员工的直接领导是 $f_i$。

该公司有严格的管理制度，每位员工只能受到本人或直接领导或间接领导的管理。具体来说，规定员工 $x$ 可以管理员工 $y$，当且仅当 $x=y$，或 $x=f_y$，或 $x$ 可以管理 $f_y$。特别地，$0$ 号员工老板只能自我管理，无法由其他任何员工管理。

现在，有一些同事要开展合作，他们希望找到一位同事来主持这场合作，这位同事必须能够管理参与合作的所有同事。如果有多名满足这一条件的员工，他们希望找到编号最大的员工。你能帮帮他们吗？

输入格式

第一行一个整数 $N$，表示员工的数量。

第二行 $N-1$ 个用空格隔开的正整数，依次为 $f_1,f_2,\dots f_{N-1}$。

第三行一个整数 $Q$，表示共有 $Q$ 场合作需要安排。

接下来 $Q$ 行，每行描述一场合作：开头是一个整数 $m$（$2\le m\le N$），表示参与本次合作的员工数量；接着是 $m$ 个整数，依次表示参与本次合作的员工编号（保证编号合法且不重复）。

保证公司结构合法，即不存在任意一名员工，其本人是自己的直接或间接领导。

输出格式

输出 $Q$ 行，每行一个整数，依次为每场合作的主持人选。

样例 #1

样例输入 #1

5
0 0 2 2
3
2 3 4
3 2 3 4
2 1 4

样例输出 #1

2
2
0

样例 #2

样例输入 #2

7
0 1 0 2 1 2
5
2 4 6
2 4 5
3 4 5 6
4 2 4 5 6
2 3 4

样例输出 #2

2
1
1
1
0

提示

样例解释 1

对于第一场合作，员工$3,4$ 有共同领导 $2$ ，可以主持合作。

对于第二场合作，员工 $2$ 本人即可以管理所有参与者。

对于第三场合作，只有 $0$ 号老板才能管理所有员工。

数据范围

对于 $50\%$ 的测试点，保证 $N\le 50$。

对于所有测试点，保证 $3\le N\le 300$，$Q\le 100$。

2024/1/28 添加一组 hack 数据。

AC代码（100分）

#include<bits/stdc++.h>
using namespace std;
/*思路： 
	找到每个的员工的全部领导，用数组标记
	如果某个人被标记了m次，这个人是所有人的公共领导
	如果有多个，输出编号最大的 	
*/
int n,f[310],q,m,x,cnt[310];//cnt是标记数组 
//函数：找一个员工的所有领导，并用数组标记 
void find(int x){
	cnt[x]++;//员工可以自己管理自己 
	while(x!=0){
		x=f[x];//向上找领导
		cnt[x]++; 
	} 
}
int main(){
	cin>>n;
	for(int i=1;i<=n-1;i++) cin>>f[i];
	cin>>q;
	while(q--){
		memset(cnt,0,sizeof(cnt));//多组测试，每次测试前初始化cnt数组 
		cin>>m;
		for(int i=1;i<=m;i++){
			cin>>x;
			find(x);//每输入一个数，找其所有领导，进行标记 
		}
		int ans;
		for(int i=n-1;i>=0;i--){//从后往前找 
			if(cnt[i]==m){//找到的第一个 
				ans=i;//就是编号最大的公共领导 
				break;
			}
		}
		cout<<ans<<endl;
	}
	return 0;
}

文末彩蛋：

点击王老师青少年编程主页有更多精彩内容