C++前缀和算法的应用：摘水果原理源码测试用例

news2024/11/14 19:42:43

本文涉及的基础知识点

C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例包括课程视频

题目

在一个无限的 x 坐标轴上，有许多水果分布在其中某些位置。给你一个二维整数数组 fruits ，其中 fruits[i] = [positioni, amounti] 表示共有 amounti 个水果放置在 positioni 上。fruits 已经按 positioni 升序排列，每个 positioni 互不相同。
另给你两个整数 startPos 和 k 。最初，你位于 startPos 。从任何位置，你可以选择向左或者向右走。在 x 轴上每移动一个单位，就记作一步。你总共可以走最多 k 步。你每达到一个位置，都会摘掉全部的水果，水果也将从该位置消失（不会再生）。
返回你可以摘到水果的最大总数。
示例 1：
输入：fruits = [[2,8],[6,3],[8,6]], startPos = 5, k = 4
输出：9
解释：
最佳路线为：

向右移动到位置 6 ，摘到 3 个水果
向右移动到位置 8 ，摘到 6 个水果
移动 3 步，共摘到 3 + 6 = 9 个水果
示例 2：
输入：fruits = [[0,9],[4,1],[5,7],[6,2],[7,4],[10,9]], startPos = 5, k = 4
输出：14
解释：
可以移动最多 k = 4 步，所以无法到达位置 0 和位置 10 。
最佳路线为：
在初始位置 5 ，摘到 7 个水果
向左移动到位置 4 ，摘到 1 个水果
向右移动到位置 6 ，摘到 2 个水果
向右移动到位置 7 ，摘到 4 个水果
移动 1 + 3 = 4 步，共摘到 7 + 1 + 2 + 4 = 14 个水果
示例 3：
输入：fruits = [[0,3],[6,4],[8,5]], startPos = 3, k = 2
输出：0
解释：
最多可以移动 k = 2 步，无法到达任一有水果的地方

参数范围：
1 <= fruits.length <= 105
fruits[i].length == 2
0 <= startPos, positioni <= 2 * 105
对于任意 i > 0 ，positioni-1 < positioni 均成立（下标从 0 开始计数）
1 <= amounti <= 104
0 <= k <= 2 * 105

分析

原理

只需要左移（右移）一次。假定左移了两次，分别移动了x1,x2,假定x1<x2。则不移动x1，水果不会少。
分四种情况：
一，左移到底。
二，先左移，后右移。
三，右移到底。
四，先右移，再左移。
一是四的特殊请，三是二的特殊情况。

步骤

一，先获取前缀和。
二，枚举左移。右移为0或负数，忽视，因为劣于左移到底。k为0是，此条不符合。
三，枚举右移。

注意

坐标无限，但前缀和有限[0,iMaxPos]。


左移后的坐标	可能小于0
左移后的坐标	可能大于iMax
右移后的坐标	可能大于iMax
k为0时要左特殊处理。

变量解释


vNum	各坐标水果数量
vSum	/vSum[i]记录[0,i)草莓的总数量
iMoveLeft	左移距离
iMoveRight	右移距离
left	移动到的最左坐标
right	移动到最右坐标

代码

核心代码

class Solution {
public:
int maxTotalFruits(vector<vector>& fruits, int startPos, int k) {
const int iMaxPos = fruits.back()[0];
vector vNum(iMaxPos + 1);
for (const auto&v : fruits)
{
vNum[v[0]] = v[1];
}
vector vSum = { 0 };//vSum[i]记录[0,i)草莓的总数量
for (int i =0; i <= iMaxPos; i++)
{
vSum.emplace_back(vSum.back() + vNum[i]);
}

	int iRet = 0;
	for (int iMoveLeft = 0; iMoveLeft <= k / 2; iMoveLeft++)
	{//先左后右
		const int iMoveRight = k - iMoveLeft * 2;
		if (iMoveRight < 0)
		{
			continue;
		}
		const int left = max(0, startPos - iMoveLeft);
		if (left > iMaxPos)
		{
			continue;
		}
		const int right = min(iMaxPos, startPos + iMoveRight);
		//可收集[left,right+1)的草莓
		const int cur = vSum[right + 1] - vSum[left];
		iRet = max(iRet, cur);
	}

	for (int iMoveRight = 0; iMoveRight <= k / 2; iMoveRight++)
	{//先左后右
		const int iMoveLeft = k - iMoveRight * 2;
		if (iMoveLeft < 0)
		{
			continue;
		}
		const int left = max(0, startPos - iMoveLeft);
		if (left > iMaxPos)
		{
			continue;
		}
		const int right = min(iMaxPos, startPos + iMoveRight);
		//可收集[left,right+1)的草莓
		const int cur = vSum[right + 1] - vSum[left];
		iRet = max(iRet, cur);
	}
	return iRet;
}

};

