题目描述:求

\[\sum_{i=1}^ni^kr^i
\]

对某个质数取模。\(T\)组数据。

数据范围:\(n,r\le 10^{18},\sum k\le 2.56\times 10^6\)


数论神题。。。

我们设\(S_k(n)=\sum_{i=1}^{n-1}i^kr^i\),通过推通项公式,你会发现这个结论:

\[S_k(n)=r^nF_k(n)-F_k(0)
\]

其中\(F_k(n)\)是一个关于\(n\)且次数不高于\(k\)的多项式。

如何证明呢?首先你可以通过错位相减法得到一个递推公式,然后就可以数学归纳法了。

然后我们就是要求出\(F_k(0),F_k(1),F_k(2),\ldots,F_k(k+1)\),这样就可以通过拉格朗日插值求出\(F_k(n)\)。

我们注意到\(S_k(n)-S_k(n-1)=r^nF_k(n)-r^{n-1}F_k(n-1)=r^{n-1}(n-1)^k\),所以\(F_k(n)=\dfrac{F_k(n-1)+(n-1)^k}{r}\)。

然后我们知道,\(k\)次多项式的\(k+1\)阶差分为\(0\),即

\[\sum_{i=0}^{k+1}(-1)^i\binom{k+1}{i}F_k(k+1-i)=0
\]

于是可以设\(F_k(0)=x\),然后不停代入,然后用这个式子解方程。具体实现可以写一个二项式binom来做。

注意特判一些特殊情况。时间复杂度\(O(k)\)。

code
#include<bits/stdc++.h>
#define Rint register int
using namespace std;
typedef long long LL;
const int N = 1000012, mod = 1e9 + 7;
inline int kasumi(int a, int b){
int res = 1;
while(b){
if(b & 1) res = (LL) res * a % mod;
a = (LL) a * a % mod; b >>= 1;
}
return res;
}
inline void upd(int &a, int b){a += b; if(a >= mod) a -= mod;}
inline int add(int a, int b){return (a + b >= mod) ? (a + b - mod) : (a + b);}
inline int sub(int a, int b){return (a < b) ? (a + mod - b) : (a - b);}
int T, k, fac[N], inv[N];
LL n, r;
struct binom {
int a, b;
inline binom(int _a = 0, int _b = 0): a(_a), b(_b){}
inline binom operator + (const binom &o) const {return binom(add(a, o.a), add(b, o.b));}
inline binom operator - (const binom &o) const {return binom(sub(a, o.a), sub(b, o.b));}
inline binom operator * (int v) const {return binom((LL) a * v % mod, (LL) b * v % mod);}
} S[N];
inline int lagrange(int *A, int k, LL n){
static int pre[N], suf[N];
pre[0] = suf[k + 1] = 1;
for(Rint i = 1;i <= k;i ++) pre[i] = (n - i + 1) % mod * pre[i - 1] % mod;
for(Rint i = k;i;i --) suf[i] = (n - i) % mod * suf[i + 1] % mod;
int ans = 0;
for(Rint i = 0;i <= k;i ++){
int tmp = (LL) inv[i] * inv[k - i] % mod * pre[i] % mod * suf[i + 1] % mod;
if((k - i) & 1) tmp = mod - tmp;
upd(ans, (LL) A[i] * tmp % mod);
}
return ans;
}
int pri[N], tot, po[N], minp[N];
bool notp[N];
inline void init(int n){
notp[0] = notp[1] = 1;
for(Rint i = 2;i <= n;i ++){
if(!notp[i]) pri[++ tot] = i;
for(Rint j = 1;j <= tot && i * pri[j] <= n;j ++){
notp[i * pri[j]] = 1; minp[i * pri[j]] = pri[j];
if(!(i % pri[j])) break;
}
}
fac[0] = 1;
for(Rint i = 1;i <= n;i ++) fac[i] = (LL) fac[i - 1] * i % mod;
inv[n] = kasumi(fac[n], mod - 2);
for(Rint i = n;i;i --) inv[i - 1] = (LL) inv[i] * i % mod;
}
inline void calc(int k){
po[0] = 0; po[1] = 1;
for(Rint i = 2;i <= k + 1;i ++)
if(notp[i]) po[i] = (LL) po[minp[i]] * po[i / minp[i]] % mod;
else po[i] = kasumi(i, k);
}
inline int C(int n, int m){if(n < 0 || m < 0 || n < m) return 0; return (LL) fac[n] * inv[m] % mod * inv[n - m] % mod;}
inline int solve(LL n, int k, int r){
static int F[N];
int invr = kasumi(r, mod - 2);
if(!r) return 0;
memset(F, 0, (k + 2) << 2);
calc(k);
if(r == 1){
for(Rint i = 1;i <= k + 1;i ++) F[i] = add(F[i - 1], po[i]);
return n <= k + 1 ? F[n] : lagrange(F, k + 1, n);
}
if(!n) return 0;
if(!k) return (LL) sub(kasumi(r, n + 1), r) * kasumi(r - 1, mod - 2) % mod;
S[0] = binom(1, 0);
for(Rint i = 1;i <= k + 1;i ++)
S[i] = (S[i - 1] + binom(0, po[i - 1])) * invr;
binom qwq;
for(Rint i = 0;i <= k + 1;i ++)
if(i & 1) qwq = qwq + S[k + 1 - i] * C(k + 1, i);
else qwq = qwq - S[k + 1 - i] * C(k + 1, i);
F[0] = (LL) (mod - qwq.b) * kasumi(qwq.a, mod - 2) % mod;
for(Rint i = 1;i <= k + 1;i ++) F[i] = (LL) add(F[i - 1], po[i - 1]) * invr % mod;
int ans = lagrange(F, k + 1, n + 1);
return (LL) sub((LL) ans * kasumi(r, (n + 1) % (mod - 1)) % mod, F[0]);
}
int main(){
scanf("%d", &T); init(1000010);
while(T --){
scanf("%lld%lld%d", &n, &r, &k);
printf("%d\n", solve(n, k, r % mod));
}
}

