UOJ #584. 天天去哪吃

【题目描述】：

吃饭是自古以来令人头疼的问题，最后天天决定写一个程序来替他做决定。

现在已知有n个餐厅，编号从0到n-1。记ai表示第i天决定吃什么，令

t[i][0]=(A×a[i-1]+B×a[i-2])%n，(其实就是一个随机数)

t[i][j]=(t[i][j-1]+1)%n，j≥1

天天不想去最近吃过的餐厅，所以如果在前n/2(整除)天中吃过某个餐厅，就不想再去这个餐厅。所以我们要找一个最小的k，使得t[i][k]不为前n/2(整除)天去过的餐厅。那么天天第i天就想去t[i][k]餐厅。

现在已知a1和a2，想让你输出a3…am，即接下来m-2天，天天会去哪里吃饭。
【输入描述】：

第一行两个正整数n和m。其中2≤n≤10^5，3≤m≤2×10^5；

第二行两个整数，分别为A和B。其中0≤A,B≤10^9；

第三行两个整数，为a[1]和a[2]。0≤a[1],a[2]＜n，且a[1]≠a[2]；
【输出描述】：

一行m-2个整数，表示接下来m-2天里，天天去哪里吃饭。
【样例输入】：

3 4
7 11
2 0

【样例输出】：

1 2

【样例说明】：

t[3][0]=(7×a[2]+11×a[1])%n=1，前面n/2=1天没有出现，所以第三天去1号餐厅。

t[4][0]=1，前面1天已经出现，所以第四天去t[4][1]=2号餐厅。
【时间限制、数据范围及描述】：

时间：1s 空间：256M

对于60% 的数据：2≤n,m≤10^4；

对于100% 的数据：2≤n≤10^5，3≤m≤2×10^5；且保证A,B为随机的两个素数，所以可以认为t[i][0]为一个随机数。

本题直接用双端队列模拟即可.

Code:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<ctime>
#include<deque>
using namespace std;
const long long N=500005;
long long n,m,A,B,a[N];
bool vis[N];
deque<long long> Q;
int main(){
	scanf("%lld%lld%lld%lld%lld%lld",&n,&m,&A,&B,&a[1],&a[2]);
	vis[a[1]]=1;
	vis[a[2]]=1;
	Q.push_back(a[1]);
	Q.push_back(a[2]);
	for(int i=3;i<=m;i++){
		long long x=(A*a[i-1]+B*a[i-2])%n;
		if(vis[x]){
			for(int i=x;i<n;i++){
				if(!vis[i]){
					x=i;
					break;
				}
			}
		}
		if(vis[x]){
			for(int i=0;i<x;i++){
				if(!vis[i]){
					x=i;
					break;
				}
			}
		}
		Q.push_back(x);
		vis[x]=1;
		a[i]=x;
		if(Q.size()>n/2){
			vis[Q.front()]=0;
			Q.pop_front();
		}
	}
	for(int i=3;i<=m;i++){
		printf("%lld ",a[i]);
	}
	return 0;
}