Add tests

Author: adamant-pwn

verify/poly/linrec_single.test.cpp

@@ -0,0 +1,33 @@
+// @brief Kth term of Linearly Recurrent Sequence
+#define PROBLEM "https://judge.yosupo.jp/problem/kth_term_of_linearly_recurrent_sequence"
+#pragma GCC optimize("Ofast,unroll-loops")
+#include "cp-algo/math/poly.hpp"
+#include <bits/stdc++.h>
+
+using namespace std;
+using namespace cp_algo::math;
+
+const int mod = 998244353;
+using base = modint<mod>;
+using polyn = poly_t<base>;
+
+void solve() {
+    int64_t d, k;
+    cin >> d >> k;
+    vector<base> a(d), c(d);
+    copy_n(istream_iterator<base>(cin), d, begin(a));
+    copy_n(istream_iterator<base>(cin), d, begin(c));
+    polyn Q = polyn(1) - polyn(c).mul_xk(1);
+    polyn P = (polyn(a) * Q).mod_xk(d);
+    cout << polyn::kth_rec(P, Q, k) << endl;
+}
+
+signed main() {
+    //freopen("input.txt", "r", stdin);
+    ios::sync_with_stdio(0);
+    cin.tie(0);
+    int t = 1;
+    while(t--) {
+        solve();
+    }
+}

verify/poly/prod_sequence.test.cpp

@@ -0,0 +1,48 @@
+// @brief Product of Polynomial Sequence
+#define PROBLEM "https://judge.yosupo.jp/problem/product_of_polynomial_sequence"
+#pragma GCC optimize("Ofast,unroll-loops")
+#include "cp-algo/math/poly.hpp"
+#include <bits/stdc++.h>
+
+using namespace std;
+using namespace cp_algo::math;
+
+const int mod = 998244353;
+using base = modint<mod>;
+using polyn = poly_t<base>;
+
+void solve() {
+    int N;
+    cin >> N;
+    vector<polyn> polys(N);
+    multiset<polyn, decltype([](polyn const& a, polyn const& b){
+        return a.deg() < b.deg();
+    })> que = {polyn(1)};
+    int D = 0;
+    for(int i = 0; i < N; i++) {
+        int d;
+        cin >> d;
+        D += d;
+        vector<base> a(d + 1);
+        copy_n(istream_iterator<base>(cin), d + 1, begin(a));
+        que.insert(polyn(a));
+    }
+    while(que.size() > 1) {
+        auto A = *begin(que);
+        que.erase(begin(que));
+        auto B = *begin(que);
+        que.erase(begin(que));
+        que.insert(A * B);
+    }
+    begin(que)->print(D + 1);
+}
+
+signed main() {
+    //freopen("input.txt", "r", stdin);
+    ios::sync_with_stdio(0);
+    cin.tie(0);
+    int t = 1;
+    while(t--) {
+        solve();
+    }
+}