BZOJ 3142: [Hnoi2013]数列

还算是比较轻松愉快的一道题

假设已知每天股价之差的序列，那么给定第一天的股价，就能知道每天的股价了

那么要让最后一天的股价不超过N，则第一天的股价<=N-q1-q2-....-qk-1

所以一个序列对答案的贡献也就出来了

那么求个和就好了，大概是N*M^(K-1)-M*(M+1)/2*M^(K-2)*(K-1)

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
ll p;
ll qpow(ll a,ll b){
	ll ans=1;
	for(;b;b>>=1,a=a*a%p)if(b&1)ans=ans*a%p;
	return ans;
}
int main(){
	//freopen("a.in","r",stdin);
	ll n,k,m;
	scanf("%lld%lld%lld%lld",&n,&k,&m,&p);
	ll p1=n%p*qpow(m,k-1)%p;
	ll p2=m*(m+1)/2%p*qpow(m,k-2)%p*(k-1)%p;
	ll ans=((p1-p2)%p+p)%p;
	printf("%lld\n",ans);
	return 0;
}