并查集 leetcode 684 冗余连接

题目内容

在本问题中, 树指的是一个连通且无环的无向图。

输入一个图，该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组。每一个边的元素是一对[u, v] ，满足 u < v，表示连接顶点u 和v的无向图的边。

返回一条可以删去的边，使得结果图是一个有着N个节点的树。如果有多个答案，则返回二维数组中最后出现的边。答案边 [u, v] 应满足相同的格式 u < v。

示例 1：

输入: [[1,2], [1,3], [2,3]]
输出: [2,3]
解释: 给定的无向图为:
1
/
2 - 3

示例 2：

输入: [[1,2], [2,3], [3,4], [1,4], [1,5]]
输出: [1,4]
解释: 给定的无向图为:
5 - 1 - 2
| |
4 - 3

注意:

输入的二维数组大小在 3 到 1000。
二维数组中的整数在1到N之间，其中N是输入数组的大小。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/redundant-connection

c语言解答

//初始化parent ranke数组,让其设置为-1 
void initialise(int parent[], int rank[],int edgesSize){
	int i;
	for(i=0;i<=edgesSize;i++){
		parent[i]=-1;
		//初始高度设置为0 
		rank[i]=0;
	}
}

// 找根节点 
int find_root(int x, int parent[]){
	//假设根节点是自己 
	int x_root = x;
	//如果根节点不是自己,一直找到根节点 
	while(parent[x_root] != -1){
		x_root = parent[x_root];
	}
	//返回根节点 
	return x_root;
}

//合并两个结点,如果两个结点没有共同的根节点,就可以成功合并返回1,否则返回0 
/* 1 - union successfully, 0 -failed*/
int union_vertices(int x, int y,int parent[],int rank[]){
	//找到x,y的根节点 
	int x_root=find_root(x,parent);
	int y_root=find_root(y,parent);
	//比较根节点是否相同 ,相同返回0 
	if(x_root == y_root){
		return 0;
	}
	// 根节点不同进行合并,然后返回1 
	else{
		if(rank[x_root]>rank[y_root]){
			parent[y_root] = x_root;
		}
		else if(rank[y_root]>rank[x_root]){
			parent[x_root] = y_root;
		}
		//高度一致的时候 
		else{
			parent[x_root] = y_root;
			rank[y_root]++;
        }
	} 
    return 1;
	
} 
int* findRedundantConnection(int** edges, int edgesSize, int* edgesColSize, int* returnSize){
    int parent[edgesSize+1],rank[edgesSize+1];
    initialise(parent,rank,edgesSize);
    int i;
    for(i=0;i<edgesSize;i++){
    // x最大值就是边数,因为1-edgesSize的数组成.
		int x = edges[i][0];
		int y = edges[i][1];
		//有环的话就会打印有环,从而退出程序 
		if(union_vertices(x,y,parent,rank)==0){
            *returnSize=2;
			return edges[i];	
		}
	}
    *returnSize=0;
    return NULL;

}

总结:

刚开始不明白为什么parent数组的大小要加1,题中边数不就等于结点数么,现在明白了,好比边数是5的话,你parent只能初始化0 1 2 3 4,但是这道题边数是从1开始的.该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树,所以必须加上1,因为x的值可能是5,这样会导致数组越界的问题.