题目描述
数学家高斯小时候偶然间发现一种有趣的自然数集合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;
}
若有更简便的方法,请在评论区提出,谢谢