poj 1026 Cipher (置换群)

求一个置换的第k次置换的结果。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define B(x) (1<<(x))
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
void cmax(ll& a,ll b){ if(b>a)a=b; }
void cmin(ll& a,ll b){ if(b<a)a=b; }
void add(int& a,int b,int mod){ a=(a+b)%mod; }
void add(ll& a,ll b,ll mod){ a=(a+b)%mod; }
const int oo=0x3f3f3f3f;
const int MOD=2015;
const int maxn=220;
char str[maxn];
int a[maxn],vis[maxn];
int cir[maxn][maxn],num[maxn];
char ans[maxn];

int main(){
    //reopen("G:\\read.txt","r",stdin);
    int n,k,cnt;
    while(scanf("%d",&n)!=EOF){
        if(n==0)break;
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        memset(vis,0,sizeof vis);
        cnt=0;
        for(int i=1;i<=n;i++){
            if(!vis[i]){
                vis[i]=1;
                num[cnt]=0;
                int temp=a[i];
                cir[cnt][num[cnt]++]=temp;
                while(!vis[temp]){
                    vis[temp]=1;
                    temp=a[temp];
                    cir[cnt][num[cnt]++]=temp;
                }
                cnt++;
            }
        }
        while(scanf("%d",&k)){
            if(k==0)break;
            gets(str);
            int len=strlen(str);
            for(int i=len;i<=n;i++)str[i]=' ';
            str[n+1]='\0';
            for(int i=0;i<cnt;i++)
                for(int j=0;j<num[i];j++)
                    ans[cir[i][(j+k)%num[i]]]=str[cir[i][j]];
            ans[n+1]='\0';
            printf("%s\n",ans+1);
        }
        puts("");
    }
    return 0;
}