日常刷题(10)

1. 计算礼品发放的最小分组数目

1.1. 题目描述

又到了一年的末尾，项目组让小明负责新年晚会的小礼品发放工作。

为使得参加晚会的同事所获得的小礼品价值相对平衡，需要把小礼品根据价格进行分组，但每组最多只能包括两件小礼品，并且每个分组的价格总和不能超过一个价格上限。

为了保证发放小礼品的效率，小明需要找到分组数目最少的方案。

你的任务是写一个程序，找出分组数最少的分组方案，并输出最少的分组数目。

把礼物捆组合为两个一组或者单个一组，要求组合的单价不能超过阈值，且组数最小

1.2. 输入描述

第一行数据为分组礼品价格之和的上限

第二行数据为每个小礼品的价格，按照空格隔开，每个礼品价格不超过分组价格和的上限

1.3. 输出描述

输出最小分组数量

1.4. 用例

输入 5
1 2 5
输出 2
说明 无

1.5. 题目解析

双指针

1.6. 代码

class Solution:
    def __call__(self, limit: int, gifts: List[int]):
        gifts.sort()
        l, r = 0, len(gifts) - 1
        ans = 0
        while l <= r:
            sum_price = gifts[l] + gifts[r]
            if sum_price <= limit:
                l += 1
            ans += 1
            r -= 1
        return ans

2. 计算三叉搜索树的高度

2.1. 题目描述

定义构造三叉搜索树规则如下：

每个节点都存有一个数，当插入一个新的数时，从根节点向下寻找，直到找到一个合适的空节点插入。查找的规则是：

• 如果数小于节点的数减去500，则将数插入节点的左子树
• 如果数大于节点的数加上500，则将数插入节点的右子树
• 否则，将数插入节点的中子树

给你一系列数，请按以上规则，按顺序将数插入树中，构建出一棵三叉搜索树，最后输出树的高度。

2.2. 输入描述

第一行为一个数 N，表示有 N 个数，1 ≤ N ≤ 10000

第二行为 N 个空格分隔的整数，每个数的范围为[1,10000]

2.3. 输出描述

输出树的高度（根节点的高度为1）

2.4. 用例

• 用例1

输入 5
5000 2000 5000 8000 1800
输出 3

对于1800这个值
第一层：1800 < 5000 - 500，去左边
第二层：2000 - 500 <= 1800 <= 2000 + 5，去中间

• 用例2

输入 3
5000 4000 3000
输出 3

对于3000这个值
第一层：3000 < 5000 - 500，去左边
第二层：3000 < 4000 - 500，去左边

• 用例3

输入 9
5000 2000 5000 8000 1800 7500 4500 1400 8100
输出 4

对于8100这个点
第一层：5000 + 500 < 8100，去右边
第二层：8000 + 500 < 8100，去右边
第三层：7500 + 500 < 8100，去右边

2.5. 题目解析

构造个树结构，然后数一下层数就好

2.6. 代码

class Tree:
    def __init__(self, val):
        self.val = val
        self.nodes = [None, None, None]  # 左中右

    def append(self, val):
    	# 看看新节点应该去哪个位置
    	# 由于等于两个阈值时也要放在中间，所以向外扩张了一个非常小的数
        pos = sorted([self.val - 500 - 1e-4, self.val + 500 + 1e-4, val]).index(val)
        # 处理两类情况
        if self.nodes[pos] is None:
            self.nodes[pos] = Tree(val)
        else:
            self.nodes[pos].append(val)

    def __len__(self):
        nodes = list(filter(lambda x: x is not None, self.nodes))
        if not nodes:
            return 1
        return 1 + max(map(len, nodes))


class Solution:
    def __call__(self, nodes: List[int]):
        root = Tree(nodes[0])
        for node in nodes[1:]:
            root.append(node)
        return len(root)