特殊数据结构-前缀和

定义: 针对数组A[1,n],,前k个数组元素之和就是前缀和的概念。
下面给出样例题:
题目: 327. 区间和的个数
网址: https://leetcode-cn.com/problems/count-of-range-sum/

代码

class Solution {
public:
    int countRangeSum(vector<int>& nums, int lower, int upper) {
        int n = nums.size();
        if(n==0 ||(lower>upper)) return 0;
        vector<long> data;
        long sum=0;
        data.push_back(sum);
        for(auto& digit: nums){
            sum += digit;
            data.push_back(sum);
        }
        return countRangeSort(data, lower, upper, 0, n);
        
    }

    int countRangeSort(vector<long>& data, int lower, int upper, int left, int right){
        if(left==right)
            return 0;
        else{
            int mid = (left + right)/2;
            int n1 = countRangeSort(data, lower, upper, left, mid);
            int n2 = countRangeSort(data, lower, upper, mid+1, right);
            int ret = n1 + n2;

            //计算当前状态的所有符合的和
            int l = mid+1;
            int r = mid+1;
            int i = left;
            while(i<=mid){   //至于为什么能够用归并,其实是利用了前缀数长的序列一定是在右边,所以一定再排序前面先进行统计,否则就出问题。
                while(l<=right && data[l] - data[i] < lower) l++;
                while(r<=right && data[r] - data[i] <= upper ) r++;
                ret += (r-l);
                i++;
            }

            //排序
            vector<long> tmp(right-left+1);   //这个地方实现方法多种,可以考虑是移位实现不使用额外空间
            i=left;
            int j = mid+1;
            int k = 0;
            while(i<=mid && j<=right){
                if(data[i]<=data[j])
                    tmp[k++] = data[i++];
                else
                    tmp[k++] = data[j++];
            }
            while(i<=mid)
                tmp[k++]=data[i++];
            while(j<=right)
                tmp[k++]=data[j++];
            for(i=0;i<=right-left;i++){
                data[i+left]=tmp[i];
            }
            return ret;
        }
    }
};

这个代码是排序-区间和中归并算法的代码,其中的核心一共两个,一个就是前缀和的概念,另一个就是归并排序。

上一篇:第九届蓝桥杯【C++省赛B组】【第六题:递增三元组】——二分解法(附解题代码)


下一篇:LeetCode 验证二叉搜索树