poj 1026 Cipher

http://poj.org/problem?id=1026

这道题题意是给你一个置换群，再给你一个字符串，输出经过k次置换的字符串。 就是找循环节。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 3000
using namespace std;

int a[maxn],key[maxn];
int n,k;
char str[maxn],ans[maxn];
void get_key()
{
    for(int i=1; i<=n; i++)
    {
        int t1=1;
        int id=a[i];
        while(id!=i)
        {
            id=a[id];
            t1++;
        }
        key[i]=t1;
    }
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0) break;
        memset(a,0,sizeof(a));
        memset(key,0,sizeof(key));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        get_key();
        while(scanf("%d",&k)!=EOF)
        {
            if(k==0) break;
            getchar();
            gets(str+1);
            int len=strlen(str+1);
            for(int i=len+1; i<=n; i++) str[i]=' ';
            str[n+1]='';
            for(int i=1; i<=n; i++)
            {
                int id1=i;
                for(int j=0; j<k%key[i]; j++)
                {
                    id1=a[id1];
                }
                ans[id1]=str[i];
            }
            ans[n+1]='';
            for(int i=1; i<=n; i++)
            {
                printf("%c",ans[i]);
            }
            printf("
");
        }
        printf("
");
    }
    return 0;
}