【华为OD】2024D卷——停车场车辆统计

简单.is.good2024-09-06 9:12

复制代码

题目描述：
特定大小的停车场，数组cars[]表示，其中1表示有车，0表示没车。
车辆大小不一，小车占一个车位（长度1），货车占两个车位（长度2），卡车占三个车位（长度3），
统计停车场最少可以停多少辆车，返回具体的数目。

输入描述：
整型字符串Q数组cars[]，其中1表示有车，0表示没车，数组长度小于1000。
输出描述：
整型数字字符串，表示最少停车数目。

补充说明：
示例1
输入：
1,0,1
输出： 2
说明：
1个小车占第1个车位第二个车位空
1个小车占第3个车位最少有两辆车

示例2
输入：
1,1,0,0,1,1,1,0,1
输出：
3
说明：
1个货车占第1、2个车位第3、4个车位空
1个卡车占第5、6、7个车位第8个车位空
1个小车占第9个车位最少3辆车

解题思路：

使用贪心策略：遇到车位 = 1时，尝试停放卡车；不行再停放货车；再不行停放小车，即可得最少停放数量

代码部分

python 复制代码

def count_car(nums):
    ncount = 0
    length = len(nums)
    i = 0
    while i < length:
        if nums[i] == 1:
            tmp = 1
            while i + 1 < length and nums[i + 1] == 1:  #统计连续为1车位
                tmp += 1
                i += 1
            ncount += tmp // 3   #先停放卡车
            ncount += (tmp % 3) // 2     #再停放货车
            ncount += (tmp % 3) % 2     #再停放小车
            i += 1
        else:   #直接跳过0车位
            i += 1
    return ncount
nums = list(map(int, input().split(',')))
print(count_car(nums))

换一种更容易懂的写法

python 复制代码

    while i < length:
        if nums[i] == 1:
            if i + 2 < len(nums) and nums[i + 1] == 0 and nums[i + 2] == 0:
                # 停放卡车  
                nums[i + 2] = 1
                nums[i + 1] = 1
                nums[i] = 1
                ncount += 1
                i += 3
            elif i + 1 < len(nums) and nums[i + 1] == 0:
                # 停放货车  
                nums[i + 1] = 1
                nums[i] = 1
                ncount += 1
                i += 2
            else:
                # 停放小车  
                nums[i] = 1
                ncount += 1
                i += 1
        else:
            i += 1

知识点：逻辑、贪心

结语：越简单的题目解法应该越多，请路过大神留下新的思路供本小白学习一下，打开思路