Submission #36276866

---

Source Code Expand

#pragma GCC target("arch=skylake-avx512") //アーキテクチャーの指定
#pragma GCC optimize("O3") //03最適化
#include <stdio.h>
using ll = long long;
#define MOD 998244353
#define MAX 200001


int main(){
    //inv[i] = (mod上のiの逆数)
    int inv[MAX]{1, 1};
    for(int i = 2; i < MAX; i++){
        inv[i] = MOD - (ll(MOD / i) * inv[MOD % i] % MOD);
    }

    //入力
    int N;
    scanf("%d", &N);
    int A[N + 1]{};
    for(int i = 1; i <= N; i++){
        scanf("%d", A + i);
    }

    //公式解説におけるmax(Ai, Aj)の総和
    ll sum_all = 0;
    //公式解説の解法1
    for(int K = 1; K <= N; K++){
        //max(Ai, AK)の総和を愚直に計算
        ll sum_temp = 0;
        for(int i = 1; i < K; i++){
            sum_temp += (A[i] > A[K] ? A[i] : A[K]);
        }
        //max(Ai, Aj)の総和を更新
        sum_all = (sum_all + sum_temp * 2 + A[K]) % MOD;
        //Kの逆数を2回かけて答えを求める
        int ans = sum_all * inv[K] % MOD * inv[K] % MOD;
        printf("%d\n", ans);
    }

    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:18:10: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   18 |     scanf("%d", &N);
      |     ~~~~~^~~~~~~~~~
./Main.cpp:21:14: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   21 |         scanf("%d", A + i);
      |         ~~~~~^~~~~~~~~~~~~

Judge Result