描述
大数学家高斯小时候偶然间发现一种有趣的自然数集合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 100 28 5437
样例输出
418 900585
思维跟不上了这一题,还是看其他人的解析会的,暑假一定多刷题了。
这题一开始我是用一个vector和一个queue,queue弹出一个x就装进vector里,并把(2 * x + 1)和(3 * x +1)放进queue里,然后直到vector里的元素达到n个时跳出来,再排序输出最后一个就是第n个,样例直接寄。
然后又想了一下,发现是要全部的,意思是我队列里会有比这个最大元素还要小的元素存在,所以我又来了个最大mx = 3 * n + 1, 直到队列首个元素大于mx时,即终止。1 100过了,但是后面的没过,说明还是会有比他小的没有放到vector里的。
然后最后看别人的code,终于AC。
用两个队列存储,一个队列存2 * n + 1; 一个队列存3 * n + 1。然后判断哪个队列的首元素比较小就先弹出,相等时两个都弹出(因为Blah数集是自然数集合,就没有重复的元素),这样子他们两个队列里面的元素一定就是从小到大排序(因为n是从小到大增大的),同时对弹出的计数,第n个极为答案。
#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;
}