Question

我试图找到树中两个节点之间的最大距离。这是我的计划：

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;


public class distance {

    static ArrayList<ArrayList<Integer>> list=new ArrayList<ArrayList<Integer>>();
    static ArrayList<Integer> visited=new ArrayList<Integer>();
    static ArrayList<Integer> depth=new ArrayList<Integer>();
    static ArrayList<Integer> depth1=new ArrayList<Integer>();
    static int sum=-1;
    static boolean[] arr;
    static int root=0;


    public static void main(String args[])
    {
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();     //no of nodes

        for(int i=0;i<=n;i++)   
        {
            list.add(new ArrayList<Integer>());
        }
        int a;
        int b;
        for(int i=0;i<n-1;i++)  //populating the adjacency list
        {
            a=in.nextInt();
            b=in.nextInt();
            list.get(a).add(b);
            list.get(b).add(a);
        }

        arr=new boolean[n+1];

        dfs(root);

        int final_sum=0;
        Collections.sort(depth1);
        System.out.println(depth1.get(depth1.size()-1)+depth1.get(depth1.size()-2));

        }
    public static void dfs(int n)
    {   
        arr[n]=true;
        visited.add(n);
        sum=sum+1;

        if(list.get(n).size()==1)
        {   
            depth.add(sum); //add the depth to the arraylist when we reach a leaf node.Note the this will not run if the root node has only one child but I've ignored the case.
        }
        if(n==root)
        {
            for(int j=0;j<list.get(0).size();j++)
            {   
                dfs(list.get(0).get(j)); //recursion on each child of the root node
                sum=0;
                Collections.sort(depth);
                depth1.add(depth.get(depth.size()-1));
                depth.clear();
            }

        }
        else
        {
            for(int l:list.get(n))
            if(!arr[l]==true)
            {
                dfs(l);
                sum--;

            }
        }


    }

}

该程序似乎正在运行;但是没有显示某些测试用例的正确输出。我采取的方法是：

查找根节点的子节点数（我总是将根节点设为0）。
查找每个子树的最大深度（与根节点的子节点一样多的子树）。
将每个子树的最大深度存储在ArrayList中，对其进行排序并打印最后两个值的总和。

有人可以指出我程序中的错误吗？

Answer 1

错误在算法中首先出现。您假设两个节点之间的最大距离始终包含根节点，而这并不总是成立。

以下是一个例子：

最长路径中的节点标有红色。最长路径的长度为7并包含5个节点，而您的算法只查找通过根的路径，因此会打印<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android" android:id="@+id/layoutHdr" android:layout_width="match_parent" android:layout_height="wrap_content" android:orientation="horizontal" > <ImageView android:layout_width="@dimen/icon_size_small" android:layout_height="@dimen/icon_size_small" android:layout_marginLeft="15dp" android:gravity="center_horizontal" android:src="@drawable/appicon" /> <TextView android:id="@+id/permissionHeader" android:layout_width="match_parent" android:layout_height="wrap_content" android:layout_marginLeft="10dp" android:layout_marginRight="10dp" android:gravity="left" android:text="@string/permission_popup_mandate_header" android:textColor="@color/black" android:textSize="@dimen/font_size_dialog_title" android:textStyle="bold" />作为答案。

树中两个节点之间的最大距离

1 个答案: