AOJ2170 并查集的应用

1、题目大意

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

2、解题思路

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

3、代码

#include <iostream>
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<queryCount;i++){
			char operation;
			int nodeId;
			//去掉上一行的回车符 
			getchar();
			scanf("%c %d",&operation,&nodeId);		
			switch(operation){
				//标记节点 
				case mark:{
					markNode(nodeId);					
					break;
				}
				//查询节点 
				case query:{
					calculateNearestNodeId(nodeId);
					break;
				}
				//其他输入 
				default:{					
					break;
				}
			}
		}
		//输出结果 
		printf("%lld\n",sumOfAllQuery);					
	}
	return 0;
}

四、简单总结

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

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