C++队列练习：Blah数集（Blah‘s Number set）

时空限制

CPU占用时长: 1秒 内存使用限制: 128MB

问题描述

大数学家高斯小时候偶然间发现一种有趣的自然数集合Blah，对于以a为基的集合Ba定义如下：

(1) a是集合Ba的基，且a是Ba的第一个元素；

(2) 如果x在集合Ba中，则2x + 1和3x + 1也都在集合Ba中；

(3) 没有其他元素在集合Ba中了。

现在小高斯想知道如果将集合Ba中元素按照升序排列，第N个元素会是多少？注意：集合中没有重复的元素。

输入格式

输入包括很多行，每行输入包括两个数字，集合的基a(1<=a<=50))以及所求元素序号n(1<=n<=1000000)

输出格式

对于每个输入，输出集合Ba的第n个元素值。

输入输出样例

样例1

输入样例

1 100

输出样例

418

样例2

输入样例

28 5437

输出样例

900585

参考代码一(队列)

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
int main() {
	ll a, n;
	while (cin >> a >> n) {
		queue<ll> q1, q2;
		q1.push(2 * a + 1);
		q2.push(3 * a + 1);
		int ct = 1;
		ll ans = a;
		while (ct < n) {
			if (q1.front() < q2.front()) {
				ans = q1.front(); q1.pop();
			}
			else if(q1.front() > q2.front()){
				ans = q2.front(); q2.pop();	
			}
			else {
				ans = q2.front();
				q2.pop(); q1.pop();
			}
			q1.push(2 * ans + 1);
			q2.push(3 * ans + 1);
			ct++;
		}
		cout << ans << endl;
	}
	return 0;
}