Blah数集
大数学家高斯小时候偶然间发现一种有趣的自然数集合Blah,对于以a
为基的集合Ba
定义如下:
(1)a
是集合Ba
的基,且a
是Ba
的第一个元素;
(2)如果x
在集合Ba
中,则2x+1
和3x+1
也都在集合Ba
中;
(3)没有其他元素在集合Ba中了。
现在小高斯想知道如果将集合Ba
中元素按照升序
排列,第N
个元素会是多少?
暴力思路
i=1
q.push(a)
while(!q.empty() && i < N){
x=q.pop() // 出队
q.push(2x+1)
q.push(3x+1)
sort(q) // 入队的2x+1可能小于上次入队的3x+1(当N较大时可能会超时)
}
cout<<q.pop()
改进思路
- 数组
ba
存放集合 - 两个队列
q1
、q2
,分别放2x+1
、3x+1
——保证2个队列各自严格升序
q1.push(a)
、q2.push(a)
- 每次分别取队头
x1
、x2
,二者最小值x
放入ba
- 从
q1
或q2
删除x
(若x1==x2,则都删
) q1.push(2x+1)
、q2.push(3x+1)
- 最后输出
ba[N]
while (cin >> a >> n) {
queue<int> q1;
queue<int> q2;
q1.push(a);
q2.push(a);
for (int i = 1; i <= n; i++) {
int e1 = q1.front();
int e2 = q2.front();
int min = e1 > e2 ? e2 : e1;
if (e1 > e2) {
q2.pop();
} else if (e1 < e2) {
q1.pop();
} else {
q1.pop();
q2.pop();
}
arr[i] = min;
q1.push(2 * min + 1);
q2.push(3 * min + 1);
}
cout << arr[n] << endl;
}
标签:q1,q2,最大,矩阵,pop,3x,push,Ba
From: https://www.cnblogs.com/algorithm-hu/p/17662066.html