Kill Process Python Solution with Tests

lcpy gen -n 582   # by problem number
lcpy gen -s kill_process   # by problem name

Input: pid = [1,3,10,5], ppid = [3,0,5,3], kill = 5
Output: [5,10]
Explanation: The processes colored in red are the processes that should be killed.

Input: pid = [1], ppid = [0], kill = 1
Output: [1]

class Solution:
    # Time: O(n)
    # Space: O(n)
    def kill_process(self, pid: list[int], ppid: list[int], kill: int) -> list[int]:
        children: dict[int, list[int]] = {}
        for child, parent in zip(pid, ppid, strict=True):
            children.setdefault(parent, []).append(child)
        killed: list[int] = []
        stack = [kill]
        while stack:
            cur = stack.pop()
            killed.append(cur)
            stack.extend(children.get(cur, []))
        return killed