社区讨论

奇怪的逆元题目求优化

学术版参与者 7已保存回复 21

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
21 条
当前快照
1 份
快照标识符
@mhjcupph
此快照首次捕获于
2025/11/04 00:29
4 个月前
此快照最后确认于
2025/11/04 06:10
4 个月前
查看原帖
一行一个数,表示 i=1nai1×998244353ni(modp)\sum_{i=1}^n a_i^{-1} \times 998244353^{n-i} \pmod p
对于 100%100\% 的数据,有 1n5000000,1ai<p,p=109+71 \le n \le 5000000 , 1 \le a_i < p , p=10^9+7
512 MB,0.5s,求问怎么卡常/优化
CPP
#include <bits/stdc++.h>
using namespace std;
#define int long long
int fac[5000010];
const int mod = 1e9 + 7;
inline int qmi(int a, int b) {
    int res = 1;
    while (b) {
        if (b & 1) {
            res = res * a % mod;
        }
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}
signed main() {
    ios::sync_with_stdio(false);
    ios_base::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int n, ans = 0, a;
    cin >> n;
    fac[0] = 1;
    for (int i = 1; i <= n; i++) {
        fac[i] = fac[i - 1] * 998244353 % mod;
    }
    for (int i = 1; i <= n; i++) {
        cin >> a;
        ans = (ans + qmi(a, mod - 2) * fac[n - i] % mod);
    }
    cout << ans % mod;
    return 0;
}

回复

21 条回复,欢迎继续交流。

正在加载回复...