百练 2818 密码

链接地址：http://poj.grids.cn/practice/2818/

题目：

描述

Bob 和 Alice 开始使用一种全新的编码系统。它是一种基于一组私有钥匙的。他们选择了n个不同的数a1 ， . . .，an, 它们都大于0小于等于n。 机密过程如下：待加密的信息放置在这组加密钥匙下，信息中的字符和密钥中的数字一一对应起来。信息中位于i位置的字母将被写到加密信息的第ai个位置, ai 是位于i位置的密钥。加密信息如此反复加密，一共加密 k 次。

信息长度小于等于n。如果信息比 n 短, 后面的位置用空格填补直到信息长度为n。

请你帮助 Alice 和 Bob 写一个程序，读入密钥，然后读入加密次数 k 和要加密的信息，按加密规则将信息加密。

输入

输入包括几块。每块第一行有一个数字n, 0 < n <= 200. 接下来的行包含n个不同的数字。数字都是大于0小于等于n的。下面每行包含一个k和一个信息字符串，它们之间用空格格开。每行以换行符结束，换行符不是要 加密的信息。每个块的最后一行只有一个0。 最后一个块后有一行，该行只有一个0。

输出

输出有多个块，每个块对应一个输入块。每个块包含输入中的信息经过加密后的字符串，顺序与输入顺序相同。所有加密后的字符串的长度都是 n。 每一个块后有一个空行。

样例输入

10
4 5 3 7 2 8 1 6 10 9
1 Hello Bob
1995 CERC
0
0

样例输出

BolHeol  b
C RCE

代码

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
using namespace std;
int main()
{
    //freopen("F:\\input.txt","r",stdin);
    int n;
    while(scanf("%d",&n),n != 0)
    {
        int *nums = new int[n];
        for(int i = 0; i < n; i++) cin>>nums[i];
        
        //预处理
        int* circles = new int[n];
        for(int i = 0; i < n; i++)
        {
            int pos = i;
            int j = 0;
            do
            {
                pos = nums[pos] - 1;
                j++;
            }while(pos != i);
            circles[i] = j;
        }
        
                
        int k;
        while(scanf("%d",&k),k != 0)
        {
            getchar();
            
            char* chs = new char[n+1];
            memset(chs,' ',sizeof(char)*(n+1));
            
            gets(chs);
            
            int length = strlen(chs);
            chs[length] = ' ';
            chs[n] = '\0';
            
            /*char* chs1 = new char[n+1];
            char* chs2 = new char[n+1];
            strcpy(chs1,chs);
            strcpy(chs2,chs);
            
            for(int i = 0;i < k; i++)
            {
                for(int j = 0; j < n; j++)
                {
                    chs2[nums[j]-1] = chs1[j];
                }
                
                if(strcmp(chs,chs2) == 0)
                {
                    i = k/(i+1)*(i+1) - 1;
                    strcpy(chs2,chs);
                } 
                
                strcpy(chs1,chs2);
            }
            cout<<chs1<<endl;
            */
            
            char *chs3 = new char[n+1];
            for(int i = 0; i < n; i++)
            {
                int nk = k % circles[i];
                int pos = i;
                for(int j = 0; j < nk; j++)
                {
                    pos = nums[pos] - 1;
                }
                chs3[pos] = chs[i];
            }
            chs3[n] = '\0';
            cout<<chs3<<endl;
            
        }
        cout<<endl;
    }
    
    return 0;
}

思路：
（1）注意要优化，不然会超时
（2）读入读出要注意