Java解leetcode,助力面试之中等10道题(十)
第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
解题思路
本题使用并查集,查询两个节点是否属于同一连结分量,如果不属于,则合并两个节点为同一索引,如果属于同一分量,则返回该连结分量
代码
// 冗余连接:并查集
class Solution {
public int[] findRedundantConnection(int[][] edges) {
int nodesCount = edges.length;
int[] parent = new int[nodesCount + 1];
for (int i = 1; i <= nodesCount; i++) {
parent[i] = i;
}
for (int i = 0; i < nodesCount; i++) {
int[] edge = edges[i];
int node1 = edge[0], node2 = edge[1];
if (find(parent, node1) != find(parent, node2)) {
//判断两个顶点是否属于相同连接分量
union(parent, node1, node2);//不属于则合并节点
} else {
return edge;//如果属于相同连结分量,则返回该连结分量
}
}
return new int[0];
}
public void union(int[] parent, int index1, int index2) {
//合并
parent[find(parent, index1)] = find(parent, index2);//如果两个节点不属于相同连接分量,则合并一个节点的索引为另一个节点的值
}
public int find(int[] parent, int index) {
//查找
if (parent[index] != index) {
//如果当前节点的索引不是本身,则说明合并过,找它的父节点
parent[index] = find(parent, parent[index]);
}
return parent[index];
}
}
时间复杂度为O(nlog n),n为图中节点数
空间复杂度为O(n)
第687题 最长同值路径
给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例 1:
5
/ \
4 5
/ \ \
1 1 5
输入 | 输出 |
---|---|
[5,4,5,1,1,null,5] | 2 |
示例 2:
1
/ \
4 5
/ \ \
4 4 5
输入 | 输出 |
---|---|
[1,4,5,4,4,null,5] | 2 |
解题思路
递归查找,判断左右节点值是否等于根节点的值,如果不等于继续递归判断左右子节点,如果有相等的值,则将结果增加。
代码
// 最长同值路径:递归
class Solution {
int ans;
public int longestUnivaluePath(TreeNode root) {
ans = 0;
arrowLength(root);
return ans;
}
public int arrowLength(TreeNode node) {
if (node == null) return 0;
int left = arrowLength(node.left);//查询左右节点
int right = arrowLength(node.right);
int arrowLeft = 0, arrowRight = 0;
if (node.left != null && node.left.val == node.val) {
//如果左节点等于根节点,则将arrowleft加1
arrowLeft += left + 1;
}
if (node.right != null && node.right.val == node.val) {
//同理
arrowRight += right + 1;
}
ans = Math.