《算法零基础100讲》(第59讲) 前缀和(三) 线性前缀和统计

最新推荐文章于 2024-10-02 23:40:03 发布

yaCoding

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量290

点赞数

分类专栏：《算法零基础100讲》文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/qq_52230126/article/details/122291870

版权

《算法零基础100讲》专栏收录该内容

11 篇文章 2 订阅

订阅专栏

这篇博客探讨了三个关于数组操作的算法问题：航班预订统计、有序数组中差绝对值之和以及可获得的最大点数。解决方案包括使用滑动窗口和前缀和技巧，有效地计算每个航班的预订座位数、数组中每个元素与其他元素差的绝对值之和，以及在拿走一定数量卡牌时的最大点数。

摘要由CSDN通过智能技术生成

1109. 航班预订统计

/*
1.这里有 n 个航班，它们分别从 1 到 n 进行编号。
2.有一份航班预订表 bookings ，表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti （包含 firsti 和 lasti ）的 每个航班 上预订了 seatsi 个座位。
3.请你返回一个长度为 n 的数组 answer，里面的元素是每个航班预定的座位总数*/
class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int [] nums = new int [n];
        for(int[] booking :bookings) {
            nums[booking[0] -1] += booking[2];
            if(booking[1] < n) {
                nums[booking[1]] -= booking[2];
            }
        }
        for(int i = 1; i < n; i++) {
            nums[i] += nums[i -1];
        }
        return nums;
    }
}
//时间复杂度：O(m+n)

1685. 有序数组中差绝对值之和

/*
1.给你一个 非递减 有序整数数组 nums 。
2.请你建立并返回一个整数数组 result，它跟 nums 长度相同，且result[i] 等于 nums[i] 与数组中所有其他元素差的绝对值之和。
3.换句话说，result[i]等于 sum(|nums[i]-nums[j]|) ，其中 0 <= j < nums.length 且 j != i（下标从 0 开始 ）*/
class Solution {
    public int[] getSumAbsoluteDifferences(int[] nums) {
        int n = nums.length;
        int[] result = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            result[i] = result[i - 1] + nums[i - 1];
        }
        int[] ans = new int[n];
        for (int i = 0; i < n; i++) {
            ans[i] = (i + 1) * nums[i] - result[i + 1];
            ans[i] += result[n] - result[i + 1] - (n - i - 1) * nums[i];
        }
        return ans;
    }
}

1423. 可获得的最大点数

/*
1.几张卡牌 排成一行，每张卡牌都有一个对应的点数。点数由整数数组 cardPoints 给出。
2.每次行动，你可以从行的开头或者末尾拿一张卡牌，最终你必须正好拿 k 张卡牌。
3.你的点数就是你拿到手中的所有卡牌的点数之和。
4.给你一个整数数组 cardPoints 和整数 k，请你返回可以获得的最大点数*/
class Solution {
/*思路:记数组cardPoints的长度为n，由于只能从开头和末尾拿k张卡牌，所以最后剩下的必然是连续的n−k张卡牌。我们可以通过求出剩余卡牌点数之和的最小值，来求出拿走卡牌点数之和的最大值*/
    public int maxScore(int[] cardPoints, int k) {
        int n = cardPoints.length;//滑动窗口大小为n-k
        int m = n − k;//选取前n-k个作为初始值
        int sum = 0;
        for(int i = 0; i < m; i++) {
            sum += cardPoints[i];//对剩余的卡片进行求和
        }
        int minSum = sum;
        //由于剩余卡牌是连续的，使用一个固定长度为 n−k 的滑动窗口对数组 cardPoints 进行遍历
        for (int i = m; i < n; i++) {
             // 滑动窗口每向右移动一格，增加从右侧进入窗口的元素值，并减少从左侧离开窗口的元素值
            sum += cardPoints[i] - cardPoints[i -m];
            minSum = Math.min(minSum, sum);
        }
        //求出滑动窗口最小值，然后用所有卡牌的点数之和减去该最小值，即得到了拿走卡牌点数之和的最大值
        return Arrays.stream(cardPoints).sum() - minSum;
    }
}
//参考题解.