力扣OJ 503. 下一個更大元素 II

給定一個循環數組（最後一個元素的下一個元素是數組的第一個元素），輸出每個元素的下一個更大元素。數字 x 的下一個更大的元素是按數組遍歷順序，這個數字之後的第一個比它更大的數，這意味着你應該循環地搜索它的下一個更大的數。如果不存在，則輸出 -1。

示例 1:

輸入: [1,2,1]
輸出: [2,-1,2]
解釋: 第一個 1 的下一個更大的數是 2；
數字 2 找不到下一個更大的數； 
第二個 1 的下一個最大的數需要循環搜索，結果也是 2。

 

//翻轉vector
template<typename T>
vector<T> frev(vector<T> v)
{
    vector<T> ans;
    for(int i=v.size()-1;i>=0;i--)ans.push_back(v[i]);
    return ans;
}
//vector乘一個數
template<typename T1,typename T2>
void fcheng(vector<T1> &v,T2 n)
{
    for(int i=v.size()-1;i>=0;i--)v[i]*=n;
}
//vector加一個數
template<typename T1,typename T2>
void fjia(vector<T1> &v,T2 n)
{
    for(int i=v.size()-1;i>=0;i--)v[i]+=n;
}
//id處，覆蓋或者添加x
template<typename T>
void finsert(vector<T>&v,int id,T x)
{
    if(id<0||id>v.size())return;
    if(id==v.size())v.push_back(x);
    v[id]=x;
}

//返回vector每個數前面最近的滿足pr關係的數的ID,-1 或者 0到size-1
template<typename T,class P>inline
vector<int>firstInLeft(vector<T>v,P pr)
{
    vector<int> ans;
    if(v.size()==0)return ans;
    stack<T>st;
    st.push(0);
    ans.push_back(-1);
    for(int i=1;i<v.size();i++)
    {
        while(!st.empty() && !pr(v[st.top()],v[i]))st.pop();
        if(st.empty())ans.push_back(-1);
        else ans.push_back(st.top());
        st.push(i);
    }
    return ans;
}

//返回vector每個數前面最近的比它小的數的ID,-1 或者 0到size-1
template<typename T>
vector<int> fminlef(vector<T> v)
{
    return firstInLeft(v,[](T a,T b){return a<b;});
}
//返回vector每個數前面最近的比它小或等於的數的ID,-1 或者 0到size-1
template<typename T>
vector<int> fminlef2(vector<T> v)
{
    return firstInLeft(v,[](T a,T b){return a<=b;});
}
//返回vector每個數前面最近的比它大的數的ID,-1 或者 0到size-1
template<typename T>
vector<int> fmaxlef(vector<T> v)
{
    fcheng(v,-1);
    vector<int>ans=fminlef(v);
    return ans;
}
//返回vector每個數前面最近的比它大或等於的數的ID,-1 或者 0到size-1
template<typename T>
vector<int> fmaxlef2(vector<T> v)
{
    fcheng(v,-1);
    vector<int>ans=fminlef2(v);
    return ans;
}


//返回vector每個數後面最近的比它小的數的ID,size 或者 0到size-1
template<typename T>
vector<int> fminrig(vector<T> v)
{
    vector<int>v1=frev(v),v2=fminlef(v1);
    fcheng(v2,-1);
    fjia(v2,v.size()-1);
    return frev(v2);
}
//返回vector每個數後面最近的比它小或等於的數的ID,size 或者 0到size-1
template<typename T>
vector<int> fminrig2(vector<T> v)
{
    vector<int>v1=frev(v),v2=fminlef2(v1);
    fcheng(v2,-1);
    fjia(v2,v.size()-1);
    return frev(v2);
}
//返回vector每個數後面最近的比它大的數的ID,size 或者 0到size-1
template<typename T>
vector<int> fmaxrig(vector<T> v)
{
    vector<int>v1=frev(v),v2=fmaxlef(v1);
    fcheng(v2,-1);
    fjia(v2,v.size()-1);
    return frev(v2);
}
//返回vector每個數後面最近的比它大或等於的數的ID,size 或者 0到size-1
template<typename T>
vector<int> fmaxrig2(vector<T> v)
{
    vector<int>v1=frev(v),v2=fmaxlef2(v1);
    fcheng(v2,-1);
    fjia(v2,v.size()-1);
    return frev(v2);
}

//把v和id轉化爲v[id]
template<typename T>
vector<int> fgetNumFromId(vector<T> &v,vector<int>id)
{
    vector<T>ans;
    ans.resize(id.size());
    for(int i=0;i<id.size();i++)ans[i]= (id[i]>=0 && id[i]<v.size()) ? v[id[i]] : -1;
    return ans;
}
//把v1和v2 轉化成1個map<v1,v2>
template<typename T1,typename T2>
map<T1,T2> vectorToMap(vector<T1> &v1,vector<T2> &v2)
{
    map<T1,T2>m;
    for(int i=0;i<v1.size();i++)m[v1[i]]=v2[i];
    return m;
}
//1個vector加1個map轉化成map[vector]
template<typename T1,typename T2>
vector<T2> vmToVector(vector<T1> &v,map<T1,T2>&m)
{
    vector<T2>ans;
    ans.resize(v.size());
    for(int i=0;i<v.size();i++)ans[i]=m[v[i]];
    return ans;
}

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int>t=nums;
        t.resize(nums.size()*2);
        copy(nums.begin(),nums.end(),t.begin()+nums.size());
        vector<int>tmp=fgetNumFromId(t,fmaxrig(t));
        vector<int>ans=nums;
        copy(tmp.begin(),tmp.begin()+tmp.size()/2,ans.begin());
        return ans;
    }
};