【Leetcode刷题Python】矿泉水问题

深信服公司的算法笔试题

1 题目

小明横穿沙漠，需要携带至少x毫升的水。

有两种规格的矿泉水可供选购：小瓶矿泉水每瓶500ml，价格a元。大瓶矿泉水每瓶1500ml，价格b元。

小明打算买一些矿泉水用于横穿沙漠，为了保证至少买到x毫升的水，小明至少需要花费多少钱？

输入描述:

第一行一个正整数t(t<=1000)，表示有t组测试数据;
接下来t行，每行3个正整数：x，a，b。其中x<=1,000,000,000，表示小明至少需要x毫升水；a<=100，b<=100，分别表示小瓶和大瓶矿泉水的价格，单位：元。

输出描述:

每组测试数据输出一行，表示小明最少需要花费的钱，单位：元。

输入例子1:
3 5000 5 10 4999 5 10 5000 5 100

输出例子1:
35 35 50

2 解析

• 先看1500ml 是买大瓶的便宜还是小瓶的便宜？
• 剩下的部分肯定不足1500ml，是买1瓶大瓶的 还是 几个小瓶的？
• 把上面2部分的钱加起来

3 Python实现

import math
N = int(input())
res = []
def getmoney(x,a,b):
    low_p = min(a*3,b)
    most_p = x//1500 *low_p
    rest_w =x%1500
    rest_p = min(math.ceil(rest_w/500)*a,b)
    return most_p+rest_p
    
for i in range(N):
    x,a,b = [int(s) for s in str(input()).split()]
    res.append(getmoney(x,a,b))
for i in range(N):
    print(res[i])