题目:
找到数组中每个元素后面比它大的第一个数,然后求这个元素与后面第一个比他大的数之间的索引间隔,没找到的话返回-1。
解法1:
#include<iostream>
#include<vector>
#include<stack>
using namespace std;
//解法1:直接遍历求解
vector<int> FindMaxDistance(vector<int>nums){
vector<int>res(nums.size(),-1);
for(int i = 0; i < nums.size();i++){
for(int j = i + 1; j < nums.size(); j++){
if(nums[j] > nums[i]){
res[i] = j - i;
break;
}else{
j++;
}
}
}
for(int i = 0; i < nums.size(); i++){
cout << res[i] << " ";
}
return res;
}
int main(){
vector<int> num = {6,3, 1, 2, 4, 5};
FindMaxDistance(num);
return 0;
}
解法2:
//解法2:栈
//如果我当前的处理对象是第1个元素,如果第2个元素比我小,那么我现在要做的不是去比较第3个元素与第1个元素的关系,而是将处理的对象变成第2个元素。如果第3个元素比第2个元素大,我在回过来比较第1个元素,这样是不是就省去了很多时间?
//因此我们需要一个容器来存储未处理的元素,可以看到,元素是后进先出的,比如第2个元素,后后到来却是先得到结果的。因此我们可以用栈(stack) 来存储未处理的元素
vector<int> FindMaxDistance(vector<int> &num)
{
int len=num.size();
if(len==0) return {}; //空数组,返回空
vector<int> res(len,-1); //返回结果:初始化-1,表示未找到
stack<int> notFind; //栈:num中还未找到符合条件的元素索引
int i=0;
//遍历数组
while(i<len){
//如果栈空或者当前num元素不大于栈顶,将当前元素压栈,索引后移
if(notFind.empty() || num[notFind.top()]>=num[i]){
notFind.push(i++);
}else{ //有待处理元素,且num当前元素大于栈顶索引元素,符合条件,更新结果数组中该索引的值,栈顶出栈。
res[notFind.top()]= i - notFind.top();
notFind.pop();
}
}
for(int i = 0; i < res.size(); i++){
cout << res[i] << " ";
}
return res;
}