日常刷题(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
日常刷题(10)_第1张图片
对于1800这个值
第一层:1800 < 5000 - 500,去左边
第二层:2000 - 500 <= 1800 <= 2000 + 5,去中间

  • 用例2

输入 3
5000 4000 3000
输出 3
日常刷题(10)_第2张图片
对于3000这个值
第一层:3000 < 5000 - 500,去左边
第二层:3000 < 4000 - 500,去左边

  • 用例3

输入 9
5000 2000 5000 8000 1800 7500 4500 1400 8100
输出 4
日常刷题(10)_第3张图片
对于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)

你可能感兴趣的:(日常刷题,python,数据结构,华为od)