LeetCode 582. 杀死进程（图的遍历）

文章目录

1. 题目
2. 解题

2.1 DFS
2.2 BFS

1. 题目

给 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]
解释: 
           3
         /   \
        1     5
             /
            10
杀掉进程 5 ，同时它的后代进程 10 也被杀掉。
 
注意:
被杀掉的进程编号一定在 PID 序列中。
n >= 1.

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/kill-process
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

建图，遍历

2.1 DFS

class Solution {
	unordered_map<int, vector<int>> m;
	vector<int> ans;
public:
    vector<int> killProcess(vector<int>& pid, vector<int>& ppid, int kill) {
    	int i, root = -1;
    	for(i = 0; i < pid.size(); ++i)
    	{
    		if(ppid[i])
    			m[ppid[i]].push_back(pid[i]);
    	}
    	dfs(kill);
    	return ans;
    }

    void dfs(int root)
    {
    	ans.push_back(root);
    	for(int i = 0; i < m[root].size(); ++i)
    		dfs(m[root][i]);
    }
};

276 ms 67.7 MB

2.2 BFS

class Solution {
    unordered_map<int, vector<int>> m;
    vector<int> ans;
public:
    vector<int> killProcess(vector<int>& pid, vector<int>& ppid, int kill) {
        int i, cur;
        for(i = 0; i < pid.size(); ++i)
        {
            if(ppid[i])
                m[ppid[i]].push_back(pid[i]);
        }
        queue<int> q;
        q.push(kill);
        while(!q.empty())
        {
            cur = q.front();
            ans.push_back(cur);
            q.pop();
            for(int i = 0; i < m[cur].size(); ++i)
                q.push(m[cur][i]);
        }
        return ans;
    }
};

264 ms 65.3 MB

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！