LintCode題解|Twitter 面試題：Tree

題目描述

給出兩個list x，y，表明x[i]與y[i]之間有一條邊，整個邊集構成一棵樹，1爲根。如今有個list a,b，表示詢問節點a[i]與b[i]是什麼關係，若是a[i]與b[i]是兄弟，即有同一個父節點，輸出1，若是a[i]與b[i]是父子關係，輸出2，不然輸出0。

思路點撥

一次dfs將樹的每一個節點的father處理出來，並保存每一個節點的兒子。

考點分析

本題考察了樹遍歷，一次樹遍歷就能將每一個節點的father處理出來，而後對每一個節點維護一個map，這個map存下全部的兒子，那麼判斷是不是兄弟關係，就判斷father是否相同，父子關係就用map判斷。

九章參考程序

www.jiuzhang.com/solution/tr…

/**
* 本參考程序來自九章算法，由 @斌助教 提供。版權全部，轉發請註明出處。
* - 九章算法致力於幫助更多中國人找到好的工做，教師團隊均來自硅谷和國內的一線大公司在職工程師。
* - 現有的面試培訓課程包括：九章算法班，系統設計班，算法強化班，Java入門與基礎算法班，Android 項目實戰班，
* - Big Data 項目實戰班，算法面試高頻題班, 動態規劃專題班
* - 更多詳情請見官方網站：http://www.jiuzhang.com/?source=code
*/ 

public class Solution {
    /**
     * @param x: The x
     * @param y: The y
     * @param a: The a
     * @param b: The b
     * @return: The Answer
     */
     
    void dfs(int x, int fa, List<List<Integer>>g, List<Map<Integer, Integer>>mp, int[] f) {
        f[x] = fa;
        for(int i = 0; i < g.get(x).size(); i++) {
            int y = g.get(x).get(i);
            if(y != fa) {
                mp.get(x).put(y, 1);
                dfs(y, x, g, mp, f);
            }
        }
    }
     
    public int[] tree(int[] x, int[] y, int[] a, int[] b) {
        // Write your code here
        List<List<Integer>>g = new ArrayList<List<Integer>>();
        List<Map<Integer, Integer>> mp = new ArrayList<Map<Integer, Integer>>();
        int[] f = new int[100005];
        int[] ans = new int[a.length];
        for(int i = 0; i < 100005; i++) {
            mp.add(new HashMap<Integer, Integer>());
        }
        for(int i = 0; i < x.length + 2; i++) {
            g.add(new ArrayList<Integer>());
        }
        for(int i = 0; i < x.length; i++) {
            g.get(x[i]).add(y[i]);
            g.get(y[i]).add(x[i]);
        }
        dfs(1, 0, g, mp, f);
        for(int i = 0; i < a.length; i++) {
            if(f[a[i]] == f[b[i]]) {
                ans[i] = 1;
            } else if(mp.get(a[i]).containsKey(b[i]) || mp.get(b[i]).containsKey(a[i])) {
                ans[i] = 2;
            } else {
                ans[i] = 0;
            }
        }
        return ans;
    }
}複製代碼