Cod sursa(job #424392)

Utilizator doru.nituNitu Doru Constantin doru.nitu Data 24 martie 2010 20:21:27
Problema Suma divizorilor Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.46 kb
#include<stdio.h>
#define mod 9901
#include<bitset>
#define ll long long 

using namespace std;

int  pr[1000000],t;

struct nod
{
	ll nr,p;
}d[1000000];

bitset<1000100> c;

ll put(ll x)
{
	return x*x;
}

ll invmod(ll a,ll b)
{
	if(b==1) return a%mod;
	else if(b%2==1) return ( (put( invmod(a, (b-1)/2 ))) %mod *a%mod)%mod;
	return (put(invmod(a,b/2)))%mod;
}

void prelucreaza()
{
	ll n=0,b=0;
	int i=0,nd=1,sd=1;
	
	scanf("%lld %lld",&n,&b);
	
	if(b==0) 
	{
		printf("%d\n",sd);
		fclose(stdin);
		fclose(stdout);

	}
	else
	{		
		for(i=1;pr[i]*pr[i]<=n;++i)
		{
			int nr=0;
			while(n%pr[i]==0)
			{
				n/=pr[i];
				nr++;
			}
			if(nr)
			{
				if(pr[i]%mod==1)
				{
					sd*=((ll)(nr*b+1)%mod);
					continue;
				}
				nd=invmod(pr[i],nr*b+1);
				nd--;
				if(nd<0) nd+=mod;
				sd*=((nd)%mod);
				sd%=mod;
				sd*=invmod(pr[i]-1,mod-2);
				sd%=mod;
			}
		}
		if(n>1)
		{
			if(n%mod==1)
			{
				sd*=((ll)(b+1)%mod);
			}
			else
			{
			nd=invmod(n,b+1);
			nd--;
			if(nd<0) nd+=mod;
			sd*=((nd)%mod);
			sd%=mod;
			sd*=invmod(n-1,mod-2);
			sd%=mod;
			}
		}
		sd%=mod;
		
		printf("%d\n",sd);
	}
	
}

int main()
{
	freopen("sumdiv.in","r",stdin);
	freopen("sumdiv.out","w",stdout);
	
	for(int i=2;i<=1000000;++i)
		if(!c[i])
		{
			pr[++pr[0]]=i;
			for(int j=i+i;j<=1000000;j+=i)c[j]=1;
		}
	
	prelucreaza();
	
	fclose(stdin);
	fclose(stdout);
	
	return 0;
}