LeetCode面试算法-力扣 582. 杀死进程

题目描述

    给n个进程，每个进程都有一个独一无二的PID(进程编号）和它的PPID(父进程编号）。每一个进程只有一个父进程，但是每个进程可能会有一个或者多个孩子进程。它们形成的关系就像一个树状结构。只有一个进程的PPID是0,意味着这个进程没有父进程。所有的PID都会是唯一的正整数。我们用两个序列来表示这些进程，第一个序列包含所有进程的PID,第二个序列包含所有进程对应的PPID。现在给定这两个序列和一个PID表示你要杀死的进程，函数返回一个 PID序列，表示因为杀这个进程而导致的所有被杀掉的进程的编号。当一个进程被杀掉的时候，它所有的孩子进程和后代进程都要被杀掉。你可以以任意顺序排列返回的PID序列。

---

示例 1：

    输入: pid = [1, 3, 10, 5]
             ppid = [3, 0, 5, 3]
             kill = 5
    输出: [5, 10]

---

public class Solution {
    public static List killProcess(List pid, List ppid, int kill) {
        Map> map = new HashMap<>();
        for (int i = 0; i < ppid.size(); i++) {
            List list = map.getOrDefault(ppid.get(i), new ArrayList<>());
            list.add(pid.get(i));
            map.put(ppid.get(i), list);
        }
        List result = new ArrayList<>();
        Queue queue = new LinkedList<>();
        queue.add(kill);
        while (!queue.isEmpty()) {
            int id = queue.poll();
            result.add(id);
            if (map.containsKey(id)) {
                queue.addAll(map.get(id));
            }
        }
        return result;
    }
}