【BZOJ3157/3516】国王奇遇记(数论)

题面

BZOJ3157

BZOJ3516

题解

先考虑怎么做\(m\le 100\)的情况、

令\(f(n,k)=\displaystyle \sum_{i=1}^n i^k m^i\),然后推式子:

\[\begin{aligned}
f(n+1,k)&=\sum_{i=1}^{n+1} i^km^i=m+\sum_{i=2}^{n+1}i^km^i\\
&=m+\sum_{i=1}^n (i+1)^km^{i+1}\\
&=m+m\sum_{i=1}^n m^i\sum_{j=0}^k{k\choose j}i^j\\
&=m+m\sum_{j=0}^{k}{k\choose j}\sum_{i=1}^n i^jm^i\\
&=m+m\sum_{j=0}^k {k\choose j}f(n,j)
\end{aligned}\]

这样子可以做到\(O(nm)\)。

考虑继续处理这个式子:

\[\begin{aligned}
f(2n,k)&=\sum_{i=1}^{2n}i^km^i=f(n,k)+m^n\sum_{i=1}^n (i+n)^km^i\\
&=f(n,k)+m^n\sum_{i=1}^n m^i\sum_{j=0}^k {k\choose j}i^jn^{k-j}\\
&=f(n,k)+m^n\sum_{j=0}^k{k\choose j}n^{k-j}\sum_{i=1}^ni^j m^i\\
&=f(n,k)+m^n\sum_{j=0}^k{k\choose j}n^{k-j}f(n,j)
\end{aligned}\]

既然这样子就可以愉快的倍增了。

时间复杂度\(O(m^2\log n)\)

#include<iostream>
using namespace std;
#define MOD 1000000007
int n,m,f[222],C[222][222],pw[222],tmp[222];
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
void Solve(int n)
{
if(n==1){for(int i=0;i<=m;++i)f[i]=m;return;}
Solve(n>>1);int pwm=fpow(m,n>>1);
pw[0]=1;for(int i=1;i<=m;++i)pw[i]=1ll*pw[i-1]*(n>>1)%MOD;
for(int i=0;i<=m;++i)tmp[i]=f[i];
for(int k=0;k<=m;++k)
for(int j=0;j<=k;++j)
tmp[k]=(tmp[k]+1ll*pwm*C[k][j]%MOD*pw[k-j]%MOD*f[j])%MOD;
for(int i=0;i<=m;++i)f[i]=tmp[i];
if(n&1)
{
for(int i=0;i<=m;++i)tmp[i]=m;
for(int k=0;k<=m;++k)
for(int j=0;j<=k;++j)
tmp[k]=(tmp[k]+1ll*m*C[k][j]%MOD*f[j])%MOD;
for(int i=0;i<=m;++i)f[i]=tmp[i];
}
}
int main()
{
cin>>n>>m;
for(int i=0;i<=m;++i)C[i][0]=1;
for(int i=1;i<=m;++i)
for(int j=1;j<=i;++j)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
Solve(n);
cout<<f[m]<<endl;
return 0;
}

然而如果你直接把上面的代码去交加强版就会\(T\)飞(时限\(1s\))

考虑更加优秀的方法。

直接令\(f(k)=\sum_{i=1}^n i^k m^i\)。

然后拿出来强行做个差:

\[\begin{aligned}
mf(k)-f(k)&=\sum_{i=1}^n i^k m^{i+1}-\sum_{i=1}^n i^km^i\\
&=n^km^{n+1}+\sum_{i=1}^{n} (i-1)^km^i-\sum_{i=1}^n i^k m^i\\
&=n^km^{n+1}+\sum_{i=1}^{n}m^i((i-1)^k-i^k)\\
&=n^km^{n+1}+\sum_{i=1}^n m^i \sum_{j=0}^{k-1}{k\choose j}(-1)^{k-j}i^j\\
&=n^km^{n+1}+\sum_{j=0}^{k-1}{k\choose j}(-1)^{k-j}\sum_{i=1}^n i^jm^i\\
&=n^km^{n+1}+\sum_{j=0}^{k-1}{k\choose j}(-1)^{k-j}f(j)\\
\end{aligned}\]

于是就可以做到\(O(m^2)\)了。

注意\(m=1\)的时候需要特判。

#include<iostream>
using namespace std;
#define MOD 1000000007
int n,m,f[1010],C[1010][1010],pw[1010],tmp[1010];
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
int main()
{
cin>>n>>m;
if(m==1){cout<<1ll*n*(n+1)/2%MOD<<endl;return 0;}
int inv=fpow(m-1,MOD-2);
for(int i=0;i<=m;++i)C[i][0]=1;
for(int i=1;i<=m;++i)
for(int j=1;j<=i;++j)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
f[0]=1ll*(fpow(m,n)+MOD-1)*inv%MOD*m%MOD;
for(int k=1,pwm=fpow(m,n+1),pw=n;k<=m;++k,pw=1ll*pw*n%MOD)
{
f[k]=1ll*pw*pwm%MOD;
for(int j=0,d=((k&1)?(MOD-1):1);j<k;++j,d=MOD-d)
f[k]=(f[k]+1ll*C[k][j]*d%MOD*f[j])%MOD;
f[k]=1ll*f[k]*inv%MOD;
}
cout<<f[m]<<endl;
return 0;
}

