【Leetcode】636. Exclusive Time of Functions

题目地址：

https://leetcode.com/problems/exclusive-time-of-functions/

对于一个单线程，它总共调用过$n$个不同的函数，给定其函数调用的log，这个log是个字符串数组，每个数组的格式形如"id:start:x"或者"id:end:x"，表示编号为id的函数在$x$这个时间戳开始（或结束）了。要求返回每个函数的独占的调用时间（也就是它们位于线程栈顶的时间。注意，开始和结束两个时刻都是属于调用函数独占时间的，这两个时刻所代表的时间，比如花了一秒，都要计入）。

思路是用栈，直接模拟函数调用过程，然后累加每个函数的独占时间。即，遇到函数开始，就把调用它的函数的独占时间累加一下，并把当前函数id入栈；遇到函数结束，则把它自己的独占时间累加一下，并且栈顶出栈（由计算机函数调用的设计，此时栈顶的id一定就是当前函数的id）。由于累加的时候，要知道栈顶函数的一段连续独占时间片段是啥时候开始的，这可以用一个变量记录一下。代码如下：

class Solution {
 public:
  vector<int> exclusiveTime(int n, vector<string>& logs) {
    vector<int> res(n);
    // 存正在调用的函数的id
    stack<int> stk;
    // last存的是栈顶函数的连续独占时间片段的开始时间
    int last = 0;
    for (auto& log : logs) {
      int pos1 = log.find(":");
      int pos2 = log.find(":", pos1 + 1);
      int id = stoi(log.substr(0, pos1));
      auto op = log.substr(pos1 + 1, pos2 - pos1 - 1);
      int time = stoi(log.substr(pos2 + 1));

      if (op == "start") {
        // 如果当前函数是开始，并且如果它是别的函数调用的，
        // 那么调用它的函数的时间应该累加一下
        if (!stk.empty()) res[stk.top()] += time - last;
        stk.push(id);
        last = time;
      } else {
        // 栈顶函数调用结束了，它的时间也要累加一下。注意这里要把调用开始的那一刻也要算进去
        res[stk.top()] += time - last + 1;
        stk.pop();
        // 此时栈顶变成了调用当前函数的那个函数，
        // 它的又一独占时间片段开始了，开始时间是当前函数结束的时间 + 1
        last = time + 1;
      }
    }
    return res;
  }
};

时空复杂度$O(N)$，$N$是log的长度，空间$O(n)$。