运行时控制线程的数量

题目:用多线程计算 1+2+3+…+…+100 = 5050
希望有4个线程,每个线程处理25个数字。
如下:

#include <iostream>
#include <thread>
#include <windows.h>
#include <vector>
#include <numeric>
using namespace std;

int parallel_accumulate(vector<int>::iterator first, vector<int>::iterator last, int init);
void accumulate_block(vector<int>::iterator first, vector<int>::iterator last, int& sum);
int main(void)
{
	vector<int> ivec(100);
	for (int i = 1; i <= 100; ++i)
		ivec.push_back(i);
	cout << parallel_accumulate(ivec.begin(),ivec.end(),0) << endl;
	return 0;
}
int parallel_accumulate(vector<int>::iterator first, vector<int>::iterator last, int init)
{
	const unsigned int length = distance(first, last);
	if (!length)
		return init;
	const unsigned int min_per_thread = 25;
	const unsigned int max_threads = (length + min_per_thread - 1) / min_per_thread;
	const unsigned int hard_thread_num = thread::hardware_concurrency();
	const unsigned int num_thread = min(hard_thread_num != 0 ? hard_thread_num : 2, max_threads);
	const unsigned int block_size = length / num_thread;
	vector<thread> threads(num_thread - 1);
	vector<int> results(num_thread);
	vector<int>::iterator block_start = first;
	for (int i = 0; i < (num_thread - 1); i++)
	{
		vector<int>::iterator block_end = block_start;
		advance(block_end, block_size);
		threads[i] = thread(accumulate_block, block_start, block_end, ref(results[i]));
		block_start = block_end;
	}
	accumulate_block(block_start, last, results[num_thread - 1]);
	for (auto &r : threads)//auto遍历线程
		r.join();
	return accumulate(results.begin(), results.end(), init);
}

void accumulate_block(vector<int>::iterator first, vector<int>::iterator last, int& sum)
{
	sum = accumulate(first, last, sum);
}

代码中使用了stl里面的容器来定义多个thread,thread是一个类,可以这样使用。主要难度再迭代器的应用上。需要反复阅读加深记忆。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值