結果

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

ソースコード

diff # 

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

const long long mod = 998244353;

long modpow(long long a, long 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<long long> A(N);
    vector<long long> T(N);
    T[0] = 1;
    for (int i = 0; i < N; i++)
        cin >> A[i];
    for (int i = 0; i < N; i++)
    {
        A[i] *= 2;
        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 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