BZOJ3142 [Hnoi2013]数列 (组合数学)

                                   数列  HYSBZ - 3142

小T最近在学着买股票，他得到内部消息：F公司的股票将会疯涨。股票每天的价格已知是正整数，并且由于客观上的原因，最多只能为N。在疯涨的K天中小T观察到：除第一天外每天的股价都比前一天高，且高出的价格（即当天的股价与前一天的股价之差）不会超过M，M为正整数。并且这些参数满足M(K-1)<N。
小T忘记了这K天每天的具体股价了，他现在想知道这K天的股价有多少种可能

Input

只有一行用空格隔开的四个数：N、K、M、P。对P的说明参见后面“输出格式”中对P的解释。
输入保证20%的数据M,N,K,P≤20000，保证100%的数据M,K,P≤109，N≤1018 。

Output

仅包含一个数，表示这K天的股价的可能种数对于P的模值。【输入输出样例】

Sample Input

7 3 2 997

Sample Output

16
【样例解释】
输出样例的16表示输入样例的股价有16种可能：
{1，2，3}，{1，2，4}，{1，3，4}，{1，3，5}， {2，3，4}，{2，3，5}，{2，4，5}，{2，4，6}， {3，4，5}，{3，4，6}，{3，5，6}，{3，5，7}，{4，5，6}，{4，5，7}，{4，6，7}，{5，6，7}

Hint

   

分析：  设a[i]为两个数之间的差值。那么就有m^(k-1)个这样的数列。

     然后我们还可以知道当确定一个数列之后，可以得到 s= n-（a[1]+a[2]+...+a[k-1]）种情况（其实就是初始值不同情况不一样）

   那么ans=(m^k-1)*s

    又因为 m个数每个数出现的概率是一样的 （题目保证 m*（k-1）<n）所以 （a[1]+a[2]+...+a[k-1]）=(k-1)*m*(m+1)/(2*m)

   那么就可以得出答案了

参考http://www.cnblogs.com/jianglangcaijin/archive/2013/08/13/3254314.html

#include<stdio.h>
#include<string.h>
long long n,k,m,p;
long long f(long long a,long long b)
{
	long long c=1;
	while(b)
	{
		if(b%2)
		{
			c*=a;
			c%=p;
		}
		a*=a;
		a%=p;
		b/=2;
	}
	return c;
}
int main()
{
	while(scanf("%lld%lld%lld%lld",&n,&k,&m,&p)==4)
	{
		if(k==1)
		printf("%lld\n",n%p);
		else
		{
			long long ans=f(m,k-2)*((((n%p)*m)%p)-((((m*(m+1)/2)%p)*(k-1))%p));
			ans%=p;
    		if(ans<0)
       		ans+=p;
			printf("%lld\n",ans%p);
		}
	}
}