BZOJ3142: [Hnoi2013]数列

将原序列差分得到序列$a[1],a[2],a[3]——a[k-1]$，那么这样的一个序列一共有$n-\sum a[i]$种可能，答案就是$\sum (n-\sum a[i])$，差分的序列有$m^{k-1}$种可能，所以将$n$提出来得到

$$n*m^{k-1}-\sum\sum a[i]$$
因为每个数字在差分序列中出现次数是一样的，所有数字出现次数为$(k-1)*m^{k-1}$，所以每个数字出现次数为$(k-1)*m^{k-2}$，所以答案可以化为

$$n*m^{k-1}-\dfrac{(k-1)*m^{k-2}*m*(m+1)}{2}$$

---

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

ll P;
ll pw(ll x,int k)
{
    x%=P; ll r=1,t=x;
    for(;k;k>>=1,t=t*t%P) if(k&1) r=r*t%P;
    return r;
}
ll n,m,k;

int main()
{
    scanf("%lld%lld%lld%lld",&n,&k,&m,&P);
    ll r=((-m*(m+1)/2ll%P*pw(m,k-2)%P*(k-1)%P+n%P*pw(m,k-1)%P)%P+P)%P;
    printf("%lld\n",r);

    return 0;
}