C++ STL set的使用,一道算法题,化繁为简、化难为易

先看问题,题目来自C语言贴吧。题目里没有什么H,那是2X+1和3X+1,尬笑。。。
C++ STL set的使用,一道算法题,化繁为简、化难为易_第1张图片

看似不难,就是排序、去重而已,元素个数如果不是50,而是10万、100万,要用纯C语言写出高效的算法,还是有一定难度的。

说到去重、排序,set说:”这活我最擅长“!set天生就是干这个的,能自动去重、自动排序。下面的代码没有什么复杂逻辑,看不到去重,也看不到排序,都是set的功劳。

在for auto循环中,加入到set中的新元素,等会也会自动被访问(被处理)。最初,当1在集合中时,2*1+1=3被加入到集合,这个3等会也会继续被访问(被处理),也就是说3*2+1=7会被加入到集合,这就是一个无限循环,这个无限循环恰恰就是我们想要的。然后,循环中一个break,完事!

自动去重、自动排序、无限迭代,这题目简直就是为C++的set量身定做的!是谁出的题?放在C语言贴吧可惜了,尬笑。。。

C++ STL set的使用,一道算法题,化繁为简、化难为易_第2张图片

数据量加到10万,set说:小菜一碟,我的效率不错吧。第100万个数据也很轻松,第1000万个数据开始有点压力了,一点点而已。

C++ STL set的使用,一道算法题,化繁为简、化难为易_第3张图片

代码如下:

#include 
#include 
using namespace std;

int main()
{
	clock_t t1 = clock();

	set nums = { 1 };
	const int data_num = 100000;
	long long sum = 0;
	int n = 0;

	for (auto num : nums)
	{
		//cout << num << " ";
		sum += num;
		n++;
		if (n == data_num)
		{
			cout << "第" << data_num << "个元素:" << num << endl;
			break;
		}

		nums.insert(nums.begin(), 2 * num + 1);
		nums.insert(nums.begin(), 3 * num + 1);
	}

	cout << "前" << data_num << "个元素的和:" << sum << endl;

	clock_t t2 = clock();
	cout << t2 - t1 << "毫秒" << endl;

	return 0;
}

还是那句话,STL让程序变得简洁、高效、安全!

你可能感兴趣的:(c++,开发语言)