test/fps_bostan_mori.test.cpp

test/fps_bostan_mori.test.cpp · GitHub で見る

検証問題

依存ライブラリ

• Fast I/O
• FPS のブロック変換
• Bostan–Mori の NTT 倍長化
• FPS の融合畳み込み
• FPS の定数係数畳み込み
• FPS の非対称畳み込み
• FPS と多項式除算の内部変換
• FPS の指数・冪の内部変換
• FPS の一点評価
• FPS 逆数のブロック変換
• FPS の内部演算
• FPS の結果配列への一括変換
• 疎 FPS の共通演算
• 疎 FPS の次数圧縮・前処理
• 疎 FPS の項収集・演算種別
• FPS 平方根のブロック変換
• FPS の作業領域
• modint998244353 の共通剰余演算
• modint998244353 の共有畳み込み
• NTT（998244353）（高速版）
• Bostan–Mori（高速版）
• 形式的冪級数（高速版）
• modint998244353

コード

#define PROBLEM "https://judge.yosupo.jp/problem/kth_term_of_linearly_recurrent_sequence"
#include "fps/bostan_mori.hpp"
#include "fixtures/fps_io.hpp"

int main() {
	FastInput input;
	FastOutput output;
	int d = 0; long long k = 0; input.read(d); input.read(k);
	auto a = read_fps(input, d), q = read_fps(input, d);
	for (auto &x : q) x = -x;
	q.insert(q.begin(), mint(1));
	fps p = (a * q).pre(d);
	output.println(bostan_mori(std::move(p), std::move(q), k).val());
}

Back to top page