代码源 493. 合并数列

493. 合并数列

有n个线性序列，第i个序列都可以表示成ki∗x+bi的形式（x=0,1,2,…）。

现在询问将这些序列的数从小到大合并起来，前m个数分别是多少（重复出现的数合并后也会多次出现）。

输入格式

输入第一行一个数n。
接下来n行每行两个数表示k_i，b_i。
最后一行一个数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;
}