似乎还要一个更强的版本,然而我不会做QwQ。

【BZOJ3157/3516】国王奇遇记(数论)的更多相关文章

  1. bzoj3157 3516 国王奇遇记

    Description Input 共一行包括两个正整数N和M. Output 共一行为所求表达式的值对10^9+7取模的值. 特判m=1 m≠1时: 设S[u]=sigma(i^u*m^i) m*S ...

  2. BZOJ3157: 国王奇遇记 & 3516: 国王奇遇记加强版

    令\[S_i=\sum_{k=1}^n k^i m^k\]我们有\[\begin{eqnarray*}(m-1)S_i & = & mS_i - S_i \\& = & ...

  3. bzoj 3157 && bzoj 3516 国王奇遇记——推式子

    题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3157 https://www.lydsy.com/JudgeOnline/problem.p ...

  4. bzoj 3157 & bzoj 3516 国王奇遇记 —— 推式子

    题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3157 https://www.lydsy.com/JudgeOnline/problem.p ...

  5. BZOJ3157/BZOJ3516 国王奇遇记(矩阵快速幂/数学)

    由二项式定理,(m+1)k=ΣC(k,i)*mi.由此可以构造矩阵转移,将mi*ik全部塞进去即可,系数即为组合数*m.复杂度O(m3logn),因为大常数喜闻乐见的T掉了. #include< ...

  6. 3157: 国王奇遇记 & 3516: 国王奇遇记加强版 - BZOJ

    果然我数学不行啊,题解君: http://www.cnblogs.com/zhuohan123/p/3726933.html const h=; var fac,facinv,powm,s:..]of ...

  7. 【BZOJ4126】【BZOJ3516】【BZOJ3157】国王奇遇记 线性插值

    题目描述 三倍经验题. 给你\(n,m\),求 \[ \sum_{i=1}^ni^mm^i \] \(n\leq {10}^9,1\leq m\leq 500000\) 题解 当\(m=1\)时\(a ...

  8. BZOJ 3516 国王奇遇记加强版(乱推)

    题意 求\(\sum_{k=1}^{n}k^mm^k (n\leq1e9,m\leq1e3)\) 思路 在<>中有一个方法用来求和,称为摄动法. 我们考虑用摄动法来求这个和式,看能不能得到 ...

  9. 【BZOJ】【3157】&【BZOJ】【3516】国王奇遇记

    数论 题解:http://www.cnblogs.com/zhuohan123/p/3726933.html copy一下推导过程: 令$$S_i=\sum_{k=1}^{n}k^im^k$$ 我们有 ...

随机推荐

  1. [2017BUAA软工助教]第0次作业小结

    BUAA软工第0次作业小结 零.题目 作业链接: This is a hyperlink 一.评分规则 本次作业满分10分: 按时提交有分 一周内补交得0分 超过一周不交或抄袭倒扣全部分数 评分规则如 ...

  2. js处理ajax返回的json数组

    一.json对象和json数组的区别 jsonObject = {} # json对象 jsonArray=[{},{}] # json数组 二.数据处理 前台接收到后台传过来的json数组实际上是一 ...

  3. 11-vue的使用

    一.安装 对于新手来说,强烈建议大家使用<script>引入 二. 引入vue.js文件 我们能发现,引入vue.js文件之后,Vue被注册为一个全局的变量,它是一个构造函数. 三.使用V ...

  4. MySQL之优化

    当 MySQL 单表记录数过大时,增删改查性能都会急剧下降,本文会提供一些优化参考,大家可以参考以下步骤来优化. 一. 单表优化 除非单表数据未来会一直不断上涨,否则不要一开始就考虑拆分,拆分会带来逻 ...

  5. jmeter的jtl日志转html报告常见报错笔记

    问题:生成的jmeter文件可以放任意位置 输入命令转换hmtl报告 PS D:\user\80003288\桌面\Ques> jmeter -g .\test1.jtl -e -o .\rep ...

  6. 小程序wepy.js框架总结

    wepy.js借鉴了Vue的语法风格和功能特性,对官方提供的框架进行了封装,更贴近于MVVM架构模式,让开发者更加容易上手,增加开发效率.(脏数据处理--是否有标识.是否有响应) 前端开发的对组件化开 ...

  7. js-跨域源资源共享(CORS)

    ### 一. CORS(Cross-Origin Resource Sharing,跨域源资源共享) 基本思想:使用自定义HTTP头部让浏览器与服务器进行沟通 发送请求时,需附加一个Origin头部 ...

  8. 配置react-sass

    在配置react-sass时遇到很多坑其中 一条如果你的.scss文件失效 请一定要在fileloader之前配置该sass-loader 配置文件如下 基于你不熟悉webpack 容易出这个错误

  9. jquery操作复选框(checkbox)的一些小技巧总结

    1.获取单个checkbox选中项(三种写法) //第一种 $("input:checkbox:checked").val() //第二种 $("input:[type= ...

  10. vscode开发中绝对让你惊艳的插件!!!(个人在用)

    识别模版引擎 1.Apache Velocity :识别Velocity(vm) 2.Art Template Helper:识别artTemplate 点击路径跳转 1.Laravel goto v ...