记录编号	91516	评测结果	AAAWWAAAAA
题目名称	1544.Bessie洗牌法	最终得分	80
用户昵称	cstdio	是否通过	未通过
代码语言	C++	运行时间	0.106 s
提交时间	2014-03-15 09:21:25	内存使用	2.70 MiB

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int SIZEM=100010;
int P[SIZEM]={0};
int N,M,Q;
vector<int> belt;//从第M号到第0号的"传送链"
vector<int> cycle[SIZEM];
bool visit[SIZEM]={0};
int numc=0;
int anspr[SIZEM]={0},anssu[SIZEM]={0};//答案的前若干个和后若干个
int value(int x){//第x张被拿走的牌
	if(x<=belt.size()-2) return anspr[x];
	if(x>N-M+1) return anssu[x-(N-M+1)];
	return M+x-(belt.size()-1);
}
void answer(void){
	int q;
	for(int i=1;i<=Q;i++){
		scanf("%d",&q);
		printf("%d\n",value(N+1-q));
	}
}
void prepare(void){
	int p=0,step=N-M+1;
	for(int i=belt.size()-2;i>0;i--) anspr[++p]=belt[i];//最开始的编号就是它们的下标
	p=N;
	for(int i=1;i<=belt.size()-2;i++) anssu[belt[i]]=p--;
	for(int i=1;i<=numc;i++){
		for(int j=0;j<cycle[i].size();j++){
			p=(j+step)%cycle[i].size();
			anssu[cycle[i][p]]=cycle[i][j];
		}
	}
}
void getcyc(void){
	int x=M;
	while(x){
		visit[x]=true;
		belt.push_back(x);
		x=P[x];
	};
	visit[0]=true;
	belt.push_back(0);
	for(int i=1;i<=M;i++){
		if(visit[i]) continue;
		x=i;
		numc++;
		do{
			visit[x]=true;
			cycle[numc].push_back(x);
			x=P[x];
		}while(x!=i);
	}
}
void read(void){
	scanf("%d%d%d",&N,&M,&Q);
	for(int i=1;i<=M;i++) scanf("%d",&P[i]),P[i]--;
}
int main(){
	freopen("shufflegold.in","r",stdin);
	freopen("shufflegold.out","w",stdout);
	read();
	getcyc();
	prepare();
	answer();
	return 0;
}