题目地址:
https://leetcode.com/problems/exclusive-time-of-functions/
对于一个单线程,它总共调用过
n
n
n个不同的函数,给定其函数调用的log,这个log是个字符串数组,每个数组的格式形如"id:start:x"
或者"id:end:x"
,表示编号为id的函数在
x
x
x这个时间戳开始(或结束)了。要求返回每个函数的独占的调用时间(也就是它们位于线程栈顶的时间。注意,开始和结束两个时刻都是属于调用函数独占时间的,这两个时刻所代表的时间,比如花了一秒,都要计入)。
思路是用栈,直接模拟函数调用过程,然后累加每个函数的独占时间。即,遇到函数开始,就把调用它的函数的独占时间累加一下,并把当前函数id入栈;遇到函数结束,则把它自己的独占时间累加一下,并且栈顶出栈(由计算机函数调用的设计,此时栈顶的id一定就是当前函数的id)。由于累加的时候,要知道栈顶函数的一段连续独占时间片段是啥时候开始的,这可以用一个变量记录一下。代码如下:
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.List;
public class Solution {
public int[] exclusiveTime(int n, List<String> logs) {
int[] res = new int[n];
// 存正在调用的函数的id
Deque<Integer> stack = new ArrayDeque<>();
// last存的是栈顶函数的连续独占时间片段的开始时间
int last = 0;
for (String log : logs) {
String[] split = log.split(":");
// 解析出函数id,是开始还是结束,以及时间戳
int id = Integer.parseInt(split[0]), time = Integer.parseInt(split[2]);
String op = split[1];
if ("start".equals(op)) {
// 如果当前函数是开始,并且如果它是别的函数调用的,那么调用它的函数的时间应该累加一下
if (!stack.isEmpty()) {
res[stack.peek()] += time - last;
}
stack.push(id);
last = time;
} else {
// 栈顶函数调用结束了,它的时间也要累加一下。注意这里要把调用开始的那一刻也要算进去
res[stack.pop()] += time - last + 1;
// 此时栈顶变成了调用当前函数的那个函数,
// 它的又一独占时间片段开始了,开始时间是当前函数结束的时间 + 1
last = time + 1;
}
}
return res;
}
}
时空复杂度 O ( N ) O(N) O(N), N N N是log的长度,空间 O ( n ) O(n) O(n)。