[HNOI2013]数列（牛客竞赛）

[HNOI2013]数列

题目描述:

链接: [HNOI2013]数列
来源：牛客网

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

输入描述:

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

输出描述:

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

示例1：

输入：

7 3 2 997

输出：

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}

设每种可能的情况为{a[1],a[2]……,a[k]}，定义差分数组b，使b[i]=a[i+1]-a[i];得到一种{b[1],b[2]……b[k-1]}，其中1<=b[i]<=m所以这种b数组一共有m^(k-1)种。
当我们确定了一种b时，以这种b为差分的a数组就一共有n-b[k-1]-b[k-2]……-b[1](第一天的股价)种。
所以全部情况就为m^(k-1)*Σ(n-b[k-1]-b[k-2]……-b[1])，即为n*m^(k-1)减去所有出现的b[i]。
将n提出来为n*m^(k-1)由于每种情况b数组都不同，不能提出，继续计算，由(k-1)*m^(k-1)得到b数组会出现多少个数，由于1<=b[i]<=m，在数组b中，所有的数字出现的概率是相等的，所以每一种数字出现的次数为(k-1)*m^(k-2)，所以出现的所有的数字的和为(1+2+3……+m)*(k-1)*m^(k-2)=(k-1)*m^(k-2)*(1+m)*m/2
这道题取模也是关键，取不好模一样过不了，实在取不好的时候让每一次运算后都取模，最后所有运算完成后再取模。
大一刚接触C++学习算法两个月，写这篇文章也是希望自己慢慢总结经验，提升自己，如有描述不准确，大佬勿喷。

代码：

#include <iostream>
using namespace std;
typedef long long ll;
ll n, m, k, p, ans;

ll pows(ll x, ll y)//快速幂
{
    ll sum = 1;
    while (y)
    {
        if (y & 1)//位运算，等价于y%2==1;
            sum = (sum * x) % p;
        x = (x * x) % p;
        y >>= 1;//位运算，等价于y/=2；
    }
    return sum;
}

int main()
{
    std::ios::sync_with_stdio(false);//让cin cout更快，不写也没事，但写了就不可以使用scanf和printf。
    cin >> n >> k >> m >> p;
    ans = (pows(m % p, k - 1) * (n % p)) % p - ((pows(m % p, k - 2) * (k - 1) % p) * (((m + 1) * m / 2) % p)) % p;
    cout << (ans + p) % p;//防止ans减成负数得到错误输出
    return 0;
}