记录编号 91517 评测结果 AAAAAAAAAA
题目名称 Bessie洗牌法 最终得分 100
用户昵称 Gravatarcstdio 是否通过 通过
代码语言 C++ 运行时间 0.109 s
提交时间 2014-03-15 09:45:47 内存使用 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 beltitem(int x){//传送链上(从0后面那个数开始)的第x个数是什么
	if(x<=belt.size()-2) return anspr[x];//开始时传送链上的编号规律和后面是不一样的
	return M+x-(belt.size()-1);
}
int value(int x){//第x张被拿走的牌
	if(x>N-M+1) return anssu[x-(N-M+1)];
	return beltitem(x);//否则就是传送链上的第x张牌了
}
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=belt.size()-1+N-M;
	for(int i=1;i<=belt.size()-2;i++) anssu[belt[i]]=beltitem(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;
}