山姆的玩具车

#396. 山姆的玩具车

描述
提交
自定义测试

【题目描述】：

Jasio 是一个三岁的小男孩,他最喜欢玩玩具了,他有n 个不同的玩具,它们都被放在了很高的架子上所以Jasio 拿不到它们. 为了让他的房间有足够的空间,在任何时刻地板上都不会有超过k 个玩具. Jasio 在地板上玩玩具. Jasio’的妈妈则在房间里陪他的儿子. 当Jasio 想玩地板上的其他玩具时,他会自己去拿,如果他想玩的玩具在架子上,他的妈妈则会帮他去拿,当她拿玩具的时候,顺便也会将一个地板上的玩具放上架子使得地板上有足够的空间. 他的妈妈很清楚自己的孩子所以他能够预料到Jasio 想玩些什么玩具. 所以她想尽量的使自己去架子上拿玩具的次数尽量的少,应该怎么安排放玩具的顺序呢?
【输入描述】：

第一行三个整数: n, k, p , 分别表示玩具的总数,地板上玩具的最多个数以及Jasio 他想玩玩具的序列的个数,接下来p行每行描述一个玩具编号表示Jasio 想玩的玩具。
【输出描述】：

一个数表示Jasio 的妈妈最少要拿多少次玩具。
【样例输入】：

3 2 7
1
2
3
1
3
1
2

【样例输出】：

4

【时间限制、数据范围及描述】：

时间：1s 空间：128M

20%的数据：1<=k<=n<=10；1<=p<=20

40%的数据：1<=k<=n<=500；1<=p<=2000

100%的数据：1<=k<=n<=100,000；1<=p<=500,000

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
const int N=500005;
const int M=100005;

int n, k, p,a[N], next[N];
bool vis[M];
int ans = 0;

struct cmp{
    bool operator()(const int&x,const int &y){
    return next[x] < next[y];
    }
};

queue <int> q[M];
priority_queue <int,vector<int>,cmp> h;

int gi(){
    int ans = 0, f = 1; char i = getchar();
    while(i<'0'||i>'9'){if(i=='-')f=-1;i=getchar();}
    while(i>='0'&&i<='9'){ans=ans*10+i-'0';i=getchar();}
    return ans * f;
}

int main(){
	n = gi(); k = gi(); p = gi();
    for(int i=1;i<=p;i++) a[i] = gi(), q[a[i]].push(i);
    for(int i=1;i<=p;i++){
        q[a[i]].pop();
        if(q[a[i]].empty()) next[i] = p+1;
        else next[i] = q[a[i]].front();
    }
    for(int i=1;i<=p;i++){
        if(!vis[a[i]]){
            if(h.size() == k){
                vis[a[h.top()]] = 0;
                h.pop();
            }
            h.push(i); vis[a[i]] = 1; ans++;
        }
        else h.push(i), k++;
    }
    printf("%d\n",ans);
    return 0;
}

标签：玩具车,Jasio,const,int,地板,玩具,山姆,include
来源： https://blog.csdn.net/hssl_ywl/article/details/100620334