POJ密码

逻辑很清晰，但写的时候犯了两个小错误，居然样例都正好过了。

1. 计算每个字符的位置时，忘了用temp保存k。
2. 给ans赋值完之后，忘了在最后加’\0’。但由于ans是全局变量，一开始自动赋值为全0，也就是’\0’的值，并且样例输入只有一组，导致可以通过。

思路： 根据密钥计算每个位置的循环次数，再计算原字符串中的每个字符应该放到哪个位置。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,code[210],k;
char str[210],ans[210];

int cycle[210];
void compute() {
	for(int i=0; i<n; i++) {
		int c=0,t=i;
		do {
			t=code[t];//通过当前的下标，计算下一次应该出现的位置 
			c++;
		} while(t!=i);
		cycle[i]=c;
	}
}
int main() {

	while(scanf("%d",&n),n) {
		memset(cycle,0,sizeof(cycle));
		for(int i=0; i<n; i++) {
			scanf("%d",&code[i]);
			code[i]--;
		}
		while(scanf("%d",&k),k) {
			getchar();
			gets(str);
			int l=strlen(str);
			int i;
			for(i=l; i<n; i++) {
				str[i]=' ';
			}
			str[i]='\0';

			compute();
			int temp=k; 
			for(int i=0; i<n; i++) {
				int t=temp%cycle[i],k=i;
				while(t--) {
					k=code[k];
				}
				ans[k]=str[i];
			}
			ans[n]='\0';
			printf("%s\n",ans);
		}
		printf("\n");
	}

}