2025A卷-生日礼物-CSDN博客

本文链接：https://blog.csdn.net/tu123456s/article/details/147346063

题目描述

小牛的孩子生日快要到了，他打算给孩子买蛋糕和小礼物，蛋糕和小礼物各买一个，他的预算不超过x元。蛋糕cake和小礼物gift都有多种价位的可供选择。
请返回小牛共有多少种购买方案。

备注：

```
1 ≤ cake.length ≤ 10^5
```
```
1 ≤ gift.length ≤10^5
```
```
1 ≤ cake[i]，gift[i] ≤ 10^5
```
```
1 ≤ X ≤ 2*10^5
```

输入描述

第一行表示cake的单价，以逗号分隔

第二行表示gift的单价，以逗号分隔

```
第三行表示x预算
```

输出描述

输出数字表示购买方案的总数

用例

【用例1】

输入

10,20,5
5,5,2
15

输出

说明

小牛有6种购买方案，所选蛋糕与所选礼物在数组中对应的下标分别是：

第1种方案: cake [0] + gift [0] = 10 + 5 = 15;
第2种方案: cake [0] + gift [1]= 10 + 5 = 15;
第3种方案: cake [0] + gift [2] = 10 + 2 = 12;
第4种方案: cake [2] + gift [0] = 5 + 5 = 10;
第5种方案: cake [2] + gift [1]= 5 + 5 = 10;
第6种方案: cake [2] + gift [2] = 5 + 2 = 7。

Python代码实现

思路：

1、将cake和gift数组，都排序

2、遍历cake数组，通过二分查找去gift数组中查找满足要求的元素有多少个

3、将找到的元素个数相加，得到总的total

注意：原题目解答时，应该需要使用Input()函数输入字符串，然后转换成对应的数组或者数字，这里为了方便调试，直接省去了这一步，但是真正答题时，需要通过input()来输入。

from typing import List


# 二分查找
# 这里需要找的是，小于或等于指定数m的坐标，所以返回mid
def binary_find(nums: List, m):
    left, right = 0, len(nums) - 1
    mid = -1
    while left <= right:   # 左闭右闭区间 [left,right]
        mid = (left + right) // 2
        if nums[mid] <= m:
            left = mid + 1
        else:
            right = mid - 1
        # print(f"left,right:{left} {right}")
    return mid


def main(cake_list: List, gift_list: List, x):
    cake_list.sort()
    gift_list.sort()
    # print(cake_list)
    # print(gift_list)
    total = 0
    for y in cake_list:
        if y >=x:
            break
        count = binary_find(nums=gift_list,m=x-y)+1
        # print(f"count: {count}")
        total += count
    print(total)


if __name__ == '__main__':
    cake = [10, 20, 5]
    gift = [5, 5, 2]
    x = 15

    # cake = [i for i in range(100000)]
    # gift = [i for i in range(100000)]
    # x = 29343

    main(cake, gift, x)