题目描述
给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<Integer> killProcess(List<Integer> pid, List<Integer> ppid, int kill) {
Map<Integer, List<Integer>> map = new HashMap<>();
for (int i = 0; i < ppid.size(); i++) {
List<Integer> list = map.getOrDefault(ppid.get(i), new ArrayList<>());
list.add(pid.get(i));
map.put(ppid.get(i), list);
}
List<Integer> result = new ArrayList<>();
Queue<Integer> 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;
}
}