题目:2555. 两个线段获得的最多奖品
思路:想要获得“最多奖品数目”,那势必让两条线段不相交。假设第一条线段在第二条线段的左边。那么先枚举第二条线段的右端点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;
}
};