https://leetcode-cn.com/problems/next-greater-element-ii/
自己写的就是真的太慢了,速度才超过5%,==
不过全都是自己写的,就比较有成就感。
就是有很多可以剪枝的地方吧,大概。当你确定了前面一个元素他的更大元素在某个位置,那么根据数量关系当前要寻找的元素的更大元素的范围,是可以缩小的,感觉我的代码可能有一些赘余的操作吧,所以就很慢。
但是我的空间复杂度是真的小。(得意!
都写在了代码里面。
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int des=1;//标记要比较的跨越步长
int n=nums.size();//元素的个数
vector<int> res(n,-1);//返回答案的容器
int max=n;//des的自加上限
for(int i=0;i<n;){//从0开始逐步比较
if(nums[(des+i)%n]>nums[i]||des==max){//当找到一个更大的值的时候,或者当已经无路可走的时候
res[i]=nums[(des+i)%n];//答案都可以确定为
if(res[i]==nums[i]) res[i]=-1;
i++;
while(i<n&&nums[i-1]==nums[i]){
//用于简化相等时候的一切都可连等下去
res[i]=res[i-1];
des--;
i++;
}
if(i==n) break;//当确定了所有的值时候
if(nums[i]>nums[i-1]) max=n;//
else max=des-1;//否则若是小于它,则不会超过上一个的步长减去一个1
des=1;//步长初始化
}
else if(max>des) des++;
}
return res;
}
};
实际上用单调栈能更快如下:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n = nums.size();
vector<int> ret(n, -1);
stack<int> stk;
for (int i = 0; i < n * 2 - 1; i++) {
while (!stk.empty() && nums[stk.top()] < nums[i % n]) {
ret[stk.top()] = nums[i % n];
stk.pop();
}
stk.push(i % n);
}
return ret;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/next-greater-element-ii/solution/xia-yi-ge-geng-da-yuan-su-ii-by-leetcode-bwam/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。