Cod sursa(job #1684977)

Utilizator badea_adi1999Badea Adrian Catalin badea_adi1999 Data 11 aprilie 2016 13:37:40
Problema Rsir Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <cstdio>
using namespace std;
long long n;
int t0,t1,t2,a,b,x,y,z,m;
int l,s;


inline void next(int &p0,int &pp0,int &p1,int &pp1)
{
    int p2=(a*pp0%m + b*pp1%m + x*p0 + y*p1 +z)%m;
    p0=p1;pp0=pp1;
    p1=p2;pp1=p2*p2%m;
}
int main()
{
    int i,j;
    freopen("rsir.in","r",stdin);
    freopen("rsir.out","w",stdout);
    scanf("%d%d",&t0,&t1);
    scanf("%d%d%d%d%d%d%lld",&a,&b,&x,&y,&z,&m,&n);
    if (n==0) {
        printf("%d\n",t0);
        return 0;
    }
    if (n==1) {
        printf("%d\n",t1);
        return 0;
    }
    int p0,p1,pp0,pp1,q0,qq0,qq1,q1;

    p0=t0%m;p1=t1%m;
    pp0=p0*p0%m;
    pp1=p1*p1%m;

    q0=t0%m;q1=t1%m;
    qq0=q0*q0%m;
    qq1=q1*q1%m;

    for (i=2,j=2;;i++,j+=2) {
        next(p0,pp0,p1,pp1);
        next(q0,qq0,q1,qq1);
        next(q0,qq0,q1,qq1);
        if (p0==q0&&p1==q1) {
            l=j-i+1;
            break;
        }
    }
    s=i;
    p0=t0%m;p1=t1%m;
    pp0=p0*p0%m;
    pp1=p1*p1%m;

    q0=t0%m;q1=t1%m;
    qq0=q0*q0%m;
    qq1=q1*q1%m;

    for (i=1;i<=l;i++)
        next(q0,qq0,q1,qq1);
    for (s=1;;s++) {
        if (p0==q0&&p1==q1)
            break;
        next(p0,pp0,p1,pp1);
        next(q0,qq0,q1,qq1);
    }
    p0=t0;p1=t1;
    pp0=p0*p0;qq0=q0*q0;
    s--;
    if (n<=s) {
        for (i=1;i<=n;i++)
            next(p0,pp0,p1,pp1);
    }
    else {
        n=s+(n-s)%l;
        for (i=1;i<=n;i++)
            next(p0,pp0,p1,pp1);
    }

    printf("%d\n",p0);

    return 0;
}