力扣(LeetCode)863

Linus脫襪子發表於2019-01-19

題目地址:
https://leetcode-cn.com/probl…
題目描述:
給定一個二叉樹(具有根結點 root), 一個目標結點 target ,和一個整數值 K 。

返回到目標結點 target 距離為 K 的所有結點的值的列表。 答案可以以任何順序返回。

解答:
這一題顯然是利用廣度優先搜尋來探測距離,然而樹是一個有向圖,我們需要一個無向圖來進行廣度優先搜尋。
因此,先把樹變成一個無向圖,然後廣度優先搜尋這個無向圖。距離不斷增加,就可以找到每個節點和目標節點的距離。
java ac程式碼:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> distanceK(TreeNode root, TreeNode target, int K) {
        List<Integer> ans = new ArrayList(1000);
      
        HashMap<TreeNode,List<TreeNode>>map = new HashMap(1<<10);
        dfs1(root,map);
        dfs2(root,map);
        ArrayDeque<TreeNode> deque = new ArrayDeque(100);
        HashSet<TreeNode>set = new HashSet(1<<10);
        HashMap<TreeNode,Integer> map2 = new HashMap(1<<10);
        set.add(target);
        deque.offer(target);
        map2.put(target,0);
        while(!deque.isEmpty())
        {
            TreeNode temp = deque.poll();
            int num = map2.get(temp)+1;
            for(TreeNode node:map.get(temp))
                if(!set.contains(node))
                {
                    set.add(node);
                    deque.offer(node);
                    map2.put(node,num);
                }
        }
        for(Map.Entry<TreeNode,Integer> entry:map2.entrySet())
            if(entry.getValue() == K)ans.add(entry.getKey().val);
        return ans;
        
        
    }
    void dfs1(TreeNode root,HashMap<TreeNode,List<TreeNode>>map)
    {
        if(root == null)return;
        map.put(root,new ArrayList(100));
        dfs1(root.left,map);
        dfs1(root.right,map);
    }
    
    void dfs2(TreeNode root,HashMap<TreeNode,List<TreeNode>>map)
    {
        if(root == null)return;
        if(root.left != null)
        {
            map.get(root).add(root.left);
            map.get(root.left).add(root);
            
        }
        if(root.right != null)
        {
            map.get(root).add(root.right);
            map.get(root.right).add(root);
        }
        dfs2(root.left,map);
        dfs2(root.right,map);
        
    }
    
}

相關文章