【leecode每日一题】636. 函数的独占时间

题目描述(链接

有一个 单线程 CPU 正在运行一个含有 n 道函数的程序。每道函数都有一个位于 0 和 n-1 之间的唯一标识符。

函数调用 存储在一个 调用栈 上 :当一个函数调用开始时,它的标识符将会推入栈中。而当一个函数调用结束时,它的标识符将会从栈中弹出。标识符位于栈顶的函数是 当前正在执行的函数 。每当一个函数开始或者结束时,将会记录一条日志,包括函数标识符、是开始还是结束、以及相应的时间戳。

给你一个由日志组成的列表 logs ,其中 logs[i] 表示第 i 条日志消息,该消息是一个按 “{function_id}:{“start” | “end”}:{timestamp}” 进行格式化的字符串。例如,“0:start:3” 意味着标识符为 0 的函数调用在时间戳 3 的 起始开始执行 ;而 “1?2” 意味着标识符为 1 的函数调用在时间戳 2 的 末尾结束执行。注意,函数可以 调用多次,可能存在递归调用 。

函数的 独占时间 定义是在这个函数在程序所有函数调用中执行时间的总和,调用其他函数花费的时间不算该函数的独占时间。例如,如果一个函数被调用两次,一次调用执行 2 单位时间,另一次调用执行 1 单位时间,那么该函数的 独占时间 为 2 + 1 = 3 。

以数组形式返回每个函数的 独占时间 ,其中第 i 个下标对应的值表示标识符 i 的函数的独占时间。

示例

示例一:

输入:n = 2, logs = ["0:start:0","1:start:2","1:end:5","0:end:6"]
输出:[3,4]
解释:
函数 0 在时间戳 0 的起始开始执行,执行 2 个单位时间,于时间戳 1 的末尾结束执行。 
函数 1 在时间戳 2 的起始开始执行,执行 4 个单位时间,于时间戳 5 的末尾结束执行。 
函数 0 在时间戳 6 的开始恢复执行,执行 1 个单位时间。 
所以函数 0 总共执行 2 + 1 = 3 个单位时间,函数 1 总共执行 4 个单位时间。

示例二:

输入:n = 1, logs = ["0:start:0","0:start:2","0:end:5","0:start:6","0:end:6","0:end:7"]
输出:[8]
解释:
函数 0 在时间戳 0 的起始开始执行,执行 2 个单位时间,并递归调用它自身。
函数 0(递归调用)在时间戳 2 的起始开始执行,执行 4 个单位时间。
函数 0(初始调用)恢复执行,并立刻再次调用它自身。
函数 0(第二次递归调用)在时间戳 6 的起始开始执行,执行 1 个单位时间。
函数 0(初始调用)在时间戳 7 的起始恢复执行,执行 1 个单位时间。
所以函数 0 总共执行 2 + 4 + 1 + 1 = 8 个单位时间。

示例三:

输入:n = 2, logs = ["0:start:0","0:start:2","0:end:5","1:start:6","1:end:6","0:end:7"]
输出:[7,1]
解释:
函数 0 在时间戳 0 的起始开始执行,执行 2 个单位时间,并递归调用它自身。
函数 0(递归调用)在时间戳 2 的起始开始执行,执行 4 个单位时间。
函数 0(初始调用)恢复执行,并立刻调用函数 1 。
函数 1在时间戳 6 的起始开始执行,执行 1 个单位时间,于时间戳 6 的末尾结束执行。
函数 0(初始调用)在时间戳 7 的起始恢复执行,执行 1 个单位时间,于时间戳 7 的末尾结束执行。
所以函数 0 总共执行 2 + 4 + 1 = 7 个单位时间,函数 1 总共执行 1 个单位时间。 

思路说明

在这里使用双栈来实现,stack1用来统计函数编号,stack2来统计函数的开始时间,end时间不需要统计,因为start我们进栈操作,end时出栈操作(并将该函数执行时间加入对应ans位置)。
注意的是,在出栈是需要给栈内元素加上已经出栈函数的耗时。

代码

public int[] exclusiveTime(int n, List<String> logs) {
        int[] ans = new int[n];
        // flag 用来标记给stack2栈中元素需要+的值
        int flag = 0;
        Stack<String> stack1 = new Stack<>();// 每个函数编号栈
        Stack<Integer> stack2 = new Stack<>();// start 栈,start进栈,end出栈
        while (!logs.isEmpty()) {
            String temp = logs.get(0);
            String[] strings = temp.split(":");
            if (strings[1].equals("start")) {
                // 进栈
                stack1.add(strings[0]);
                stack2.add(Integer.valueOf(strings[2]));
            } else {
                // 出栈
                ans[Integer.valueOf(strings[0])] += Integer.valueOf(strings[2]) - stack2.peek() + 1;// ans对应位置加函数的耗时
                flag = Integer.valueOf(strings[2]) - stack2.peek() + 1; // 当有函数出栈,就需要给里边的值+flag,因为单线程,所以+flag相当于给栈内函数减去其他函数的耗时
                Stack<Integer> te = new Stack<>();
                // +flag操作
                while (!stack2.isEmpty()) {
                    te.add(stack2.pop() + flag);
                }
                while (!te.isEmpty()) {
                    stack2.add(te.pop());
                }
                stack1.pop();
                stack2.pop();
            }
            logs.remove(0);
        }
        return ans;
    }

版权声明:本文为weixin_45532227原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。