五月集训（第八日）前缀和

一、1480. 一维数组的动态和

1.原题链接

1480. 一维数组的动态和

2.题目描述

        给你一个数组 nums 。数组「动态和」的计算公式为：runningSum[i] = sum(nums[0]…nums[i]) 。请返回 nums 的动态和。

3.解题思路

        使用前缀和解决即可。

4.源码

class Solution {
public:
    vector<int> runningSum(vector<int>& nums) {
        int n = nums.size();
        for (int i = 1; i < n; i++) {
            nums[i] += nums[i - 1];
        }
        return nums;
    }
};

二、1588. 所有奇数长度子数组的和

1.原题链接

1588. 所有奇数长度子数组的和

2.题目描述

        给你一个正整数数组 arr ，请你计算所有可能的奇数长度子数组的和。子数组 定义为原数组中的一个连续子序列。请你返回 arr 中 所有奇数长度子数组的和 。

3.解题思路

        首先计算前缀和，利用差和来解决。

4.源码

class Solution {
public:
    int sumOddLengthSubarrays(vector<int>& arr) {
        int n=arr.size();
        vector<int> temp(n+1);
        int ans =0;
        for(int i=0;i<n;++i){
            temp[i+1]=temp[i]+arr[i];
        }
        for(int i =0;i<n;++i){
            for(int j=i;j>=0;j-=2){
                ans+=temp[i+1]-temp[j];
            }
        }
        return ans;
    }
};

三、1442. 形成两个异或相等数组的三元组数目

1.原题链接

1442. 形成两个异或相等数组的三元组数目

2.题目描述

        给你一个整数数组 arr 。现需要从数组中取三个下标 i、j 和 k ，其中 (0 <= i < j <= k < arr.length) 。a 和 b 定义如下：
         a = arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1]
        b = arr[j] ^ arr[j + 1] ^ … ^ arr[k]
注意：^ 表示 按位异或 操作。
请返回能够令 a == b 成立的三元组 (i, j , k) 的数目。

3.解题思路

1）思路1

        运用前缀和来解题，首先计算每个元素的前缀和的异或并放入hash数组，最后在hash数组中查找异或并相加

2）思路2

        要求a==b，就可以转化成求数组的子串的异或为零，枚举数组子串异或为零的个数即可。

4.源码

1）

class Solution {
    unordered_map<int,int>hash;
public:
    int countTriplets(vector<int>& arr) {
        int pre,ans=0;
        for(int i =1;i<arr.size();++i){
            arr[i]=arr[i-1]^arr[i];
        }
        for(int j=1;j<arr.size();++j){
            hash.clear();
            for(int i=0;i<j;++i){
                if(i==0){
                    pre=0;
                }
                else{
                    pre=arr[i-1];
                }
                ++hash[arr[j-1]^pre];
            }
            for(int k=j;k<arr.size();++k){
                ans+=hash[arr[k]^arr[j-1]];
            }
        }
        return ans;
    }
};

2）

class Solution {
public:
    int countTriplets(vector<int>& arr) {
        int ans =0;
        for(int i =0;i<arr.size();++i){
            int temp =arr[i];
            for(int k =i+1;k<arr.size();++k){
                temp^=arr[k];
                if(temp==0){
                    ans+=k-i;
                }
            }
        }
        return ans;
    }
};

四、1094. 拼车

1.原题链接

1094. 拼车

2.题目描述

        车上最初有 capacity 个空座位。车 只能 向一个方向行驶（也就是说，不允许掉头或改变方向）给定整数 capacity 和一个数组 trips , trip[i] = [numPassengersi, fromi, toi] 表示第 i 次旅行有 numPassengersi 乘客，接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false。

3.解题思路

        运用前缀和来解题，将所有的人都安排在车上，然后判断一下是否每个时刻都满足总人数<=capacity即可。

4.源码

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        vector<int> temp(1002);
        for(int i=0;i<trips.size();++i){
            temp[trips[i][1]]+=trips[i][0];
            temp[trips[i][2]]-=trips[i][0];
        }
        for(int i=0;i<1000;++i){
            if(i){
                temp[i]+=temp[i-1];
            }
            if(temp[i]>capacity){
                return false;
            }
        }
        return true;
    }
};

总结

        第八天，继续加油呀！