题解：P11029 『DABOI Round 1』A Simple Problem

最新推荐文章于 2024-10-09 09:53:23 发布

andycode666

最新推荐文章于 2024-10-09 09:53:23 发布

阅读量951

点赞数 15

分类专栏：洛谷题解文章标签：算法 c++

本文链接：https://blog.csdn.net/andycode_/article/details/142136771

版权

洛谷题解专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目传送门

题目大意

给出 $n$ ，求 $\prod \limits_{i=1}^{n} \prod\limits_{j=1}^{i} \prod \limits_{k=1}^{j} k^k$ 对 $998244353$ 取模后的值。

思路讲解.

不知道快速幂的请移步这里。

如果直接枚举 $i, j, k$ ，那时间复杂度为遍历的时长乘上快速幂的时长，即 $O(n^3logn)$ ，肯定会 TLE。

仔细观察，不难发现。在遍历时，要使 $k^k$ 出现， $j$ 必须不小于 $k$ ，否则遍历不到 $k$ 。所以 $j$ 有 $n - k + 1$ 种选择， $i$ 有 $n - j + 1$ 种选择，所以 $k^k$ 的出现次数为 $\sum^{n}_ {j=k}\sum^{n}_ {i=j} 1$ ，用等差数列公式求和，为 $\frac{(2+n-k) \times (n-k+1)}{2}$ 。所以 $k^k$ 在 $\prod \limits_{i=1}^{n} \prod\limits_{j=1}^{i} \prod \limits_{k=1}^{j} k^k$ 中出现的次数为 $\frac{(2+n-k) \times (1+n-k)}{2}$ 。

所以我们只需遍历 $k$ ，用快速幂求一下 $(k^k)^\frac{(2+n-k) \times (1+n-k)}{2}$ ，即 $k^{k \times \frac{(2+n-k) \times (1+n-k)}{2}}$ 的值即可。算法复杂度为 $O(n \log{n})$ 。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int mod=998244353;
long long quick_mi(int a,long long b){
    long long sum=1,base=a;
    while(b){
        if(b&1)
            sum*=base,sum%=mod;
        b>>=1,base*=base,base%=mod;
    }
    return sum;
}
int main(){
    int n;
    long long ans=1;
    cin>>n;
    for(int k=1;k<=n;k++){
		long long t=1LL*(2+n-k)*(1+n-k)/2;
		ans*=quick_mi(k,1LL*k*t),ans%=mod;
		//cout<<t<<' '<<ans<<'\n';
	}
    printf("%lld",ans);
    return 0;
}