Java——百度2021校招C /PHP研发工程师笔试卷（第三批）[编程题]树上最短路

牛牛刚刚学了二叉树和最短路
他现在很好奇，对于一棵完全二叉树（可以看例子说明了解完全二叉树），给定的两个点之间的最短路的长度是多少

我们认为相邻两点之间的距离为1

输入描述:

第一行输入一个整数T，表示测试数据共有T组
对于每组测试数据，输入两个整数表示所询问的节点的标号

输出描述:

对于每组数据，输出一个整数表示答案

输入例子1:

3

1 2

4 14

8 5

输出例子1:

1

5

3

例子说明1:

 上图就是完全二叉树的样子，如果图片显示不出来可以自行搜索。

解题思路：

在全为整数的完全二叉树中，子节点/2 就是父节点的值 （例如 8的父节点为4， 8/2 = 4）

以 输入一组数据 8 9为例

使用2个ArrayList从大到小保存输入值的所有父节点和其自身

如 8 对应的arrayList1为 {8，4，2，1} 9对应的arrayList2为{9，4，2，1}，最近的相同父节点为4

4在arrayList1下标为1，在arrayList2下标为1，则距离为1+1=2.

代码：

import java.util.*;

public class test3 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = Integer.parseInt(scanner.nextLine());
        StringBuffer stringBuffer1 = new StringBuffer();
        for (int i = 0; i < n; i++) {
            String line = scanner.nextLine();
            stringBuffer1.append(line);
            stringBuffer1.append(" ");
        }
        String[] strings = stringBuffer1.toString().split(" ");
        for (int i = 0; i < n; i++) {
            System.out.println(getSteps(strings[2*i],strings[2*i+1]));
        }

    }

    private static int getSteps(String one,String two){
        int i1 = Integer.parseInt(one);
        int i2 = Integer.parseInt(two);
        ArrayList<Integer> arrayList1 = getList(i1);
        ArrayList<Integer> arrayList2 = getList(i2);
        int i = checkArray(arrayList1,arrayList2);
        return arrayList1.indexOf(i)+arrayList2.indexOf(i);
    }

    //找出最近的相同父节点的值
    private static int checkArray(ArrayList<Integer> arrayList1,ArrayList<Integer> arrayList2){
        int i = 1;
        for (Integer i11:arrayList1) {
            for (Integer i22:arrayList2) {
                if (i11==i22){
                    i=i11;
                    return i;
                }
            }
        }
        return i;
    }

    //使用ArrayList从大到小保存输入值的所有父节点和其自身
    private static ArrayList<Integer> getList(int i){
        ArrayList<Integer> arrayList = new ArrayList<>();
        if (i==1){
            arrayList.add(i);
            return arrayList;
        }
        arrayList.add(i);
        while ((i/=2)!=1){
            arrayList.add(i);
        }
        arrayList.add(1);
        arrayList.sort(Comparator.reverseOrder());
        return arrayList;
    }
}

运行结果：

3
1 2
4 14
8 5
1
5
3

结果全部正确。

总结：

总之，以这种奇怪的方式做出来了（灵感突然就来了）。不知道大家有没有更好的解法。

我会分享一些有意思的算法题，与君共勉。如果有更好的解题思路欢迎分享，如果本文有不正确的地方欢迎指正。