LOJ #541.「LibreOJ NOIP Round #1」七曜圣贤 题解

题目传送门

题目大意： 有 $b$ 种红茶，一开始有 $0$ 到 $a$ 种类的红茶，每次操作给出一个 $p_i$，如果 $p_i=-1$，那么就捡起之前最早丢掉的红茶，否则，如果你拥有种类 $p_i$ 的红茶，那么丢掉它，如果你没有并且之前没有拥有过，那么就购买这个红茶，否则，如果之前拥有过，那么就还是执行 捡起之前最早丢掉的红茶。减红茶时，如果没红茶减就忽视这次操作，然后每次操作完后求出一个 $k$，满足你不拥有种类k的红茶，但是种类0~k-1的红茶你都有。

题解

$k$ 实际上就是，最小的手里没有的红茶。

这个东西等于 $\min \{$最小的没买过的红茶$,$最小的丢掉了的红茶$\}$。

前者用个bool数组统计则可，后者可以用单调队列维护。

代码如下：

#include 
#include 
#include 
using namespace std;
#define maxn 2000010
#define inf 999999999

int T,n,a,b,d,p[maxn],mi;
namespace IO{
	int c;
	unsigned int seed;
	unsigned int randnum(){
		seed^=seed<<13;
		seed^=seed>>17;
		seed^=seed<<5;
		return seed;
	}
	inline void init_case(int &m,int &a,int &b,int &d,int p[]){
		scanf("%d %u %d %d %d %d",&m,&seed,&a,&b,&c,&d);
		for(int i=1;i<=m;i++){
			if(randnum()%c==0)p[i]=-1;
			else p[i]=randnum()%b;
		}
	}
	inline void update_ans(unsigned int &ans_sum,unsigned int cur_ans,int no){
		const static unsigned int mod=998244353;
		ans_sum^=(long long)no*(no+7)%mod*cur_ans%mod;
	}
}
struct Queue{
	int q[maxn],st,ed;
	inline void init(){st=1,ed=0;}
	inline void push(int x){q[++ed]=x;}
	inline int front(){return st<=ed?q[st]:inf;}
	inline int back(){return q[ed];}
	inline void pop_front(){st++;}
	inline void pop_back(){ed--;}
	bool empty(){return st>ed;}
}q1,q2;
unsigned int ans_sum,cur_ans;
void up(int x)
{
	cur_ans=min(q2.front(),mi);//printf("ans : %u\n",cur_ans);
	IO::update_ans(ans_sum,cur_ans,x);
}
bool v[maxn],bought[maxn];
void work1(int x)
{
	bought[p[x]]=v[p[x]]=true;
	while(bought[mi])mi++; up(x);
}
void work2(int x)
{
	if(d)return;
	v[p[x]]=false;q1.push(p[x]);
	while(!q2.empty()&&q2.back()>p[x])q2.pop_back();
	q2.push(p[x]);up(x);
}
void work3(int x)
{
	if(d||q1.empty())return;
	v[q1.front()]=true;
	if(q2.front()==q1.front())q2.pop_front();
	q1.pop_front();up(x);
}

int main()
{
	scanf("%d",&T);while(T--)
	{
		ans_sum=0;q1.init();q2.init();
		IO::init_case(n,a,b,d,p);
		memset(v,false,sizeof(v));
		memset(bought,false,sizeof(bought));
		for(int i=0;i<=a;i++)bought[i]=v[i]=true;mi=a+1;
		for(int i=1;i<=n;i++)
		{
			if(p[i]==-1)work3(i);
			else if(v[p[i]])work2(i);
			else if(!bought[p[i]])work1(i);
			else work3(i);
		}
		printf("%u\n",ans_sum);
	}
}