题目
给定一个整数数组 nums,返回区间和在 [lower, upper] 之间的个数,包含 lower 和 upper。
区间和 S(i, j) 表示在 nums 中,位置从 i 到 j 的元素之和,包含 i 和 j (i ≤ j)。
说明:
最直观的算法复杂度是 O(n2) ,请在此基础上优化你的算法。
示例:
输入: nums = [-2,5,-1], lower = -2, upper = 2,
输出: 3
解释: 3个区间分别是: [0,0], [2,2], [0,2],它们表示的和分别为: -2, -1, 2。
归并排序
思路与算法
设前缀和数组为sum,若按一般的查找的思想,由于sum数组无序,所以我们需要用双重for循环进行遍历,时间复杂度为N^2,时间复杂度过高。此时,我们就可以考虑归并排序的思想。
对任意一个区间,我们都可以用
sum[i]-sum[j]//sum[i]表示num[0--size-1]的和
来表示
结合归并排序思想,将sum数组分成两部分进行处理,前部分N1寻找符合条件区间,和后部分N2寻找符合条件区间,最后处理N1和N2合并的区间。
假设两部分数组的区间已经寻找完毕,继续寻找N1,N2合并部分的区间。如果两部分是有序的,那么就可以只用logN的速度进行查找了。
int countRangeSumRecursive(long long* sum, int lower, int upper, int left, int right) {
if (left == right) {
return 0;
} else {
int mid = (left + right) / 2;
int n1 = countRangeSumRecursive(sum, lower, upper, left, mid);
int n2 = countRangeSumRecursive(sum, lower, upper, mid + 1, right);
int ret = n1 + n2;
// 首先统计下标对的数量
int i = left;
int l = mid + 1;
int r = mid + 1;
while (i <= mid) {
while (l <= right && sum[l] - sum[i] < lower) l++;
while (r <= right && sum[r] - sum[i] <= upper) r++;
ret += (r - l);
i++;
}
// 随后合并两个排序数组
int sorted[right - left + 1];
memset(sorted, 0, sizeof(sorted));
int p1 = left, p2 = mid + 1;
int p = 0;
while (p1 <= mid || p2 <= right) {
if (p1 > mid) {
sorted[p++] = sum[p2++];
} else if (p2 > right) {
sorted[p++] = sum[p1++];
} else {
if (sum[p1] < sum[p2]) {
sorted[p++] = sum[p1++];
} else {
sorted[p++] = sum[p2++];
}
}
}
for (int i = 0; i < right - left + 1; i++) {
sum[left + i] = sorted[i];
}
return ret;
}
}
int countRangeSum(int* nums, int numsSize, int lower, int upper) {
long long s = 0;
long long sum[numsSize + 1];
sum[0] = 0;
for (int i = 1; i <= numsSize; i++) {
sum[i] = sum[i - 1] + nums[i - 1];
}
return countRangeSumRecursive(sum, lower, upper, 0, numsSize);
}