jzoj 1344. 方程的解 (Standard IO)&&洛谷1771

Description
　　佳佳碰到了一个难题，请你来帮忙解决。 对于不定方程a1+a2+……+ak-1+ak=g(x)，其中k>=2且k∈N*，x是正整数，g(x)=x^x mod 1000（即x^x除以1000的余数），x，k是给定的数。我们要求的是这个不定方程的正整数解组数。 　　
　　举例来说，当k=3,x=2时，分别为(a1,a2,a3)=(2,1,1),(1,2,1),(1,1,2).
Input
　　输入文件有且只有一行，为用空格隔开的两个正整数，依次为k，x。
Output
　　输出文件有且只有一行，为方程的正整数解组数。
Sample Input
3 2
Sample Output
3
Data Constraint
【数据范围】
对于40%的数据，ans<=10¹⁶;
对于100%的数据，k<=100,x<=2³¹-1,k<=g(x)。

//written by zzy

题目大意:

求方程sum(a1…ak)=x^x%1000的解的个数

题解：

对于g(x)用快速幂求
考虑解得个数
令g(x)=n，将n拆成n个1，因为ai>0，所以即等价于n个球中放入k个盘里，盘不能为空。
ans=C(k-1,n-1)（组合数）
隔板法即证明:
n个球中放入k个盘里，盘不能为空。
放法为C(k-1,n-1)
例：n=4,k=2
X X X X(球)
1 2 3(板)
即在三个板子里挑1个插入
方案数为C(1,3)
(参见组合数的定义C(m,n)等于从1…n中选出m个的数的方案数 等于n!/(m!(n-m)!) )
那么考虑一般情况
即在n-1个板子里挑k-1个插入
方案数为C(k-1,n-1)
证毕。
至于如何求C
将n!和m!,(n-m)!分别分解质因数
幂相减，跑遍高精乘即可。

#include<bits/stdc++.h>
#define Mod 1000
#define N 1005
#define M 1000

using namespace std;

int num,k,x,i,j,n;
int a[N],p[N],ans[M];
bool b[N];

long long g(long long s)
{
   
	if (s==1) return x;
	long long ss=g(s/2)%Mod;
	if (s&1)