1. 程式人生 > >通過按層遍歷求等權無向圖的兩點間的最短路徑

通過按層遍歷求等權無向圖的兩點間的最短路徑

    /**
     * 
     * @param a
     *            Person a
     * @param b
     *            Person b
     * @return a與b之間的距離,返回 0 如果a=b,返回 -1 如果a與b距離為無窮大,
     */
    public int getDistance(Person a, Person b) {
        if (a.equals(b)) {
            return 0;
        } else {
            int d = 1;
            Set<Person> set = new HashSet<Person>();// 存已遍歷過的person
            List<Person> list1 = new ArrayList<>();// 存下一次的person
            list1.add(a);
            set.add(a);
            //按層遍歷無向圖
            while (true) {
                List<Person> list2 = list1;// 將list1賦給list2
                list1 = new ArrayList<Person>();// 初始化list1
                // 退出while迴圈條件
                if (list2.isEmpty())
                    break;
                for (int i = 0; i < list2.size(); i++) {
                    Person temp = list2.get(i);
/*
*fgraph即無向圖的資料
 for (Person key : fgraph.targets(temp).keySet()) {
                        //確保該節點未遍歷過
                          if (!set.contains(key)) {

                            if (key.equals(b)) {
                                return d;
                            } else {
                                set.add(key);
                                list1.add(key);
                            }
                        }
                    }
                }
                d++;
            }
            return -1;

        }
    }

這是本人的實驗二作業時,本來想直接用Floyd演算法,但是想嘗試新的方法,作為新手還是小有成就感的。

思路:以兩點間的一點a作為根節點生成樹,然後按層遍歷,若遍歷到另一點b,則停止遍歷,並記錄此時的層數n,通過n便可得出a,b兩點的最短路徑,若遍歷整棵數未遇到b,則說明a,b兩點距離為無窮大。

侷限性:只能處理等權的圖。