STL常用算法——遍历算法

42 阅读2分钟

1、for_each()

for_each:遍历容器,对容器中的每一个元素调用函数或函数对象

函数原型:

for_each(iterator beg,iterator end,_func);

参数说明:

  • beg: 开始迭代器
  • end: 结束迭代器
  • _func :一元函数,可以是全局函数(函数指针),仿函数,lambda等,默认按值传递
#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>

//普通函数
void print01(int val)
{
	cout << val << " ";
}
//仿函数
class print02
{
public:
	void operator()(int val)
	{
		cout << val << " ";
	}
};

void test1()
{
	vector<int> v1;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
	}

	for_each(v1.begin(), v1.end(), print01);
	cout << endl;
	for_each(v1.begin(), v1.end(), print02());
}


int main() {
	test1();
	system("pause");
	return 0;
}

在这里插入图片描述

2、transform()

ransform() 可以将函数应用到容器的元素上,并将这个函数返回的值保存到另一个容器中,它返回的迭代器指向输出容器所保存的最后一个元素的下一个位置。

形式一

函数原型:四个参数

transform(iterator beg1,iterator end1,iterator beg2,_func);

参数说明:

  • beg1:原容器开始迭代器
  • end1:原容器结束迭代器
  • beg2:目标容器开始迭代器
  • _func:函数或者函数对象,可以有返回值
#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>

class print01
{
public:
	void operator()(int val)
	{
		cout << val << " ";
	}
};
int op_increase(int i)
{
	return i + 1;
}

class Transform1
{
public:
	int operator()(int val)
	{
		return val;
	}
};
void test1()
{
	//原容器
	vector<int> v1;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
	}
	//目标容器
	vector<int> v2;
	//设置目标容器大小 目标容器需要提前开辟空间
	v2.resize(v1.size());
	//原容器元素搬运到目标容器
	transform(v1.begin(), v1.end(), v2.begin(), Transform1());
	for_each(v2.begin(), v2.end(), print01());
	cout << endl;
	//原容器元素+1再搬运到目标容器
	transform(v1.begin(), v1.end(), v2.begin(), op_increase);
	for_each(v2.begin(), v2.end(), print01());
	cout << endl;
}

int main() {
	test1();
	system("pause");
	return 0;
}

在这里插入图片描述

形式二

函数原型:五个参数

transform(iterator beg1, iterator end1, iterator beg2, iterator beg1, _func);

参数说明:

  • 前两个参数(beg1,end1)是原容器开始迭代器和原容器结束迭代器。
  • 第3个参数是目标容器的开始迭代器,这个容器必须至少包含和原容器同样多的元素。
  • 第4个参数是一个容器的输出迭代器,它所指向的是用来保存应用函数后得到的结果的容器的开始迭代器。
  • 第5个参数是一个函数对象,它定义了一个接受两个参数的函数,这个函数接受来自两个输入容器中的元素作为参数,返回一个可以保存在输出容器中的值。
#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>

int sum(int i, int j)
{
	return i + j;
}

int increase(int i)
{
	return i + 1;
}

class print01
{
public:
	void operator()(int val)
	{
		cout << val << " ";
	}
};

void test1()
{
	vector<int> v1;
	vector<int> v2;
	for (int i = 0; i < 10; i++)
	{
		v1.push_back(i);
	}
	v2.resize(v1.size());
	transform(v1.begin(), v1.end(), v2.begin(), increase);
	for_each(v2.begin(), v2.end(), print01());//1 2 3 4 5 6 7 8 9 10
	//将v1和v2容器里的各个元素相加,将得到的结果赋值给v1
	transform(v1.begin(), v1.end(), v2.begin(), v1.begin(), sum);
	cout << endl;
	for_each(v1.begin(), v1.end(), print01());//1 3 5 7 9 11 13 15 17 19
}
int main() {
	test1();
	system("pause");
	return 0;
}

在这里插入图片描述