SP19997 MOON2 - Moon Safari (Hard) 【数论,多项式】的更多相关文章

  1. python机器学习卷积神经网络(CNN)

    卷积神经网络(CNN) 关注公众号"轻松学编程"了解更多. 一.简介 ​ 卷积神经网络(Convolutional Neural Network,CNN)是一种前馈神经网络,它的人 ...

  2. 多项式 之 快速傅里叶变换(FFT)/数论变换(NTT)/常用套路【入门】

    原文链接https://www.cnblogs.com/zhouzhendong/p/Fast-Fourier-Transform.html 多项式 之 快速傅里叶变换(FFT)/数论变换(NTT)/ ...

  3. Algorithm: 多项式乘法 Polynomial Multiplication: 快速傅里叶变换 FFT / 快速数论变换 NTT

    Intro: 本篇博客将会从朴素乘法讲起,经过分治乘法,到达FFT和NTT 旨在能够让读者(也让自己)充分理解其思想 模板题入口:洛谷 P3803 [模板]多项式乘法(FFT) 朴素乘法 约定:两个多 ...

  4. 数论 - Moon Game

    Fat brother and Maze are playing a kind of special (hentai) game in the clearly blue sky which we ca ...

  5. UOJ269【清华集训2016】如何优雅地求和【数论,多项式】

    题目描述:求 $$\sum_{k=0}^nf(k)\binom{n}{k}x^k(1-x)^{n-k}$$ 输入$n$,$f(x)$的次数上界$m$,$x$,$f(0,1,\ldots,m)$,对$9 ...

  6. 模板 - 数学 - 多项式 - 快速数论变换/NTT

    Huffman分治的NTT,常数一般.使用的时候把多项式的系数们放进vector里面,然后调用solve就可以得到它们的乘积.注意这里默认最大长度是1e6,可能需要改变. #include<bi ...

  7. 多项式乘法(FFT)模板 && 快速数论变换(NTT)

    具体步骤: 1.补0:在两个多项式最前面补0,得到两个 $2n$ 次多项式,设系数向量分别为 $v_1$ 和 $v_2$. 2.求值:用FFT计算 $f_1 = DFT(v_1)$ 和 $f_2=DF ...

  8. NOI 2019 省选模拟赛 T1【JZOJ6082】 染色问题(color) (多项式,数论优化)

    题面 一根长为 n 的无色纸条,每个位置依次编号为 1,2,3,-,n ,m 次操作,第 i 次操作把纸条的一段区间 [l,r] (l <= r , l,r ∈ {1,2,3,-,n})涂成颜色 ...

  9. JZYZOJ 2041 快速数论变换 NTT 多项式

    http://172.20.6.3/Problem_Show.asp?id=2041 https://blog.csdn.net/ggn_2015/article/details/68922404 代 ...

随机推荐

  1. 用StatSVN统计svn项目中每人代码提交量

    用StatSVN统计SVN服务器项目的代码量 下载并安装SVN统计工具StatSVN 1)下载地址:sourceforge.net/projects/statsvn/ 2)解压压缩包,到一个目录,如D ...

  2. js 时间常用处理方法

    众所周知,JavaScript核心包含Data()构造函数,用来创建表示时间和日期的对象. 今天主要跟大家梳理一下,常用的时间.日期处理方法,方便大家使用和理解 格式化时间 老生常谈,大概会这么写 1 ...

  3. JavaScript学习笔记(6月份)

    由于笔记比较杂,本身学习程度并不理想,所以暂时没有整理这些繁杂的笔记. ps:博客园markdown用起来和看起来都舒服太多了,这才是我了解的那个markdown,又回来了! 笔记 DOM对象 doc ...

  4. Oracle数据库之初识部分知识

    Oracle是比MySql更为严格的数据库.使用时需要更加严谨. 一.安装注意事项: 1.选择地址时需要注意好不能有汉字,以免造成安装的时候注册表显示监听失败: 2.可视化窗口PLSQL编辑器(相当于 ...

  5. 如何在SAP Cloud Platform上进行第一个integration flow开发

    登录SAP Cloud Platform integration tenant,点击Edit图标: 创建一个新的Content package: 保存content package后,点击artifa ...

  6. SQL SERVER升级2017

    SQL SERVER升级2017 摘要 本文只介绍了SQL SERVER升级到2017(在简单环境下),分为开始升级前的检查事项,升级操作步骤,升级后对新实例的配置. 检查事项 1.检查当前版本是否可 ...

  7. jquery获取元素的display属性是不是none?

    1.代码: $(".max_find_wl_info").click(function(){ if($('.max_wuliu').css('display') === 'none ...

  8. beta版本——第三次冲刺

    第三次冲刺 (1)SCRUM部分☁️ 成员描述: 姓名 李星晨 完成了哪个任务 认证学校那一栏增加检测机制的ui设计 花了多少时间 1h 还剩余多少时间 1h 遇到什么困难 没有困难 这两天解决的进度 ...

  9. Codeforces C. Jzzhu and Cities(dijkstra最短路)

    题目描述: Jzzhu and Cities time limit per test 2 seconds memory limit per test 256 megabytes input stand ...

  10. 51nod 2502 最多分成多少块

    小b有个长度为n的数组a,她想将这个数组排序. 然而小b很懒,她觉得对整个数组排序太累了,因此她请你将a分成一些块,使得她只需要对每一块分别排序,就能将整个数组排序. 请问你最多能把a分成多少块. 保 ...