partial_sum 的使用

partial_sum 的使用

partial_sum 是C++ STL 算法组件中的其中一个算法，其作用是计算某个序列局部元素的和。它有四个重载函数。要使用 partial_sum 需要引用头文件 numeric。

下面是一个例子：
头文件、命名空间的引用，main方法，定义容器并赋值，输出容器，使用partial_sum：

代码：

#include<iostream>
#include <vector>//vector
#include <numeric>
#include <time.h> //随机数
using namespace std;
//自定义方法
int func(int x, int y) { return x - y; }
void main()
{
	//定义容器
	vector<int>vec;
	srand(time(0));//随机数
	//赋值
	for (int i = 0; i < 5; i++)
	{
		int num = rand() % 100;
		vec.push_back(num);
	}
	//迭代器
	vector<int>::iterator vi;
	//输出
	cout << "vec : \n";
	for (vi = vec.begin(); vi < vec.end(); vi++) { cout << *vi << '\t'; } //输出
	cout << endl << endl;

	// 结果存放
	// 声明容器、数组，用于存放计算结果
	int arr[10] = { 0 };
	vector<int>vec2;
	vec2.resize(10);//设置容器的大小


	// 局部总和 第一个重载
	//partial_sum(容器要计算的起始位置，容器要计算的结束位置，结果存放的起始位置)
	partial_sum(vec.begin(), vec.end(), arr);
	cout << "1 arr : \n"; 
	for (int i = 0; i < 10; i++) { cout << arr[i] << '\t'; }//输出
	cout << endl << endl;
	
	//第二个重载
	// partial_sum(容器要计算的起始位置，容器要计算的结束位置，结果存放的起始位置，自定义函数)
	partial_sum(vec.begin(), vec.end(), arr, func);
	cout << "2 arr : \n";
	for (int i = 0; i < 10; i++) { cout << arr[i] << '\t'; }//输出
	cout << endl << endl;

	//第三个重载
	//partial_sum(容器要计算的起始位置，容器要计算的结束位置，结果存放的起始位置)
	partial_sum(vec.begin(), vec.end(), vec2.begin());
	cout << "3 vec2 : \n";
	for (int i = 0; i < 10; i++) { cout << vec2[i] << '\t'; }//输出
	cout << endl << endl;

	//第四个重载
	// partial_sum(容器要计算的起始位置，容器要计算的结束位置，结果存放的起始位置，自定义函数)
	partial_sum(vec.begin(), vec.end(), vec2.begin(), func);
	cout << "4 vec2 : \n";
	for (int i = 0; i < 10; i++) { cout << vec2[i] << '\t'; }//输出
	cout << endl<< endl;

	// 结果存放在容器自身
	partial_sum(vec.begin(), vec.end(), vec.begin());
	cout << "5 vec : \n";
	for (vi = vec.begin(); vi < vec.end(); vi++) { cout << *vi << '\t'; }//输出
}

运行结果：

从运行结果来看：
在没有自定义方法的情况下，partial_sum 算法默认计算局部和，每一次计算 第n项 与 前n项和 的和。
如 第一个重载(运行结果是 1 arr），第三个重载(运行结果是3 vec2);

而自定义方法返回的是 前n项的差减去第n项的结果。
如 第二个重载(运行结果是2 arr)，第四个重载(运行结果是4 vec2);

存放结果的可以是数组，如 第一个重载(运行结果是 1 arr），第二个重载(运行结果是2 arr)；
也可以是容器，如第三个重载(运行结果是3 vec2)，第四个重载(运行结果是4 vec2)；
还可以是容器自身，如运行结果5 vec；

感谢您的浏览。