编程问题:
给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间,CPU 可以完成一个任务,或者处于待命状态。
然而,两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。
你需要计算完成所有任务所需要的 最短时间 。
示例:
- 输入:[73, 74, 75, 71, 69, 72, 76, 73]
- 输出:[1, 1, 4, 2, 1, 1, 0, 0]
解法:
1.暴力搜索
循环数组中每一个温度,对于每一个温度分别再其后面找第一个大于它的温度,如果找到则计算索引差值,否则,取0。
时间复杂度O(N^2)
空间复杂度O(1)
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> res;
int i, j;
for(i = 0; i < T.size(); i++)
{
for (j = i + 1; j < T.size(); j++)
{
if(T[i] < T[j])
{
res.push_back(j-i);
break;
}
}
if(j == T.size())
res.push_back(0);
}
return res;
}
};
2.递减栈
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离。直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来。
时间复杂度O(N),其中 N 是温度列表的长度。正向遍历温度列表一遍,对于温度列表中的每个下标,最多有一次进栈和出栈的操作。
空间复杂度O(N),需要维护一个单调栈存储温度列表中的下标。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> res(T.size(), 0);
stack<int> s;
for (int i = 0; i < T.size(); i++)
{
while(!s.empty() && T[i] > T[s.top()])
{
res[s.top()] = i - s.top();
s.pop();
}
s.push(i);
}
return res;
}
};