C++ 算法:区间和的个数

涉及知识点

归并排序

题目

给你一个整数数组 nums 以及两个整数 lower 和 upper 。求数组中,值位于范围 [lower, upper] (包含 lower 和 upper)之内的 区间和的个数 。

区间和 S(i, j) 表示在 nums 中,位置从 i 到 j 的元素之和,包含 i 和 j (i ≤ j)。

示例 1:

输入:nums = [-2,5,-1], lower = -2, upper = 2

输出:3

解释:存在三个区间:[0,0]、[2,2] 和 [0,2] ,对应的区间和分别是:-2 、-1 、2 。

示例 2:

输入:nums = [0], lower = 0, upper = 0

输出:1
参数范围

1 <= nums.length <= 105

-231 <= nums[i] <= 231 - 1

-105 <= lower <= upper <= 105

题目数据保证答案是一个 32 位 的整数

2023年3月版(树状数组)

template

class CTreeArr

{

public:

CTreeArr(int iSize) :m_vData(iSize+1)

{

}

void Add(int index, T value)

{

index++;

while (index < m_vData.size())

{

m_vData[index] += value;

index += index&(-index);

}

}

T Sum(int index)

{

index++;

T ret = 0;

while (index )

{

ret += m_vData[index];

index -= index&(-index);

}

return ret;

}

private:

vector m_vData;

};

class Solution {

public:

int countRangeSum(vector& nums, int lower, int upper) {

//std::multiset dp;

vector vCan;

long long llAdd = 0;

for (const auto& n : nums)

{

llAdd += n;

vCan.emplace_back(n - llAdd);

}

std::sort(vCan.begin(), vCan.end());

auto itEnd = std::unique(vCan.begin(), vCan.end());

vCan.erase(itEnd, vCan.end());

CTreeArr tree(vCan.size());

llAdd = 0;

long long iRet = 0;

for (const auto& n : nums)

{

llAdd += n;

const int iIndex = std::lower_bound(vCan.begin(), vCan.end(), (long long)n - llAdd) - vCan.begin();

tree.Add(iIndex, 1);

const int iBeginIndex = std::lower_bound(vCan.begin(), vCan.end(), (long long)lower - llAdd) - vCan.begin();

const int iEndIndex = std::lower_bound(vCan.begin(), vCan.end(), (long long)upper - llAdd+1) - vCan.begin();

if (0 == iEndIndex)

{

continue;

}

iRet += tree.Sum(iEndIndex - 1);

if (iBeginIndex > 0)

{

iRet -= tree.Sum(iBeginIndex - 1);

}

}

复制代码
	 return iRet;
 }

};

2023年8月 归并排序版

class CCountRangeSum : public CMergeSortIndex

{

public:

CCountRangeSum(const vector& vPreSum,int lower, int upper):CMergeSortIndex(vPreSum)

{

m_iLower = lower;

m_iUpper = upper;

}

using CMergeSortIndex::CMergeSortIndex;

int m_iRet = 0;

protected:

virtual void OnSortLeftRightEnd(int left, int mid, int right)

{

int i2 = mid,i3=mid;

for (int i1 = left; i1 < mid; i1++)

{

while ((i2 < right) && (m_nums[m_vIndexs[i2]] - m_nums[m_vIndexs[i1]] < m_iLower))

{

i2++;

}

while ((i3 < right) && (m_nums[m_vIndexs[i3]] - m_nums[m_vIndexs[i1]] <= m_iUpper))

{

i3++;

}

m_iRet += i3 - i2;

}

}

int m_iLower ;

int m_iUpper ;

};

class Solution {

public:

int countRangeSum(vector& nums, int lower, int upper) {

vector vPreSum(1);

for (const auto& n : nums)

{

vPreSum.emplace_back(vPreSum.back() + n);

}

CCountRangeSum crs(vPreSum, lower, upper);

crs.Sort();

return crs.m_iRet;

}

};

扩展阅读

视频课程

有效学习:明确的目标 及时的反馈 拉伸区(难度合适),可以先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快

速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

相关下载

想高屋建瓴的学习算法,请下载《闻缺陷则喜算法册》doc版
https://download.csdn.net/download/he_zhidan/88348653

鄙人想对大家说的话
闻缺陷则喜是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。
墨家名称的来源:有所得以墨记之。
如果程序是一条龙,那算法就是他的是睛

测试环境

操作系统:win7 开发环境: VS2019 C++17

或者 操作系统:win10 开发环境:

VS2022 C++17

相关推荐
exm-zem12 分钟前
多用户图书管理系统
c++
☆璇14 分钟前
【数据结构】排序
c语言·开发语言·数据结构·算法·排序算法
我要成为c嘎嘎大王14 分钟前
【C++】初识C++(1)
开发语言·c++
艾莉丝努力练剑3 小时前
【LeetCode&数据结构】单链表的应用——反转链表问题、链表的中间节点问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
_殊途5 小时前
《Java HashMap底层原理全解析(源码+性能+面试)》
java·数据结构·算法
还债大湿兄5 小时前
《C++内存泄漏8大战场:Qt/MFC实战详解 + 面试高频陷阱破解》
c++·qt·mfc
珊瑚里的鱼8 小时前
LeetCode 692题解 | 前K个高频单词
开发语言·c++·算法·leetcode·职场和发展·学习方法
AI+程序员在路上8 小时前
QTextCodec的功能及其在Qt5及Qt6中的演变
开发语言·c++·qt
Risehuxyc8 小时前
C++卸载了会影响电脑正常使用吗?解析C++运行库的作用与卸载后果
开发语言·c++
秋说9 小时前
【PTA数据结构 | C语言版】顺序队列的3个操作
c语言·数据结构·算法