leetcode582

Posted 2020-09-19 AsenYang

public class Solution {
    public IList<int> KillProcess(IList<int> pid, IList<int> ppid, int kill)
        {
            if (kill == 0)
            {
                return pid;
            }

            int n = pid.Count;
            Dictionary<int, List<int>> tree = new Dictionary<int, List<int>>();
            for (int i = 0; i < n; i++)
            {
                tree.Add(pid[i], new List<int>());
            }
            for (int i = 0; i < n; i++)
            {
                if (tree.ContainsKey(ppid[i]))
                {
                    var children = tree[ppid[i]];
                    children.Add(pid[i]);
                    if (!tree.ContainsKey(ppid[i]))
                    {
                        tree.Add(ppid[i], children);
                    }
                }
            }

            List<int> result = new List<int>();
            traverse(tree, result, kill);

            return result;
        }

        private void traverse(Dictionary<int, List<int>> tree, List<int> result, int pid)
        {
            result.Add(pid);

            var children = tree[pid];
            foreach (var child in children)
            {
                traverse(tree, result, child);
            }
        }
}

https://leetcode.com/problems/kill-process/#/solutions