LRU-K 缓存(队列+模拟)

题目链接：题目详情 - 7-7 LRU-K 缓存 (pintia.cn)

样例1输入： 

2 5 17
9 5 6 7 8 3 8 9 5 9 8 3 4 7 5 6 9

样例1输出：

4 9
8 3 7 5 6

样例2输入：

3 5 10
9 5 6 7 8 3 8 9 5 9

样例2输出：

7 3 8 5 9
-

分析：题意非常简单，就是按照题意所描述的直接进行模拟即可，但是我们可以发现假如当前访问x，但是x已经在优先队列里面了，那么我们要把所有的元素都拿出来，然后再把x放到队尾，这样复杂度就是O(n)的，那么肯定会超时，那么我们怎么优化这个过程呢？就是我们队列里面存放的是这个数x最后一次出现的位置，假如我们现在访问x，而x已经在队列中了，那么我们就把x的有效位置设置为++idx，其中idx记录当前第几个数。那么我们判断队首元素应不应该出队列就通过判断当前队列的实际容量是否小于等于n以及当前队列的队首元素是否有效来进行判断即可。既然加上了一个位置，那么我们队列里面装的数据类型就是pair类型了，优化了这个点之后就可以AC了。

细节见代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=1e6+10;
int pos1[N],pos2[N];//pos1/2[i]分别统计i在q1和q2两个队列中的有效位置 
int vis[N];//vis[i]统计i在第一个队列中被几次命中 
queue<pair<int,int> > q1,q2;
int main()
{
	int k,n,m;
	cin>>k>>n>>m;
	int idx=0;
	int t1=0,t2=0;//t1,t2分别记录q1和q2两个队列中的有效元素个数 
	for(int i=1;i<=m;i++)
	{
		while(t1>n)
		{
			pair<int,int> p=q1.front();
			q1.pop();
			if(pos1[p.first]==p.second)
			{
				t1--;
				vis[p.first]=0;
				pos1[p.first]=0;
			}
		}
		while(t2>n)
		{
			pair<int,int> p=q2.front();
			q2.pop();
			if(pos2[p.first]==p.second)
			{
				t2--;
				pos2[p.first]=0;
			} 
		}
		int x;
		scanf("%d",&x);
		if(pos2[x])//在第二个队列里面 
		{
			pos2[x]=++idx;//赋予一个新的编号，默认放至队尾
			q2.push({x,idx});
		}
		else if(pos1[x])//在第一个队列里面 
		{
			vis[x]++;
			if(vis[x]==k)
			{
				pos1[x]=0;//将其从队列1中移除
				t1--;
				vis[x]=0;//将其在队列1中出现次数设置为0 
				pos2[x]=++idx;//加入队列2中 
				t2++;
				q2.push({x,idx});
			}
			else
			{
				pos1[x]=++idx;
				q1.push({x,idx});
			}
		}
		else//既不在队列1里面，也不在队列2里面，就将其加入至队列1 
		{
			vis[x]++;
			pos1[x]=++idx;
			t1++; 
			q1.push({x,idx});
		}
	}
	//装完最后一个元素还要保证q1和q2两个队列是有效的，也就是不能超队列容量
	while(t1>n)
	{
		pair<int,int> p=q1.front();
		q1.pop();
		if(pos1[p.first]==p.second)
		{
			t1--;
			vis[p.first]=0;
			pos1[p.first]=0;
		}
	}
	while(t2>n)
	{
		pair<int,int> p=q2.front();
		q2.pop();
		if(pos2[p.first]==p.second)
		{
			t2--;
			pos2[p.first]=0;
		} 
	}
	if(!t1) printf("-");
	else
		while(t1)
		{
			pair<int,int> p=q1.front();
			q1.pop();
			if(pos1[p.first]==p.second)
			{
				printf("%d",p.first),t1--;
				if(t1) printf(" ");
			}
			
		}
	puts("");
	if(!t2) printf("-");
	else
		while(t2)
		{
			pair<int,int> p=q2.front();
			q2.pop();
			if(pos2[p.first]==p.second)
			{
				printf("%d",p.first),t2--;
				if(t2) printf(" ");
			}
		}
	return 0;
}