[CSP-S模拟测试]:chinese(数学)
题目传送门(内部题25)
输入格式
一行三个整数$n,m,k$。
输出格式
一行一个整数表示答案。
样例
样例输入:
2 2 2
样例输出:
8
数据范围与提示
样例解释:
$f_0=10,f_1=4,f_2=2,f_3=f_4=0$。
数据范围:
对于所有数据,$2\leqslant n,m\leqslant {10}^9,1\leqslant k\leqslant {10}^6$。
题解
考虑$\sum \limits_{i=0}^{nm}i\times f_i$的意义:所有方案中炼字的个数之和。
统计答案时可以考虑$[1,k]$每个字对答案的贡献,即每个字在多少种方案中成为炼字。在方格的一个确定位置$(x,y)$,字符$i$对答案的贡献($(x,y)$位置的数是$i$且$i$是炼字的方案数)是${(i−1)}^{n−1}{(i − 1)}^{m−1}k^{nm−n−m+1}$。由于诗作中的所有位置都是等价的,那么最后的答案就是
$nm\sum \limits_{i=1}^k {(i−1)}^{n−1} {(i− 1)}^{m−1} k^{nm−n−m+1}$。
时间复杂度:$\Theta(k)$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | #include<bits/stdc++.h> using namespace std; long long n,m,k; long long ans; long long qpow( long long x, long long y) { long long res=1; while (y) { if (y&1)res=res*x%1000000007; x=x*x%1000000007; y>>=1; } return res; } int main() { scanf( "%lld%lld%lld" ,&n,&m,&k); for ( int i=1;i<=k;i++)ans=(ans+qpow(i-1,n+m-2)*qpow(k,n*m-n-m+1)%1000000007)%1000000007; ans=ans*n%1000000007*m%1000000007; printf( "%lld" ,ans); return 0; } |
rp++
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步