Cod sursa(job #3135237)

Utilizator maraboneaMara Bonea marabonea Data 2 iunie 2023 14:00:26
Problema Ridicare la putere in timp logaritmic Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.98 kb
#include <fstream>
#define MOD 1999999973
using namespace std;
ifstream fin("lgput.in");
ofstream fout("lgput.out");
int n,p;
int sol[3][3],put[3][3];
void pwr(int a[3][3],int b[3][3])
{
    long long  c[3][3];
    for(int i=1;i<=2;i++)
    {
        for(int j=1;j<=2;j++)
        {
            c[i][j]=0;
        }
    }

     for(int i = 1; i <=2; i++)
        for(int j = 1; j <=2; j++)
            {
               for(int k = 1; k <=2; k++)
                    c[i][j] = (c[i][j] + 1LL * a[i][k] * b[k][j])%MOD;
            }
     for(int i=1;i<=2;i++)
    {
        for(int j=1;j<=2;j++)
        {
            a[i][j]=c[i][j];
        }
    }
}
void solve()
{
    sol[1][1]=sol[2][2]=1;
    put[1][1]=n;
    put[2][2]=1;


    while(p)
    {
        if(p % 2)
        {
            pwr(sol,put);
        }

        pwr(put,put);

        p = p/ 2;
    }
    fout<<(sol[1][1]%MOD)<<" ";
}
int main()
{
    fin>>n>>p;
    solve();
   return 0;
}