题目描述
n个节点n-1条边的无向连通图,两个点a,b,a位于1,b位于x,两点移动速度相同,求a和b移动到同一节点所需的最多节点数。
输入
第一个参数为 n ,(1≤n≤100,000)
第二个参数为 x ,(1≤x≤n)
第三个参数为大小为 n−1 的点对 的集合,其中 表示结点ui与结点vi
之间有一条边,1≤ui,vi≤n
返回
最多需要经过的节点数(包括 1 号节点在内)
示例1
输入:5,2,[(1,2),(2,3),(3,4),(2,5)]
返回值:4
题目分析
连通图的概念是图中的任意两点之间都能有路径到达,当n个结点只有n-1条边形成的连通图一定是树状的,题目设定两个点a,b,a点固定从1出发,b点会从给定的x处出发,要想两点移动到同一点,可以获取从1到所有其他结点的距离dis1[]以及从x到其他结点的距离dis2[],求同一点的dis1[i]和dis2[i]的移动结点数,因为一定要包括1号结点,所以考虑dis1[i]中的最大结点数。
解题思路
方法1:DFS遍历整个树,获取每个结点到1和x的结点距离
dfs的思路比较直观:
1.创建邻接表graph,根据Edge数组初始化邻接表,创建结点数组dis1和dis2来记录到某个结点经过的结点数;
2.可以根据dis[i]是否为0 来标记结点是否已经访问过;
3.dfs从根开始遍历,访问根的子节点(邻接表),每经过一个结点,都要增加一个结点数;
4.遍历dis1和dis2,求出到i结点的最大结点数。
方法2:BFS遍历整个树,获取每个结点到1和x的结点距离
dfs遍历会需要较多的栈空间,一旦超过设定的栈深度,就会报栈溢出错误,使用bfs可以避免;
代码实现
方法1:DFS(会报栈溢出错误)
//邻接表 List<List<Integer>> graph; // 从1结点到某结点路径上结点的个数 int[] dis1; // 从x结点到某结点路径上结点的个数 int[] dis2; public int solve (int n, int x, Point[] Edge) { // 初始化邻接表 graph = new ArrayList<>(); for(int i=0;i<n;i++){ graph.add(new ArrayList<>()); } for(int i=0;i<Edge.length;i++){ int u = Edge[i].x-1; int v = Edge[i].y-1; graph.get(u).add(v); graph.get(v).add(u); } // 初始化结点数组 dis1 = new int[n]; dis2 = new int[n]; // 初始化1到1(下标对应0)的距离为1 dis1[0] = 1; // 初始化x到x(下标对应x-1)的距离为1 dis2[x-1] = 1; // dfs递归遍历,获得从1结点到其他路径上结点的个数 dfs(0, dis1); // dfs递归遍历,获得从x结点到其他路径上结点的个数 dfs(x-1, dis2); int res = 0; for(int i=0;i<n;i++){ if(dis1[i] > dis2[i]){ res = Math.max(res, dis1[i]); } } return res; } public void dfs(int i, int[] dis){ for(Integer next:graph.get(i)){ // 从i到next结点未初始化,则未经过 if(dis[next] == 0){ // 到next经过了一个i结点 dis[next] = dis[i] + 1; dfs(next, dis); } } }
时间复杂度:,首先需要创建长度为n的邻接表,同时也要遍历Query数组,在dfs的过程中,在每个结点都只有一个子节点的情况下,递归深度达到最大n,所以花费的总时间为n级别;
空间复杂度:,空间上需要2(n-1)来存邻接结点(n个结点的树有n-1条边,边上的两个结点都会在邻接表中重复出现,所以是2(n-1)),还需要深度最大为n的递归栈空间,因为栈深度的限制,使用dfs时会报栈溢出错误;
方法2:BFS(借助队列,避免栈溢出)
//邻接表 List<List<Integer>> graph; // 从1结点到某结点路径上结点的个数 int[] dis1; // 从x结点到某结点路径上结点的个数 int[] dis2; public int solve (int n, int x, Point[] Edge) { // 初始化邻接表 graph = new ArrayList<>(); for(int i=0;i<n;i++){ graph.add(new ArrayList<>()); } for(int i=0;i<Edge.length;i++){ int u = Edge[i].x-1; int v = Edge[i].y-1; graph.get(u).add(v); graph.get(v).add(u); } // 初始化结点数组 dis1 = new int[n]; dis2 = new int[n]; // 初始化1到1(下标对应0)的距离为1 dis1[0] = 1; // 初始化x到x(下标对应x-1)的距离为1 dis2[x-1] = 1; // dfs递归遍历,获得从1结点到其他路径上结点的个数 bfs(0, dis1); // dfs递归遍历,获得从x结点到其他路径上结点的个数 bfs(x-1, dis2); int res = 0; for(int i=0;i<n;i++){ if(dis1[i] > dis2[i]){ res = Math.max(res, dis1[i]); } } return res; } public void bfs(int i, int[] dis){ //借助队列 Queue<int[]> queue=new LinkedList<>(); // 队列中存储结点和结点距离 queue.offer(new int[]{i,dis[i]}); while(!queue.isEmpty()){ // 取出当前节点 int[] node=queue.poll(); int id=node[0]; int d=node[1]; // 遍历所有邻结点 for(Integer next:graph.get(id)){ if(dis[next]==0){ // 当还有子节点时,放入队列中,且距离加上该结点 dis[next]=d+1; queue.offer(new int[]{next,dis[next]}); } } } }
时间复杂度:,与dfs类似需要遍历数组,bfs使用的队列,省掉了递归的栈时间,增加了遍历邻接表的时间,时间复杂度为n级别;
空间复杂度:,与dfs相似,减少了栈空间,增加了队列来存储结点值,空间复杂度级别为n。