[BZOJ3142][HNOI2013]数列

这题我的公式好像不能化简，于是膜拜了一下题解。
先做差分$A[1]...A[k-1]$，那么满足这个差分的初始票价为的方案数有：
$n-\sum\limits_{i=1}^{k-1}A[i]$，那么总方案有：
$\sum\limits_{所有序列}(n-\sum\limits_{i=1}^{k-1}A[i])$，
序列显然有$m^{(k-1)}$种，把上式拆开得：
$nm^{(k-1)}-\sum\limits_{所有序列}\sum\limits_{i=1}^{k-1}A[i]$
交换$\sum$得：
$nm^{(k-1)}-\sum\limits_{i=1}^{k-1}\sum\limits_{所有序列}A[i]$
$A[i]$可以取1到m，除去i的序列数量有$m^{(k-2)}$个，所以原式等于：
$nm^{(k-1)}-(k-1)m^{(k-2)}\frac{m(m+1)}{2}$，
这种方法避免了我方法里面的讨论，而且运用了算贡献的思想，我的方法则是分段讨论差分和的取值，不如直接把差分设出来。

#include<cstdio>
#include<algorithm>
#include<iostream>
#define LL long long
using namespace std;
LL p;
LL ksm(LL x,LL t)
{
    LL res=1,d=x;
    while(t)
    {
        if(t&1)
            res=(res*d)%p;
        d=(d*d)%p;
        t>>=1;
    }
    return res;
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("series.in","r",stdin);
    freopen("series.out","w",stdout);
#endif
    LL n,m,k;
    cin>>n>>k>>m>>p;
    n%=p;
    LL ans1=(n*ksm(m,k-1))%p,ans2=ksm(m,k-2)*((m*(m+1)/2)%p)%p;
    ans2%=p;
    ans2=(ans2*(k-1))%p;
    cout<<(ans1-ans2+p)%p<<endl;
}