算法 | 每日一题 | 两个线段获得的最多奖品 | 滑动窗口

星不易

已于 2024-09-11 11:10:32 修改

阅读量327

点赞数 9

分类专栏：不易|力扣算法每日一题文章标签： java 算法滑动窗口

于 2024-09-11 11:09:52 首次发布

本文链接：https://blog.csdn.net/m0_57130776/article/details/142133807

版权

不易|力扣算法每日一题专栏收录该内容

33 篇文章 2 订阅

订阅专栏

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

原题地址：力扣每日一题：两个线段获得的最多奖品

在 X轴上有一些奖品。给你一个整数数组 prizePositions ，它按照非递减顺序排列，其中 prizePositions[i] 是第 i 件奖品的位置。数轴上一个位置可能会有多件奖品。再给你一个整数 k 。
你可以同时选择两个端点为整数的线段。每个线段的长度都必须是 k 。你可以获得位置在任一线段上的所有奖品（包括线段的两个端点）。注意，两个线段可能会有相交。
比方说 k = 2 ，你可以选择线段 [1, 3] 和 [2, 4] ，你可以获得满足 1 <= prizePositions[i] <= 3 或者 2 <= prizePositions[i] <= 4 的所有奖品 i 。
请你返回在选择两个最优线段的前提下，可以获得的最多奖品数目。

以 mid 为分界，若有重叠或能全部覆盖就直接拿到所有奖品。否则就先跑右边的那条线段，以mid为左端点，以right - 1为右端点，将长度调整到为k时就可以找奖品最多的那一段了。左边的线段以mid为右端点，移动左端点left，长度为k时开始找奖品最多的一段。把ans放在temp前面是因为避免mid重叠，ans要使用上一次遍历得到的temp。

class Solution {
    public int maximizeWin(int[] prizePositions, int k) {
        int n = prizePositions.length;
        //拿到全部奖品
        if (k * 2 + 1 >= prizePositions[n - 1] - prizePositions[0]) {
            return n;
        }

        int ans = 0, temp = 0;
        int left = 0, right = 0;
        for (int mid = 0; mid < n; mid++) {
            //第二条线段右端为right - 1，左端为mid
            while (right < n && prizePositions[right] - prizePositions[mid] <= k) {
                right++;
            }
            //此时temp中右端点为当前mid - 1，因此不会重叠
            ans = Math.max(ans, temp + right - mid);
            //第一条线段左端为left，右端为mid
            while (prizePositions[mid] - prizePositions[left] > k) {
                left++;
            }
            temp = Math.max(temp, mid - left + 1);
        }
        return ans;
    }
}