【最多颜色的车辆】

题目描述

【最多颜色的车辆】

题目描述

在一个狭小的路口，每秒只能通过一辆车，假好车辆的颜色只有 3 种，找出 N 秒内经过的最多颜色的车辆数量。

三种颜色编号为0 ，1 ，2

输入描述

第一行输入的是通过的车辆颜色信息

[0,1,1,2] 代表4 秒钟通过的车辆颜色分别是 0 , 1 , 1 , 2

第二行输入的是统计时间窗，整型，单位为秒

输出描述

输出指定时间窗内经过的最多颜色的车辆数量。

示例1   输入输出示例仅供调试，后台判断数据一般不包含示例

输入

0 1 2 1
3

输出

2

样例解释

在 3 秒时间窗内，每个颜色最多出现 2 次。例如：[1,2,1]

示例1   输入输出示例仅供调试，后台判断数据一般不包含示例

输入

0 1 2 1

2

输出

1

样例解释

在 2 秒时间窗内，每个颜色最多出现1 次。

解题：滑动窗口的应用

代码一：

s = list(map(int,input().split()))#0 1 2 1
r = int(input())
l = 0
d = {0:0,1:0,2:0}
while r <= len(s):
    d[0] = max(d[0],s[l:r].count(0))
    d[1] = max(d[1],s[l:r].count(1))
    d[2] = max(d[2],s[l:r].count(2))
    l += 1
    r += 1

ret = sorted(d.values(),key=lambda x:x,reverse=True)
print(ret[0])

代码二：

# 输入获取
arr = input().split()
n = int(input())
 
 
# 算法入口
def getResult(arr, n):
    count = {
        "0": 0,
        "1": 0,
        "2": 0
    }
 
    l = 0
    r = l + n
 
    for c in arr[l:r]:
        count[c] += 1
 
    maxV = max(count.values())
 
    while r < len(arr):
        add = arr[r]
        remove = arr[l]
 
        r += 1
        l += 1
 
        count[add] += 1
        count[remove] -= 1
 
        maxV = max(maxV, count[add])
 
    return maxV
 
 
# 算法调用
print(getResult(arr, n))