LeetCode 962. 最大宽度坡(C++、python)

给定一个整数数组 A是元组 (i, j),其中  i < j 且 A[i] <= A[j]。这样的坡的宽度为 j - i

找出 A 中的坡的最大宽度,如果不存在,返回 0 。

示例 1:

输入:[6,0,8,2,1,5]
输出:4
解释:
最大宽度的坡为 (i, j) = (1, 5): A[1] = 0 且 A[5] = 5.

示例 2:

输入:[9,8,1,0,1,9,4,0,4,1]
输出:7
解释:
最大宽度的坡为 (i, j) = (2, 9): A[2] = 1 且 A[9] = 1.

提示:

2 <= A.length <= 50000

0 <= A[i] <= 50000

C++

class Solution {
public:
    static bool compare(pair<int,int> a, pair<int,int> b)
    {
        if(a.first==b.first)
        {
            return a.second<b.second;
        }
        else
        {
            return a.first<b.first;
        }
    }
    int maxWidthRamp(vector<int>& A) 
    {
        int n=A.size();
        int res=0;
        vector<pair<int,int>> tmp;
        for(int i=0;i<n;i++)
        {
            tmp.push_back(make_pair(A[i],i));
        }
        sort(tmp.begin(),tmp.end(),compare);
        int index=tmp[0].second;
        for(int i=1;i<n;i++)
        {
            res=max(res,tmp[i].second-index);
            index=min(index,tmp[i].second);
        }
        return res;
    }
};

python

class Solution:
    def maxWidthRamp(self, A: List[int]) -> int:
        n=len(A)
        tmp=[]
        for i in range(n):
            tmp.append([A[i],i])
        tmp=sorted(tmp,key=lambda x:(x[0],x[1]))
        res=0
        index=tmp[0][1]
        for i in range(1,n):
            res=max(res,tmp[i][1]-index)
            index=min(index,tmp[i][1])
        return res
        

猜你喜欢

转载自blog.csdn.net/qq_27060423/article/details/88407213
今日推荐