Blah数集

题目描述

数学家高斯小时候偶然间发现一种有趣的自然数集合Blah。对于以a为基的集合Blah定义如下：
　　1)a是集合Blah的基，且a是Blah的第一个元素；
　　2)如果x在集合Blah中，则2x+1和3x+1也都在集合Blah中；
　　3)没有其他元素在集合Blah中了。
　　现在小高斯想知道如果将集合Blah中元素按照升序排列，第n个元素会是多少？注意：集合中没有重复的元素。

输入输出格式

输入格式：
一行两个正整数，分别表示集合的基a以及所求元素序号n,1≤a≤50,1≤n≤1000000。

输出格式：
一行一个正整数，表示集合Blah的第n个元素值。

输入输出样例

输入样例：

1 100

输出样例：

418

---

分析

题目说如果集合中有x，则一定有2x+1和3x+1，并且最后要输出从小到大的第n个。
由于x>1，所以一定x<2x+1<3x+1，因此我们可以定义两个队列，分别存放2x+1和3x+1，每一次都取出其中小的那个，执行到第n次时就能得出第n小的数

Code

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
long long a,n;
queue<int> q1;  //存放2x+1
queue<int> q2;  //存放3x+1
int main()
{
	scanf("%lld%lld",&a,&n);
	n--;
	while(n--)
	{
		q1.push(2*a+1);
		q2.push(3*a+1);
		if(q1.front()<q2.front())
		{
			a=q1.front();
			q1.pop();
		}
		else if(q1.front()>q2.front())
		{
			a=q2.front();
			q2.pop();
		}
		else
		{
			a=q1.front();
			q1.pop();
			q2.pop();
		}
	}
	printf("%lld",a);
	return 0;
}

若有更简便的方法，请在评论区提出，谢谢