582. Kill Process Medium

来源:互联网 发布:二元自动交易软件 编辑:程序博客网 时间:2024/05/19 19:14

这道题虽然形式上是树,其实用不到树的相关知识。因为是新题,还不知道以后有没有人写出更快的算法。这里参考答案中的C++代码。


首先要知道,map和set是STL中很有用的数据结构,但是平时用的不多所以不熟悉。map是红黑树实现的,可以定义key与value的一一对应关系,它查找的复杂度仅为O(log n),插入、删除操作都很快。set同样使用红黑树实现,插入、删除都很快,并且会自动排序。使用set和map可以大大降低算法复杂度。


这道题我们先梳理一下数据,用set来存储每一个节点的子节点,然后用map让set与每一个节点对应起来。然后用DFS或BFS都可以。这里给出DFS的做法。


class Solution {public:    vector<int> killProcess(vector<int>& pid, vector<int>& ppid, int kill) {        vector<int> killed;        map<int, set<int>> children;        for (int i = 0; i < pid.size(); i++) {            children[ppid[i]].insert(pid[i]);        }        queue<int> q;        q.push(kill);        while (!q.empty()) {            int p = q.front(); q.pop();            killed.push_back(p);            for (int child : children[p]) {                q.push(child);            }        }        return killed;    }};

0 0
原创粉丝点击