493. 合并数列
有n个线性序列,第i个序列都可以表示成ki∗x+bi的形式(x=0,1,2,…)。
现在询问将这些序列的数从小到大合并起来,前m个数分别是多少(重复出现的数合并后也会多次出现)。
输入格式
输入第一行一个数n。
接下来n行每行两个数表示ki,bi。
最后一行一个数m,代表答案需要前m个数。
输出格式
输出m个数,表示答案。
样例输入
2
1 2
5 2
8
样例输出
2 2 3 4 5 6 7 7
样例解释
两个数列分别为2,3,4,5,6,7,8,… 和2,7,12,…。
数据规模
对于100%的数据,保证1≤n≤20000,1≤m≤10000,1≤ki,bi≤1000。
思路
使用结构体数组,val表示当前的值,tt表示每次要加上的数
代码
/*
* 合并数列,多个数列来合成
* 用最小堆来更新最小值
*/
#include<stdio.h>
struct {
int val;//当前值
int tt;//差值
}heap[20001], t;
int len = 0;
void UP(int p) {
while (p > 1 && heap[p].val < heap[p/2].val) {
t = heap[p];
heap[p] = heap[p/2];
heap[p/2] = t;
p /= 2;
}
}
void Down(int p) {
while (p + p <= len) {//判断左子树是否存在
int j = p + p;
if (j + 1 <= len && heap[j + 1].val < heap[j].val)
j++;//右子树存在并且右子树小于左子树,让父亲和更小的交换
if (heap[p].val <= heap[j].val) //父亲最小,不用交换
break;
t = heap[p];
heap[p] = heap[j];
heap[j] = t;
p = j;//更新位置
}
}
int main() {
int n, m, k, b;
scanf("%d", &n) ;
for (int i = 1; i <= n; i++) {
scanf("%d%d", &k, &b);
heap[++len].val = b;
heap[len].tt = k;
UP(len);//因为插入到了末尾,所以向上比较
}
scanf("%d", &k);
for (int i = 1; i <= k; i++) {
printf("%d ", heap[1].val);//堆顶是最小值,直接输出
heap[1].val += heap[1].tt;//加上每次要加上的差值
Down(1);//重新向下比较
}
return 0;
}