BSOJ:3768 玩具车--贪心+堆

3768 -- 【模拟试题】玩具车
Description
　　Jasio 是一个三岁的小男孩，他最喜欢玩玩具了，他有n 个不同的玩具,它们都被放在了很高的架子上，所以Jasio 拿不到它们。为了让他的房间有足够的空间，在任何时刻地板上都不会有超过k个玩具。
　　Jasio在地板上玩玩具。Jasio的妈妈则在房间里陪他的儿子，当Jasio 想玩地板上的其他玩具时，他会自己去拿，如果他想玩的玩具在架子上，他的妈妈则会帮他去拿，当她拿玩具的时候，顺便也会将一个地板上的玩具放上架子使得地板上有足够的空间。
　　他的妈妈很清楚自己的孩子，所以他能够预料到Jasio想玩些什么玩具。所以她想尽量的使自己去架子上拿玩具的次数尽量的少，应该怎么安排放玩具的顺序呢?
Input
　　输入文件的第一行三个整数：n, k, p (1<=k<=n <=100000, 1<=p<=500000)，分别表示玩具的个数，地板上玩具的最多个数以及Jasio他想玩玩具的序列个数，接下来p行每行描述一个整数表示Jasio想玩的玩具编号。
Output
　　输出文件只有一行为一个整数，表示Jasio 的妈妈最少要拿多少次玩具。
Sample Input
3 2 7
1
2
3
1
3
1
2
Sample Output
4
Hint
【数据范围】
　　对于50%的数据1<=n<=500，1<=k<=200，1<=p<=2000

　　对于100%的数据1<=n<=100000，1<=k<=100000，1<=p<=500000

贪心策略是个人都想得到吧..当空间不足时，删除下一个使用时间最晚的玩具..毕竟样例给得很清楚了。

那么这个就是明显的使用堆了，删除时取堆顶pop即可，当然记一个hash记录这个玩具在不在地板上。

然而写的时候WA了无数次..原因一是当发现地板上有这个玩具时，没有把新的next时间截给push进去，导致时间乱完了..

二是改正这个错误之后，一开始我只是单纯地把k个元素push进去，其实开始时必须要把地板铺满才能结束预处理，也即当hash[a[i]]=1的时候要记录再多入一个，不然k个push完后不一定能铺的满地板。

使用堆的时候注意每次pop无效数据。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<queue>
#define ll long long
#define pa pair<ll,ll>
using namespace std;
priority_queue<pa> Q;
inline ll read()
{
	ll bj=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-')bj=-1;
		ch=getchar();
	}
	ll ret=0;
	while(ch>='0'&&ch<='9')ret=ret*10+ch-'0',ch=getchar();
	return ret*bj;
}
ll n,k,p,a[1000005]={0},pos[1000005]={0},next[1000005]={0};
bool ha[1000005]={0};
void init()
{
	n=read();k=read();p=read();
	for(ll i=1;i<=p;i++)a[i]=read();
	for(ll i=1;i<=p;i++)
	{
		if(!pos[a[i]]){pos[a[i]]=i;continue;}
		else 
		{
			next[pos[a[i]]]=i;
			pos[a[i]]=i;
		}
	}
	for(ll i=1;i<=p;i++)if(next[i]==0)next[i]=p+1;
	
} 
void greed()
{
	ll ans=0;
	for(ll i=1;i<=k;i++)
	{
		if(ha[a[i]])
		{
			Q.push(make_pair(next[i],a[i]));
			k++;
			continue;
		}
		ans++;
		Q.push(make_pair(next[i],a[i]));
		ha[a[i]]=1;
	}
	for(ll i=1;i<=p;i++)
	{
		if(ha[a[i]])
		{
			Q.push(make_pair(next[i],a[i]));
			continue;
		}
	    while(!Q.empty()&&ha[Q.top().second]==0)Q.pop();
		ha[Q.top().second]=0;
		Q.pop();
		Q.push(make_pair(next[i],a[i]));
		ha[a[i]]=1;
		ans++;
	}
	printf("%lld\n",ans);
}
int main()
{
	init();
	greed();
return 0;
}
/*
3 2 7
1
2
3
1
3
1
2
*/