Ankit_132

E

Oct 28th, 2023
253
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.69 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. const int MOD = 998244353;
  5.  
  6. long long int fp(long long int a, long long int b) {
  7.     long long int res = 1;
  8.     while(b) {
  9.         if(b & 1)
  10.             (res *= a) %= MOD;
  11.         (a *= a) %= MOD;
  12.         b >>= 1;
  13.     }
  14.  
  15.     return res;
  16. }
  17.  
  18. int main()
  19. {
  20.     long long int N;
  21.     cin >> N;
  22.     vector<long long int> A(N + 1), dp(N + 1), sum(N + 1);
  23.  
  24.     long long int inv = fp(N, MOD-2);
  25.  
  26.     for (int i = 1; i <= N; i++) {
  27.         cin >> A[i];
  28.     }
  29.  
  30.     for (int i = 0; i < N; i++) {
  31.         dp[i + 1] = (sum[i] * inv + inv) % MOD;
  32.         sum[i + 1] = (sum[i] + dp[i + 1]) % MOD;
  33.     }
  34.  
  35.     long long ans = 0;
  36.  
  37.     for (int i = 1; i <= N; i++) {
  38.         ans = (ans + dp[i] * A[i]) % MOD;
  39.     }
  40.  
  41.     cout << ans << endl;
  42. }
Advertisement
Add Comment
Please, Sign In to add comment