力扣OJ 496. 下一個更大元素 I

給定兩個 沒有重複元素 的數組 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下一個比其大的值。

nums1 中數字 x 的下一個更大元素是指 x 在 nums2 中對應位置的右邊的第一個比 x 大的元素。如果不存在，對應位置輸出 -1 。

 

示例 1:

輸入: nums1 = [4,1,2], nums2 = [1,3,4,2].
輸出: [-1,3,-1]
解釋:
    對於num1中的數字4，你無法在第二個數組中找到下一個更大的數字，因此輸出 -1。
    對於num1中的數字1，第二個數組中數字1右邊的下一個較大數字是 3。
    對於num1中的數字2，第二個數組中沒有下一個更大的數字，因此輸出 -1。
示例 2:

輸入: nums1 = [2,4], nums2 = [1,2,3,4].
輸出: [3,-1]
解釋:
    對於 num1 中的數字 2 ，第二個數組中的下一個較大數字是 3 。
    對於 num1 中的數字 4 ，第二個數組中沒有下一個更大的數字，因此輸出 -1 。
 

提示：

nums1和nums2中所有元素是唯一的。
nums1和nums2 的數組大小都不超過1000。

 

 

這個題目很簡單，我主要用來鍛造我的模板：

//翻轉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);
}

//把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> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int>ans=fgetNumFromId(nums2,fmaxrig(nums2));
        map<int,int>m=vectorToMap(nums2,ans);
        ans=vmToVector(nums1,m);
        return ans;
    }
};