【做练习】Sequence(二叉树,堆)堆的原理和实现

1. 题目

总时间限制: 3000ms 内存限制: 65536kB

描述

给定m个数字序列,每个序列包含n个非负整数。我们从每一个序列中选取一个数字组成一个新的序列,显然一共可以构造出 n m n^m nm个新序列。接下来我们对每一个新的序列中的数字进行求和,一共会得到 n m n^m nm个和,请找出最小的n个和

输入

输入的第一行是一个整数T,表示测试用例的数量,接下来是T个测试用例的输入
每个测试用例输入的第一行是两个正整数m(0 < m <= 100)和n(0 < n <= 2000),然后有m行,每行有n个数,数字之间用空格分开,表示这m个序列
序列中的数字不会大于10000
输出
对每组测试用例,输出一行用空格隔开的数,表示最小的n个和

样例输入

1
2 3
1 2 3
2 2 3

样例输出

3 3 4


2. 分析

这道题考察二叉树(堆)。

2.1. 堆

是一种基于完全二叉树的数据结构。堆分为最小值堆最大值堆
最小值堆为例,如果一棵二叉树满足:从根到叶的任意一条路径,总满足单调递增关系,那么它就是一个最小值堆——换句话说,最小值堆中,任意节点所存的值,总是大于它的父节点,这样,最小值堆的根节点一定是所有节点中的最小值点。

堆提供三种基本操作。以最小值堆为例:

  • 查询最小值:即查询根节点的值
  • 插入任意元素:先把新的节点插入到完全二叉树的末尾。如果这个节点比父节点更小,则将它和父节点交换。重复交换直到它成为根节点,或不小于它的父节点。
  • 删除最小值:把最后一个节点和根节点交换。然后查看新的根节点,如果存在子节点比它小,则将它与最小的子节点交换。反复交换,直到它不存在子节点比它小。

在STL中,我们可以使用priority_queue(优先队列),它包装了一个堆。但为了复习堆的原理和实现细节,我在代码实现中,实现了一个堆的模板类。

2.2 解题思路

首先,我们记输入数组为 A 1 , A 2 , . . . , A m A_1 ,A_2,...,A_m A1,A2,...,Am
若记 S i S_i Si为前 i i i个数组各选一个数求和的最小的 n n n个值。我们会发现:可以仅仅用 S i − 1 S_{i-1} Si1 A i A_i Ai推出 S i S_i Si。我们只需计算所有 A i [ a ] + S i − 1 [ b ] , 0 ≤ a , b < n , A_i[a]+S_{i-1}[b],0\leq a, b < n, Ai[a]+Si

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值