poj 3708 Recurrent Function

#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
#include<set>
#include<string>
#include<queue>
#include<cmath>
using namespace std;
#define INF 0x3f3f3f3f
const int N_MAX = 10+500;
typedef long long ll;
// 大数类
class bign
{
#define  base  1000
#define  digit  3
private:
    int _arr[110];
    int _m;
    void bign::_simplify(void)
    {
        for (int i = 0; i <= _m; i++)
        {
            if (i == _m && _arr[i] >= base) _m++;
            _arr[i + 1] += _arr[i] / base;
            _arr[i] %= base;
        }
    }
public:
    bign::bign(void) : _m(0) { memset(_arr, 0, sizeof(_arr)); }
    bign::bign(int init) : _m(0)
    {
        memset(_arr, 0, sizeof(_arr));
        _arr[0] = init;
        _simplify();
    }
    friend istream& operator >> (istream& fin, bign& a)
    {
        char init[10010]; int len, b, t;
        fin >> init;
        len = strlen(init); a._m = -1;
        for (int i = len - 1; i >= 0;)
        {
            t = 0, b = 1;
            for (int j = 0; j < digit && i >= 0; j++, i--)
            {
                t += (init[i] - ‘0‘) * b;
                b *= 10;
            }
            a._arr[++a._m] = t;
        }
        return fin;
    }
    friend bign operator / (bign a, int b)
    {
        for (int i = a._m; i >= 0; i--)
        {
            if (a._arr[i] < b && i == a._m && i != 0) a._m--;
            if (i > 0) a._arr[i - 1] += (a._arr[i] % b) * base;
            a._arr[i] /= b;
        } return a;
    }
    friend int operator % (bign a, int b)
    {
        for (int i = a._m; i >= 0; i--)
        {
            if (i == 0) return a._arr[i] % b;
            else a._arr[i - 1] += (a._arr[i] % b) * base;
        }
    }
    friend bool operator == (bign a, bign b)
    {
        if (a._m != b._m) return false;
        for (int i = 0; i <= a._m; i++)
            if (a._arr[i] != b._arr[i]) return false;
        return true;
    }
};

pair<ll,ll> trans(ll *a,ll m,ll k ) {//采用a置换,m为需要置换的数，k为m置换的终点,返回循环次和圈的大小
    ll num = 0;//num为从m到k的循环次数，不存在则为-1
    ll tmp = m;
    while (tmp!=k) {
        if (num != 0 && tmp == m) { num = -1; break; }
        num++;
        tmp = a[tmp];
    }
    if (num == -1)return make_pair(-1, 0);
    int loop = 0;
    while (1) {
        if (loop != 0 && tmp == k) { break; }
        loop++;
        tmp = a[tmp];
    }
    return make_pair(num,loop);
}

ll gcd(ll a,ll b) {
    if (b == 0)return a;
    return gcd(b, a%b);
}

ll extgcd(ll a,ll b,ll &x,ll &y) {
    if (b == 0) {
        x = 1; y = 0;
        return a;
    }
    ll ans = extgcd(b,a%b,x,y);
    ll tmp = x;
    x = y;
    y = tmp - a / b*y;
    return ans;
}
ll mod_inverse(ll a,ll m) {
    ll x, y;
    extgcd(a,m,x,y);
    return (m + x%m) % m;
}


pair<ll, ll>linear_congruence(const ll *A,const ll *B,const ll*M,const int& num) {
    ll x = 0,m = 1;
    for (int i = 0; i < num;i++) {
        ll a = A[i] * m, b = B[i] - A[i] * x, d = gcd(M[i], a);
        if (b%d != 0)return make_pair(0, -1);
        ll t = b / d*mod_inverse(a / d, M[i] / d) % (M[i] / d);
        x = x + m*t;
        m *= M[i] / d;
    }
    return make_pair((x%m+m)%m, m);
}

ll d;
ll a[N_MAX], b[N_MAX];
bign k, m;
ll kd[N_MAX], md[N_MAX];//存储k和m的D进制形式
ll A[N_MAX], B[N_MAX], M[N_MAX];
int main() {
    while(scanf("%lld",&d)&&d!=-1) {
        for (int i = 1; i < d;i++)scanf("%lld",&a[i]);
        for (int i = 0; i < d;i++)scanf("%lld",&b[i]);
        cin >> m >> k;
        int num = 0;
        while (!(m==0)) {
            md[num++] = m%d;
            m =m/ d;
        }
        int num2 = 0;
        while (!(k == 0)) {
            kd[num2++] = k%d;
            k = k / d;
        }
        if (num != num2) { puts("NO"); continue; }
        pair<ll, ll>P;
        bool flag = 0;
        for (int i = 0; i < num-1;i++) {
            P = trans(b,md[i],kd[i]);
            if (P.first == -1) { puts("NO"); flag = 1; break; }
            A[i] = 1, B[i] = P.first, M[i] = P.second;
        }
        if (flag)continue;

        P = trans(a, md[num - 1], kd[num - 1]);
        if (P.first == -1) { puts("NO"); continue; }
        A[num - 1] = 1, B[num - 1] = P.first, M[num - 1] = P.second;
        P = linear_congruence(A,B,M,num);
       if(P.second==-1) { puts("NO"); continue; }
       else printf("%lld\n",P.first);
    }
    return 0;
}