(nice!!!)LeetCode 2555. 两个线段获得的最多奖品(贪心、二分查找、滑动窗口)

题目:2555. 两个线段获得的最多奖品

(nice!!!)LeetCode 2555. 两个线段获得的最多奖品(贪心、二分查找、滑动窗口)_第1张图片
思路:想要获得“最多奖品数目”,那势必让两条线段不相交。假设第一条线段在第二条线段的左边。那么先枚举第二条线段的右端点i,然后找到第二条线段最远的左端点x。则第一条线段的右端点一定在x的左侧,因此只需要记录区间[0,x-1]之间的“线段长度为k”所涵盖的“最多奖品数目”。这里用数组dp来维护即可,因为区间dp[x-1]其实在遍历第二条线段时,就可以求出来。细节看注释。
方法一:二分查找,查找第二条线段最远的左端点x。时间复杂度为0(nlogn)。

class Solution {
public:
    int maximizeWin(vector<int>& prizePositions, int k) {
        int n=prizePositions.size();
        //状态dp[i]表示:区间[0,i-1]之间的“线段长度为k”所涵盖的“最多奖品数目”
        vector<int> dp(n+1,0);
        //记录答案
        int ans=0;
        //遍历第二条线段的右端点
        for(int i=0;i<n;i++){
        	//通过二分法,找到第二条线段的左端点
            int x=lower_bound(prizePositions.begin(),prizePositions.end(),prizePositions[i]-k)-prizePositions.begin();
            //更新2条线段所涵盖的“最多奖品数目”
            //第二条线段的数量为:i-x+1;第一条线段的数量为:dp[x]。
            //这里是dp[x],而不是dp[x-1]的原因是避免出现0-1=-1的情况导致下标越界
            ans=max(ans,i-x+1+dp[x]);
            //更新区间[0,i]之间的“线段长度为k”所涵盖的“最多奖品数目”
            dp[i+1]=max(dp[i],i-x+1);
        }
        return ans;
    }
};

方法二:滑动窗口,使用双指针来维护第二条线段的左右端点,时间复杂度为0(n)。

class Solution {
public:
    int maximizeWin(vector<int>& prizePositions, int k) {
        int n=prizePositions.size();
        //状态dp[i]表示:区间[0,i-1]之间的“线段长度为k”所涵盖的“最多奖品数目”
        vector<int> dp(n+1);
        //记录答案
        int ans=0;
        //遍历第二条线段的右端点i,以及维护第二条线段的左端点j
        for(int i=0,j=0;i<n;i++){
        	//如果第二条线段的长度大于k时
            while(prizePositions[i]-prizePositions[j]>k) j++;
            //更新2条线段所涵盖的“最多奖品数目”
            //第二条线段的数量为:i-j+1;第一条线段的数量为:dp[j]。
            //这里是dp[j],而不是dp[j-1]的原因是避免出现0-1=-1的情况导致下标越界
            ans=max(ans,i-j+1+dp[j]);
            //更新区间[0,i]之间的“线段长度为k”所涵盖的“最多奖品数目”
            dp[i+1]=max(dp[i],i-j+1);
        }
        return ans;
    }
};

你可能感兴趣的:(LeetCode,leetcode,算法,c++,数据结构,贪心算法,二分查找,滑动窗口)