测试用例

template
void Assert(const vector& v1, const vector& v2)
{
if (v1.size() != v2.size())
{
assert(false);
return;
}
for (int i = 0; i < v1.size(); i++)
{
assert(v1[i] == v2[i]);
}
}

template
void Assert(const T& t1, const T& t2)
{
assert(t1 == t2);
}

int main()
{
Solution slu;
vector<vector> fruits;
int startPos = 0;
int k = 0;
int res;

fruits = {{2, 8}, {6, 3}, {8, 6}};
startPos =5 ;
k =4 ;
res = slu.maxTotalFruits(fruits, startPos, k);
Assert(9, res);

fruits = {{0, 9}, {4, 1}, {5, 7}, {6, 2}, {7, 4}, {10, 9}};
startPos = 5;
k = 4;
res = slu.maxTotalFruits(fruits, startPos, k);
Assert(14, res);

fruits = { {0,10000} };
startPos = 20000;
k = 20000;
res = slu.maxTotalFruits(fruits, startPos, k);
Assert(10000, res);

fruits = { {20000,10000} };
startPos = 20000;
k = 0;
res = slu.maxTotalFruits(fruits, startPos, k);
Assert(10000, res);


//CConsole::Out(res);

}

2023年3月旧代码

class Solution {
public:
int maxTotalFruits(vector<vector>& fruits, int startPos, int k) {
m_c = fruits.size();
int iMaxLeftIndex = std::lower_bound(fruits.begin(), fruits.end(),startPos, [](const vector& v, int i)
{
return v[0] < i;
}) - fruits.begin() - 1;
std::map<int, int> mLeftMoveNum;
for (int i = iMaxLeftIndex ; (i >= 0) && (startPos - fruits[i][0] <= k); i–)
{
const int iLeftMove = startPos - fruits[i][0];
mLeftMoveNum[iLeftMove] = fruits[i][1] + (mLeftMoveNum.empty() ? 0 : mLeftMoveNum.rbegin()->second);
}
int iMinRightIndex = iMaxLeftIndex + 1;
int iRet = 0;
if ((iMinRightIndex < m_c) && (fruits[iMinRightIndex][0] == startPos))
{
iRet += fruits[iMinRightIndex][1];
iMinRightIndex++;
}
std::map<int, int> mRightMoveNum;
for (int i = iMinRightIndex; (i < m_c) && (fruits[i][0] - startPos <= k); i++)
{
const int iRightMove = fruits[i][0] - startPos;
mRightMoveNum[iRightMove] = fruits[i][1] + (mRightMoveNum.empty() ? 0 : mRightMoveNum.rbegin()->second);
}

	 int iMaxExcludeStart = 0;
	 for (int left = 0; left <= k / 2; left++)
	 {
		 const int right = k - left * 2;
		 int iCur = 0;
		 {
			 auto itLeft = mLeftMoveNum.upper_bound(left);
			 if (mLeftMoveNum.begin() != itLeft)
			 {
				 iCur += (--itLeft)->second;
			 }
		 }
		 {
			 auto itRight = mRightMoveNum.upper_bound(right);
			 if (mRightMoveNum.begin() != itRight)
			 {
				 iCur += (--itRight)->second;
			 }
		 }
		 iMaxExcludeStart = max(iMaxExcludeStart, iCur);
	 }
	 for (int right = 0; right <= k / 2; right++)
	 {
		 const int left = k - right * 2;
		 int iCur = 0;
		 {
			 auto itLeft = mLeftMoveNum.upper_bound(left);
			 if (mLeftMoveNum.begin() != itLeft)
			 {
				 iCur += (--itLeft)->second;
			 }
		 }
		 {
			 auto itRight = mRightMoveNum.upper_bound(right);
			 if (mRightMoveNum.begin() != itRight)
			 {
				 iCur += (--itRight)->second;
			 }
		 }
		 iMaxExcludeStart = max(iMaxExcludeStart, iCur);
	 }
	 iRet += iMaxExcludeStart;
	 return iRet;
 }
 int m_c;

};