結果

問題 No.1540 級数おもちゃ
ユーザー logx
提出日時 2021-06-06 12:42:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 99 ms / 2,000 ms
コード長 1,185 bytes
コンパイル時間 824 ms
コンパイル使用メモリ 75,260 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-22 19:07:50
合計ジャッジ時間 4,023 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<tuple>
#include<atcoder/modint>
using mint=atcoder::modint998244353;
void operator+=(std::tuple<mint,mint,mint> &l,std::tuple<mint,mint,mint> r){
    std::get<0>(l)+=std::get<0>(r);
    std::get<1>(l)+=std::get<1>(r);
    std::get<2>(l)+=std::get<2>(r);
    return;
}
std::tuple<mint,mint,mint> operator*(std::tuple<mint,mint,mint> l,mint r){
    std::get<0>(l)*=r;
    std::get<1>(l)*=r;
    std::get<2>(l)*=r;
    return l;
}
void print(std::tuple<mint,mint,mint> x){
    std::cout << std::get<0>(x).val() << ' ' << std::get<1>(x).val() << ' ' << std::get<2>(x).val() << '\n';
}

int main(){
    int n;
    std::cin >> n;
    int a[n+1];
    for(int i=1;i<=n;i++)std::cin >> a[i];
    assert(1<=n && n<=100000);
    for(int i=1;i<=n;i++)assert(1<=a[i] && a[i]<=1000000);
    std::tuple<mint,mint,mint> res;
    std::tuple<mint,mint,mint> ans[n+1];
    for(int j=1;j<=n;j++){
        if(j==1)ans[j]={0,0,mint(1)/6};
        else if(j==2)ans[j]={1,-mint(1)/2,0};
        else{
            ans[j]={0,-1/mint(2).pow(j-1)/(j-1),0};
            ans[j]+=ans[j-2]*(mint(j-2)/mint(j-1));
        }
        res+=ans[j]*(a[j]*mint(2).pow(j));
    }
    print(res);
}
0