poj1026 Cipher

题目意思可概括为给定集合S = {1,..,n}的一个双射关系f, 求经过k次复合之后元素i对应的元素f^k(i) (i∈S)。

由于函数是双射，一个原像对应唯一一个像，同样一个像有唯一一个原像，考虑整个映射关系，存在整数m∈ Z,使得f^m=f⁰=I。

即具有周期性。

每个元素映射回它自己有独立的周期T(i),整个映射的周期T=lcm(T(i)), i ∈ S。

独立处理更快，但对于本题也是刚刚卡过。

当然如果事先把所有询问读入加以预处理或者直接全部预处理会更快。

样例代码860ms卡过。

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

 

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 200 + 10;
int n;
char s[maxn];
int f[maxn];
int period[maxn];
int repeats;

struct Node{
    int from, to;
}node[maxn];

int gcd(int a, int b){
    if(!b) return a;
    return gcd(b, a % b);
}

bool cmp(Node a, Node b){
    return a.to < b.to;
}

void init(){
    for(int i = 0; i < n; i++){
        int t = f[i], cnt = 1;
        while(i != t) t = f[t], ++cnt;
        period[i] = cnt;
    }
}

void solve(){
    for(int i = 0; i < n; i++) node[i].from = node[i].to = i;
    for(int i = 0; i < n; i++){
        int p1 = repeats % period[i];
        while(p1--) node[i].to = f[node[i].to];
    }
    sort(node, node + n, cmp);
    for(int i = 0; i < n; i++) printf("%c", s[node[i].from]);
    printf("
");
}

int main(){
    while(~scanf("%d", &n) && n){
        for(int i = 0, j; i < n; i++){
            scanf("%d", &j);
            f[i] = j - 1;
        }
        init();
        while(~scanf("%d", &repeats) && repeats){
            getchar();
            gets(s);
            int len = strlen(s);
            for(int i = len; i < n; i++) s[i] = ' ';
            s[n] = '';
            solve();
        }
        printf("
");
    }
    return 0;
}