AOJ2170 并查集的应用

1、题目大意

我被给予一颗树T,树上有N个节点,节点被从1到N标号。1号节点总是节点中的根节点,
对于每个节点,有以下两种操作
M v,标记节点v
Q v。输出距离v最近的祖先节点。最开始,只有根节点1被标记,每一个点都是它自己的祖先。
你的工作是写一个程序,计算出所有查询需要输出的总和。

2、解题思路

本题目虽然被列为并查集,但是题目本身和并查集的关系并不是很大,也没有用到路径压缩。思路就是给每个节点两个属性。一个是父亲节点编号,另一个是节点是否被标记。这样通过简单的for循环就可以解题。

3、代码

#include 
using namespace std;
//最大的节点数量 
const int MAX_NODE=100009;
//最多查询次数 
const int MAX_QUERY=100009;
//m代表标记节点 
const char mark='M';
//q代表查询节点 
const char query='Q';
//节点数组 
int ancestorOfNode[MAX_NODE];
//是否标记节点
bool isMarkedNode[MAX_NODE]; 
//最终输出的结果 
long long sumOfAllQuery=0LL;
//节点的数量 
int nodeCount;
//查询的数量 
int queryCount;
//输入数据 
void getInputData(){
	scanf("%d%d",&nodeCount,&queryCount);
	int fatherOfNodei;
	//输入根节点以外的每一个节点的父节点 
	for(int i=2;i<=nodeCount;i++){
		scanf("%d",&fatherOfNodei);
		ancestorOfNode[i]=fatherOfNodei;
	}
}
//初始化,设置每个节点都为未标记 
void initData(){
	//初始化结果为0 
	sumOfAllQuery=0LL;
	//根节点的父节点是它本身
	ancestorOfNode[1]=1;
	isMarkedNode[1]=true;
	//1号节点为根节点,始终被标记 
	for(int i=2;i<=nodeCount;i++){
		isMarkedNode[i]=false;
	}
}
//标记节点 
void markNode(int nodeId){
	isMarkedNode[nodeId]=true;
}
//计算最近的祖先节点ID 
void calculateNearestNodeId(int nodeId){
	int currentNodeId=nodeId;
	while(true){
		if(isMarkedNode[currentNodeId]){
			sumOfAllQuery=sumOfAllQuery+currentNodeId;
			break;
		}else{
			currentNodeId=ancestorOfNode[currentNodeId];
		}
	}
}
int main(){
	while(true){
		//输入数据 
		getInputData();
		//初始化数据 
		initData();
		//节点数和查询数都为零,则程序结束 
		if(nodeCount==0&&queryCount==0){
			break;
		}
		for(int i=0;i

四、简单总结

在比赛中时间紧迫可以忽略代码的可读性,但是平时练习中,可以提高一下代码的可读性,既方便以后复习,也方便滤清接下里的思路。

代码的关键在于小巧灵活,解题的关键在于逻辑清晰。

你可能感兴趣的:(ACM,简单题目,算法,c++,数据结构)