HDU 本原串 (快速幂)

Problem Description

由0和1组成的串中，不能表示为由几个相同的较小的串连接成的串，称为本原串，有多少个长为n（n<=100000000)的本原串？
答案mod2008.
例如，100100不是本原串，因为他是由两个100组成，而1101是本原串。

Input

输入包括多个数据，每个数据一行，包括一个整数n，代表串的长度。

Output

对于每个测试数据，输出一行，代表有多少个符合要求本原串，答案mod2008.

Sample Input

1
2
3
4

Sample Output

2
2
6
12

#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
using namespace std;
map<int ,int >mp;
const int mod=2008;

int powmod(int a,int n)
{
	if(n==0) return 1;
	int x=powmod(a,n/2)%mod;
	long long ans=(long long)x*x%mod;
	if(n%2==1) ans=ans*a%mod;
	return (int) ans;
}

// 关键在求n的约数 ！！！ 
int sol(int n)
{
	if(mp[n]!=0) return mp[n];
	mp[n]=powmod(2,n)-2;
	for(int i=2;i*i<=n;i++){  // 为什么不用i<=n/2， 为了防止超时，求约数（类似于求素数的方法）
		                      // y=n*m，有效的减少,进行了优化。 
		if(n%i==0) {
			mp[n]=(mp[n]-sol(i)+mod)%mod;
			if(n/i!=i) mp[n]=(mp[n]-sol(n/i)+mod)%mod;
		}
		
	}
	return mp[n];
}

int main()
{
	int n;
	while(scanf("%d",&n)!=EOF){
	   mp[0]=0;
       mp[1]=2;
       mp[2]=2;
       if(n==0) {
       	 printf("0\n");
       	 continue;
	   }
	   printf("%d\n",sol(n));
	}
	return 0;
}