POJ 1091&2014携程第一场1001 聪明的猴子

题目链接：跳蚤

Description

Z城市居住着很多只跳蚤。在Z城市周六生活频道有一个娱乐节目。一只跳蚤将被请上一个高空钢丝的正中央。钢丝很长，可以看作是无限长。节目主持人会给该跳蚤发一张卡片。卡片上写有N+1个自然数。其中最后一个是M，而前N个数都不超过M，卡片上允许有相同的数字。跳蚤每次可以从卡片上任意选择一个自然数S，然后向左，或向右跳S个单位长度。而他最终的任务是跳到距离他左边一个单位长度的地方，并捡起位于那里的礼物。 
比如当N=2，M=18时，持有卡片(10, 15, 18)的跳蚤，就可以完成任务：他可以先向左跳10个单位长度，然后再连向左跳3次，每次15个单位长度，最后再向右连跳3次，每次18个单位长度。而持有卡片(12, 15, 18)的跳蚤，则怎么也不可能跳到距他左边一个单位长度的地方。 
当确定N和M后，显然一共有M^N张不同的卡片。现在的问题是，在这所有的卡片中，有多少张可以完成任务。 

Input

两个整数N和M(N <= 15 , M <= 100000000)。

Output

可以完成任务的卡片数。

Sample Input

2 4

Sample Output

12

Hint

这12张卡片分别是： 
(1, 1, 4), (1, 2, 4), (1, 3, 4), (1, 4, 4), (2, 1, 4), (2, 3, 4), 
(3, 1, 4), (3, 2, 4), (3, 3, 4), (3, 4, 4), (4, 1, 4), (4, 3, 4) 

Source

HNOI 2001

本质：

数论、组合数学

思路：

这个做法基于一个前提：当卡片上存在两个数字互质时，该卡片可以完成任务

证明：由于每个数字可以使用多次，那么对于卡片上面的数字a和b，我们认为a-b也可以使用（每次使用跳a个单位再向相反方向跳b个单位即可），那么a-b也可以视为是在卡片上的数字，考虑辗转相除法的过程，每次都是将较大的数字减去数个较小的数字，从而得到更小的数字，那么只要我们对互质的两个数字a、b模拟辗转相除法即可得到1，证毕。

那么题目就转化成了：求m^n张卡片中，有多少张是存在两个数互质的

我们可以用总数m^n减去所有存在非1公因子的卡片得到答案

考虑枚举m的所有质因子，然后计算以该质因子为公约数的所有组合，但是这样来会造成计算重复，比如m=6，n=2的时候，在质因子为2的时候（6,6）被计算了一次，在质因子为3的时候（6,6）又被计算了一次，

这里参考了别人博客的做法： 新浪博客

首先枚举出所有非1因子，然后对于每个因子：

如果这个因子是奇数个质数的乘积（并且每个质数最多用了1次），那么加上它的所有组合

如果这个因子是偶数个质数的成绩（并且每个质数最多用了1次），那么减去它的所有组合

如果这个因子存在一个质数被用了多次，那么忽略

这样做可以保证每种情况都被计算了一次，证明如下：

运用组合数学里面的一个定理来证明：

奇数项之和等于偶数项之和，即：

C(n,0)-C(n,1)+C(n,2)-C(n,3)+C(n,4)-C(n,5)+.......C(n,n)=0

那么有 C(n,1)-C(n,2)+C(n,3)-C(n,4)+C(n,5)-.......C(n,n)=1

假设有一张卡片上的公约数为k，并且k有x个不同的质因子，那么：

在计算一个质因子的所有组合的时候，这张卡片被计算了C(x,1)次

在计算两个质因子的所有组合的时候，这张卡片被计算了C(x,2)次

在计算三个质因子的所有组合的时候，这张卡片被计算了C(x,3)次
在计算四个质因子的所有组合的时候，这张卡片被计算了C(x,4)次

...

...

那么这张卡片总共被计算了C(x,1)-C(x,2)+C(x,3)-C(x,4)+C(x,5)-.......C(x,x)=1次

证毕。

基于上面几点，就可以做了，不过要注意的是，POJ数据范围很大，要用上高精度

PS：POJ这数据水得可以。。。一堆人用__int64过掉了，不过偷懒是不好的T皿T

#include<stdio.h>
#include<string.h>
#define p 10000
__int64 fact[20005],up;
bool sai[10005];
__int64 prim[2000],num;
struct bignum
{
	__int64 n[105],l;
};
__int64 pd(__int64 x)
{
	__int64 i,k=0;
	for(i=1;i<=num&&prim[i]<=x;i++)
		if(x%prim[i]==0)
		{
			k++;
			x/=prim[i];
			if(x%prim[i]==0)
				return -1;
	//返回-1代表这个因子存在一个质数被用了多次，不用计算
		}
	if(x!=1)
		k++;//最后分解完是一个大质数，那么要统计多一个
	return k&1;
}
bignum pow(__int64 x,__int64 y)
{
	bignum z;
	memset(z.n,0,sizeof(z.n));
	z.l=1;
	z.n[1]=1;
	int i;
	while(y--)
	{
		for(i=1;i<=z.l;i++)
			z.n[i]*=x;
		for(i=1;i<=z.l;i++)
		{
			z.n[i+1]+=z.n[i]/p;
			z.n[i]%=p;
			if(i+1>z.l&&z.n[i+1]!=0)
				z.l++;
		}
	}
	return z;
}
bignum add(bignum x,bignum y)
{
	bignum z;
	z.l=x.l>y.l?x.l:y.l;
	int i;
	memset(z.n,0,sizeof(z.n));
	for(i=1;i<=z.l;i++)
	{
		z.n[i]+=x.n[i]+y.n[i];
		if(z.n[i]>p)
		{
			z.n[i]-=p;
			z.n[i+1]++;
		}
	}
	if(z.n[z.l+1]!=0)
		z.l++;
	return z;
}
bignum sub(bignum x,bignum y)
{
	__int64 i;
	for(i=1;i<=y.l;i++)
	{
		x.n[i]-=y.n[i];
		if(x.n[i]<0)
		{
			x.n[i]+=p;
			x.n[i+1]--;
		}
	}
	for(i=y.l+1;i<=x.l;i++)
		if(x.n[i]<0)
		{
			x.n[i]+=p;
			x.n[i+1]--;
		}
	if(x.n[x.l]==0)
		x.l--;
	return x;
}
int main()
{
	__int64 i,j,m,n,t;
	bignum ans;
	memset(sai,true,sizeof(sai));
	num=0;
	for(i=2;i<=10000;i++)
		if(sai[i])
		{
			prim[++num]=i;
			for(j=i*i;j<=10000;j+=i)
				sai[j]=false;
		}
	while(scanf("%I64d%I64d",&n,&m)!=EOF)
	{
		
		up=0;
		for(i=2;i*i<m;i++)
			if(m%i==0)
			{
				fact[++up]=i;
				fact[++up]=m/i;
			}
		if(i*i==m)
			fact[++up]=i;
		fact[++up]=m;
		ans=pow(m,n);
		for(i=1;i<=up;i++)
		{
			t=pd(fact[i]);
			if(t==-1)
				continue;
			if(t)
				ans=sub(ans,pow(m/fact[i],n));
			else
				ans=add(ans,pow(m/fact[i],n));
		}
		printf("%I64d",ans.n[ans.l]);
		for(i=ans.l-1;i>=1;i--)
			printf("%04I64d",ans.n[i]);
		puts("");
	}
	return 0;
}