正題

題目鏈接:http://www.51nod.com/Challenge/Problem.html#problemId=1229

---

題目大意

給出$n,k,r$求
$$\sum _{i=1}^n{i}^k{r}^i$$

$1\le T\le 20,1\le n,r\le 10^{18},1\le k\le 2000$

---

解題思路

如此明顯的式子直接開推
$$S_k=\sum _{i=1}^n{i}^k{r}^i,r{S}_k=\sum _{i=2}^{n+1}(i?1{)}^k{r}^i$$
$$(r?1)S_k=n^k{r}^{n+1}?r+\sum _{i=2}^n\left((i?1{)}^k?i^k\right)r^i$$
二項式展開$(i?1{)}^k$
$$(r?1)S_k=n^k{r}^{n+1}?r+\sum _{i=2}^n\sum _{j=0}^{k?1}(?1{)}^{k?j}(\genfrac{}{}{0ex}{}{k}{j})r^i$$
然后把$j$提到前面去
$$(r?1)S_k=n^k{r}^{n+1}?r+\sum _{j=0}^{k?1}(?1{)}^{i?j}(\genfrac{}{}{0ex}{}{k}{j})\sum _{i=2}^n{r}^i$$
$$?S_k=\frac{n^k{r}^{n+1}?r+{\sum }_{j=0}^{k?1}(?1{)}^{k?j}(\genfrac{}{}{0ex}{}{k}{j})(S_j?r)}{r?1}$$

這樣$S_k$就可以$O(k^2)$遞推了。

當然當$r=1$的時候，不能再使用這個公式，此時${\sum }_{i=1}^n{i}^k$是很經典的問題，直接拉格朗日插值插出一個$k+1$次多項式即可。

此題到這里就圓滿結束了，但是以直覺判斷上面那個式子可以卷積，拆開組合數然后瘋狂跳步一下就是
$$(r?1)\frac{S_k?r}{k!}=n^k{r}_{n+1}?r?(r?1)\frac{r}{k!}+\sum _{j=0}^{k?1}\frac{(?1{)}^{k?j}}{(k?j)!}\frac{S_j?r}{j!}$$
設
$$H(x)=\sum _{i=0}^{\infty }(n^i{r}_{n+1}?r?(r?1)\frac{r}{i!})x^i$$
$$G(x)=\sum _{i=0}^{\infty }\frac{S_i?r}{i!},F(x)=\sum _{i=1}^{\infty }\frac{(?1{)}^i}{i!}$$
那么有
$$(r?1)G(x)=H(x)+F(x)G(x)$$
$$?G(x)=\frac{H(x)}{r?1?F(x)}$$

然后多項式求逆即可，時間復雜度$O(k\log k)$，雖然這題的模數不能用，但是可以順便解決掉序列求和V5。

但是最近寫的多項式求逆有點多，咕了，所以上面的式子如果有錯我也沒辦法（（（、

---

code

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=2100,P=1e9+7; ll T,n,k,r,inv[N],fac[N],pre[N],suf[N],s[N]; ll power(ll x,ll b){ll ans=1;b%=P-1;x%=P;while(b){if(b&1)ans=ans*x%P;x=x*x%P;b>>=1;}return ans; } ll C(ll n,ll m) {return fac[n]*inv[m]%P*inv[n-m]%P;} signed main() {scanf("%lld",&T);inv[0]=fac[0]=inv[1]=1;for(ll i=2;i<N;i++)inv[i]=P-inv[P%i]*(P/i)%P;for(ll i=1;i<N;i++)fac[i]=fac[i-1]*i%P,inv[i]=inv[i-1]*inv[i]%P;while(T--){scanf("%lld%lld%lld",&n,&k,&r);r%=P;if(r==1){ll ans=0;k+=2;n%=P;pre[0]=suf[k+1]=1;for(ll i=1;i<=k;i++)pre[i]=pre[i-1]*(n-i)%P;for(ll i=k;i>=1;i--)suf[i]=suf[i+1]*(n-i)%P;for(ll i=1,p=0;i<=k;i++){p=(p+power(i,k-2))%P;(ans+=p*pre[i-1]%P*suf[i+1]%P*inv[i-1]%P*(((k-i)&1)?P-inv[k-i]:inv[k-i])%P)%=P;}printf("%lld\n",(ans+P)%P);}else{ll z=power(r,n+1),invr=power(r-1,P-2);s[0]=(z-r+P)*invr%P;n%=P;for(ll i=1,pw=n;i<=k;i++,pw=pw*n%P){s[i]=(z*pw-r+P)%P;s[i-1]=(s[i-1]-r+P)%P;for(ll j=0;j<i;j++)(s[i]+=(((i-j)&1)?(P-1):(1))*s[j]%P*C(i,j)%P)%=P;s[i]=s[i]*invr%P;}printf("%lld\n",(s[k]+P)%P);}}return 0; }

總結

以上是生活随笔為你收集整理的51nod1229-序列求和V2【数学,拉格朗日插值】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。