結果

問題 No.1299 Random Array Score
ユーザー ぷらぷら
提出日時 2020-11-27 22:14:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 772 ms / 2,000 ms
コード長 862 bytes
コンパイル時間 2,046 ms
コンパイル使用メモリ 170,256 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-26 13:11:43
合計ジャッジ時間 16,038 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int,int> P;
int INF = 3e18+7;
int mod = 998244353;
int dx[] = {1, 0,-1, 0, 1, 1,-1,-1};
int dy[] = {0, 1, 0,-1, 1,-1, 1,-1};
int mod_pow(int x,int y) {
    int res = 1;
    while(y > 0) {
        if(y%2) {
            res = res*x%mod;
        }
        x = x*x%mod;
        y/=2;
    }
    return res;
}
signed main() {
    int N,K;
    cin >> N >> K;
    vector<int>A(N);
    int sum = 0;
    for(int i = 0; i < N; i++) {
        cin >> A[i];
        sum += A[i]*mod_pow(N,K)%mod;
        sum %= mod;
        int tmp = A[i];
        tmp *= N;
        tmp %= mod;
        tmp *= (mod_pow(2,K)+mod-1)%mod;
        tmp %= mod;
        tmp *= mod_pow(N,K-1);
        tmp %= mod;
        sum += tmp;
        sum %= mod;
    }
    cout << sum*mod_pow(mod_pow(N,K),mod-2)%mod << endl;
}
0