結果

問題 No.1540 級数おもちゃ
ユーザー 遭難者
提出日時 2021-04-29 00:16:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,623 bytes
コンパイル時間 2,624 ms
コンパイル使用メモリ 199,892 KB
最終ジャッジ日時 2025-01-21 01:58:13
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other WA * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:54:11: warning: ‘a’ may be used uninitialized [-Wmaybe-uninitialized]
   54 |         a += (A[i] * T[i] % mod) * I[i][0] % mod;
      |         ~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:51:10: note: ‘a’ was declared here
   51 |     long a, b, c = 0;
      |          ^
main.cpp:56:11: warning: ‘b’ may be used uninitialized [-Wmaybe-uninitialized]
   56 |         b += (A[i] * T[i] % mod) * I[i][1] % mod;
      |         ~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:51:13: note: ‘b’ was declared here
   51 |     long a, b, c = 0;
      |             ^

ソースコード

diff # 

#include <bits/stdc++.h>
#include <iostream>
#include <string>
#include <cstdlib>
#include <cmath>
using namespace std;

const int mod = 998244353;

long modpow(long a, long b)
{
    if (b == 0)
        return 1;
    if ((b & 1) == 0)
        return modpow((a * a) % mod, b >> 1);
    return (a * modpow(a, b - 1)) % mod;
}

int main()
{
    int N;
    cin >> N;
    vector<int> A(N);
    vector<int> T(N);
    T[0] = 1;
    for (int i = 0; i < N; i++)
        cin >> A[i];
    int two = 1;
    for (int i = 0; i < N; i++)
    {
        A[i] *= two;
        A[i] %= mod;
    }
    for (int i = 1; i < N; i++)
        T[i] = (T[i - 1] * 2) % mod;
    vector<vector<long long>> I(N + 1, vector<long long>(3));
    I[0][2] = modpow(6, mod - 2);
    I[0][0] = 0;
    I[0][1] = 0;
    I[1][0] = 1;
    I[1][1] = modpow(-2, mod - 2) + mod;
    I[1][2] = 0;
    for (int i = 2; i < N; i++)
    {
        I[i][0] = ((i - 1) * I[i - 2][0] % mod) * modpow(i, mod - 2) % mod;
        I[i][1] = (((i - 1) * I[i - 2][1] % mod) * modpow(i, mod - 2) % mod - modpow(T[i], mod - 2)) % mod;
        if (I[i][1] < 0)
            I[i][1] += mod;
        I[i][2] = ((i - 1) * I[i - 2][2] % mod) * modpow(i, mod - 2) % mod;
    }
    long a, b, c = 0;
    for (int i = 0; i < N; i++)
    {
        a += (A[i] * T[i] % mod) * I[i][0] % mod;
        a %= mod;
        b += (A[i] * T[i] % mod) * I[i][1] % mod;
        b %= mod;
        c += A[i] * T[i] % mod * I[i][2] % mod;
        c %= mod;
    }
    if (a < 0)
        a += mod;
    if (b < 0)
        b += mod;
    if (c < 0)
        c += mod;
    cout << a << " " << b << " " << c << endl;
}
0