poj3708(公式化简+大数进制装换+线性同余方程组)

//
//  main.cpp
//  poj3708
//
//  Created by 陈加寿 on 15/11/28.
//  Copyright (c) 2015年 陈加寿. All rights reserved.
//

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
using namespace std;


int a[110],b[110];
char strm[110],strk[110];
int m[110],k[110];
int savem[1100],savek[1100];
int cntm,cntk;

void Tentod(int ten[],int len,int &cnt,int d,int save[])
{
    int tcnt=0;
    while(ten[tcnt]==0) tcnt++;
    cnt=0;
    while(tcnt<len)
    {
        for(int i=tcnt;i<len;i++)
        {
            ten[i+1] += (ten[i]%d)*10;
            ten[i] /= d;
        }
        save[cnt++] = ten[len]/10;
        ten[len]=0;
        while(tcnt<len&&ten[tcnt]==0) tcnt++;
    }
    /*
    for(int i=0;i<cnt;i++)
        printf("%d ",save[i]);
    printf("\n");
    */
}



/*对于x=r0(mod m0)
 x=r1(mod m1)
 ...
 x=rn(mod mn)
 输入数组m和数组r，返回[0,[m0,m1,...,mn]-1] 范围内满足以上等式的x0。
 x的所有解为:x0+z*[m0,m1,...mn](z为整数)
 */
long long cal_axb(long long a,long long b,long long mod)
{
    //防乘法溢出
    long long sum=0;
    while(b)
    {
        if(b&1) sum=(sum+a)%mod;
        b>>=1;
        a=(a+a)%mod;
    }
    return sum;
}

//ax + by = gcd(a,b)
//传入固定值a,b.放回 d=gcd(a,b), x , y
void extendgcd(long long a,long long b,long long &d,long long &x,long long &y)
{
    if(b==0){d=a;x=1;y=0;return;}
    extendgcd(b,a%b,d,y,x);
    y -= x*(a/b);
}

long long Multi_ModX(long long m[],long long r[],int n)
{
    long long m0,r0;
    m0=m[0]; r0=r[0];
    for(int i=1;i<n;i++)
    {
        long long m1=m[i],r1=r[i];
        long long k0,k1;
        long long tmpd;
        extendgcd(m0,m1,tmpd,k0,k1);
        if( (r1 - r0)%tmpd!=0 ) return -1;
        k0 *= (r1-r0)/tmpd;
        m1 *= m0/tmpd;
        r0 = ( cal_axb(k0,m0,m1)+r0)%m1;
        m0=m1;
    }
    return (r0%m0+m0)%m0;
}

int main(int argc, const char * argv[]) {
    int d;
    while(cin>>d)
    {
        if(d==-1) break;
        for(int i=1;i<d;i++) cin>>a[i];
        for(int i=0;i<d;i++) cin>>b[i];
        scanf("%s",strm);
        scanf("%s",strk);
        int len = strlen(strm);
        for(int i=0;i<len;i++)
            m[i] = strm[i]-‘0‘;
        Tentod(m,len,cntm,d,savem);
        len = strlen(strk);
        for(int i=0;i<len;i++)
            k[i] = strk[i]-‘0‘;
        Tentod(k, len, cntk, d, savek);
        // 这样就得到了。a，b。。。
        // 然后构建同模方程
        if(cntm != cntk)
        {
            printf("NO\n");
        }
        else
        {
            int flag=0;
            long long m[1010],r[1010];
            for(int i=0;i<cntm-1;i++)
            {
                int a1=savem[i],a2=savek[i];
                int tm=1;
                int ta=a1;
                while(b[ta]!=a1)
                {
                    tm++;
                    ta=b[ta];
                }
                ta=a1;
                int tr=0;
                while(ta != a2)
                {
                    tr++;
                    ta=b[ta];
                    if(ta==a1)
                    {
                        flag=1;
                        break;
                    }
                }
                m[i]=tm;
                r[i]=tr;
                if(flag==1) break;
            }//这里面都是b
            int a1=savem[cntm-1],a2=savek[cntm-1];
            int tm=1;
            int ta=a1;
            while(a[ta]!=a1)
            {
                tm++;
                ta=a[ta];
            }
            ta=a1;
            int tr=0;
            while(ta != a2)
            {
                tr++;
                ta=a[ta];
                if(ta==a1)
                {
                    flag=1;
                    break;
                }
            }
            m[cntm-1]=tm;
            r[cntm-1]=tr;
            if(flag == 1)
            {
                printf("NO\n");
            }
            else
            {
                long long ans=Multi_ModX(m, r,cntm);
                if(ans==-1) printf("NO\n");
                else cout<<ans<<endl;
            }
        }
    }
    return 0;
}