Cod sursa(job #145431)

Utilizator tudalexTudorica Constantin Alexandru tudalex Data 28 februarie 2008 20:13:55
Problema Ridicare la putere in timp logaritmic Scor Ascuns
Compilator cpp Status done
Runda Marime 1.34 kb
#include <stdio.h>
#include <string.h>

const int n_max = 10001; // Definim numarul maxim de cifre al numerelor

int a[n_max], // In vectorul a vom tine N^(2^k)
    sol[n_max]; // In vectorul sol vom tine solutia

void make_long_num(long long x, int A[]) // Functia de creere a unui numar mare dintr-un numar mic 
{
	while ( x > 0)
	{
		A[++A[0]] = x%10;
		x/=10;
	}
}

void mul(int A[], int B[]) // Functia de inmultire a doua numere mari
{
	int i, j, t, C[n_max];
	memset(C, 0, sizeof(C));
	for (i = 1; i <= A[0]; i++)
	{
		for (t=0, j=1; j <= B[0] || t; j++, t/=10)
			C[i+j-1]=(t+=C[i+j-1]+A[i]*B[j])%10;
		if (i + j - 2 > C[0]) C[0] = i + j - 2;
	}
	memcpy(A, C, sizeof(C));
}

void print(int A[])
{
	for (int i = A[0]; i > 0; -- i)
		printf("%d", A[i]);
	printf("\n");
}

int main()
{
	int i;
	long long n, p;
	freopen("lgput.in","r",stdin);
	freopen("lgput.out","w",stdout);
	scanf("%lld %lld", &n, &p);
	make_long_num(n, a); // Il scriem pe n ca numar mare in variabila a 
	sol[0] = 1;
	sol[1] = 1;
	for (i = 0; (1<<i) <= p; ++ i)  // Luam toti biti lui p la rand
	{
		if ( ((1<<i) & p) > 0) // Daca bitul i din p este 1 atunci adaugam n^(2^i) la solutie 
			mul(sol, a);
		if ( (1<<(i+1)) <= p) // Verificam sa nu inmultim degeaba pentru ca am putea sa iesim din vector
				mul(a,a); // Inmultim a cu a ca sa obtinem n^(2^(i+1))
	}
	print(sol); // Afisam solutia
}