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

相关推荐
浮生如梦_1 小时前
Halcon基于laws纹理特征的SVM分类
图像处理·人工智能·算法·支持向量机·计算机视觉·分类·视觉检测
励志成为嵌入式工程师3 小时前
c语言简单编程练习9
c语言·开发语言·算法·vim
捕鲸叉3 小时前
创建线程时传递参数给线程
开发语言·c++·算法
A charmer3 小时前
【C++】vector 类深度解析:探索动态数组的奥秘
开发语言·c++·算法
Peter_chq3 小时前
【操作系统】基于环形队列的生产消费模型
linux·c语言·开发语言·c++·后端
wheeldown4 小时前
【数据结构】选择排序
数据结构·算法·排序算法
青花瓷5 小时前
C++__XCode工程中Debug版本库向Release版本库的切换
c++·xcode
观音山保我别报错5 小时前
C语言扫雷小游戏
c语言·开发语言·算法
幺零九零零6 小时前
【C++】socket套接字编程
linux·服务器·网络·c++
TangKenny6 小时前
计算网络信号
java·算法·华为