HNOI2013 数列

题目描述

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

输入

输入只有一行用空格隔开的四个数：N、K、M、P。
对P的说明参见后面“输出格式”中对P的解释。

输出

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

样例

input：7 3 2 997

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}。

题解

我们设第$i$天的股价为$pri_i$,相邻两项的差值为$A_i=pri_{i+1}-pri_i$由于第一天的股价是不确定的，则此时的答案为$n-\sum A$.而最终的答案为$\sum (n-\sum A)$由于A中的每个数都有m种选择，将n拆出来就是$n* m$^(k-1).而后面的$\sum A$拆出来就变成了$\frac{m*(m+1)}{2}*m$^(k-2)*(k-1).

参考程序

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
using namespace std;

ll n,k,m,mo;

ll mul(ll x,ll y){
    ll ret=1;
    while(y) {
        if (y % 2==1) ret=ret*x % mo;
        x=x*x % mo;
        y=y/2;
    }
    return ret;
}

int main(){
    scanf("%lld%lld%lld%lld",&n,&k,&m,&mo);
    if (k==1) {
        printf("%lld",n);
        return 0;
    }
    ll tmp=mul(m,k-1);
    ll ans=tmp*(n % mo) % mo;
    ans=ans-m*(m+1)/2 % mo*mul(m,k-2) % mo*(k-1) % mo;
    ans=(ans % mo+mo) % mo;
    printf("%lld",ans);
    return 0;
}