最多颜色的车辆

题目描述

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

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

输入描述

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

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

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

输出描述

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

用例1

输入

0 1 2 1
3

Copy

输出

2

Copy

说明

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

用例2

输入

0 1 2 1
2

Copy

输出

1

#include <iostream>
using namespace std;
#include <string>
#include <sstream>
#include <vector>
#include <map>

int main() {
    //先把输入的所有车辆颜色按照字符串的形式接收过来
    string strs;
    getline(cin,strs);

    //将strs转换成stringstream流方便截取
    stringstream ss(strs);

    //定义一个vector容器 接受每一种颜色车的序号
    vector<int> v;

    //定义一个接受每一种颜色车的变量 方便把数据压入vector容器
    int color;

    //定义滑动窗口
    int w = 0;
    cin>>w;

    //定义记录最多次数的变量
    int max_count = 0;

    //定义map容器存储车的颜色和对应出现次数
    map<int,int> m;

    while(ss>>color)
    {
        v.push_back(color);
    }

    //现在我们的vector容器中已经有了所有车了
    //对滑动窗口有两步处理

    //初始处理 取最小值来遍历
    for(int i = 0; i < min(w,(int)v.size());i++)
    {
        int c = v[i];
        m[c]++;
        max_count = max(m[c],max_count);
    }

    //二轮处理 让滑动窗口滑动起来
    for(size_t i = w; i < v.size();i++)
    {
        int c = v[i];
        int out_num = v[i - w] ;
        m[c]++;
        m[out_num]--;
        if (m[out_num] == 0) {
            m.erase(out_num);
        }
        max_count = max(m[c],max_count);
    }

    cout<<max_count<<endl;
    
}