結果
問題 | No.2506 Sum of Weighted Powers |
ユーザー |
![]() |
提出日時 | 2023-09-10 17:01:48 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 739 ms / 2,000 ms |
コード長 | 2,520 bytes |
コンパイル時間 | 3,555 ms |
コンパイル使用メモリ | 233,624 KB |
最終ジャッジ日時 | 2025-02-16 21:29:26 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 42 |
ソースコード
#include <bits/stdc++.h>#ifdef LOCAL#include <debug.hpp>#else#define debug(...) void(0)#endif#include "atcoder/convolution"#include "atcoder/modint"using namespace std;typedef long long ll;#define all(x) begin(x), end(x)constexpr int INF = (1 << 30) - 1;constexpr long long IINF = (1LL << 60) - 1;constexpr int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};template <class T> istream& operator>>(istream& is, vector<T>& v) {for (auto& x : v) is >> x;return is;}template <class T> ostream& operator<<(ostream& os, const vector<T>& v) {auto sep = "";for (const auto& x : v) os << exchange(sep, " ") << x;return os;}template <class T, class U = T> bool chmin(T& x, U&& y) { return y < x and (x = forward<U>(y), true); }template <class T, class U = T> bool chmax(T& x, U&& y) { return x < y and (x = forward<U>(y), true); }template <class T> void mkuni(vector<T>& v) {sort(begin(v), end(v));v.erase(unique(begin(v), end(v)), end(v));}template <class T> int lwb(const vector<T>& v, const T& x) { return lower_bound(begin(v), end(v), x) - begin(v); }/*(j + k)^3 = j^3 + 3(j + k)jk + k^3(j + k)jk = ((j + k)^3 - j^3 - k^3) / 3B_j C_k x^{(j + k)jk} = (B_j x^{-j^3}) (C_k x^{-k^3}) x^{(j + k)^3}(j + k)^3 // 3 - j^3 // 3 - k^3 // 3 + ((j + k) % 3 - j % 3 - k % 3) / 3最後の項が -1 or 0*/using mint = atcoder::static_modint<943718401>;int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int n, x;cin >> n >> x;vector<int> A(n + 1), B(n + 1), C(n + 1);cin >> A >> B >> C;mint ans = 0;if (x == 0) {for (int i = 0; i <= n; i++) {ans += mint(A[i]) * B[0] * C[i];ans += mint(A[i]) * B[i] * C[0];}ans -= mint(A[0]) * B[0] * C[0];cout << ans.val() << '\n';return 0;}vector<mint> power(n + 1);for (int i = 0; i <= n; i++) power[i] = mint(x).pow(1LL * i * i * i / 3);mint inv = mint(x).inv();vector<mint> b(n + 1, 0), c(n + 1, 0);for (int j = 0; j < 3; j++) {for (int k = 0; k < 3; k++) {for (int i = 0; i <= n; i++) {mint p = power[i].inv();b[i] = (i % 3 == j ? p * B[i] : 0);c[i] = (i % 3 == k ? p * C[i] : 0);}auto a = atcoder::convolution(b, c);for (int i = 0; i <= n; i++) ans += a[i] * power[i] * A[i] * (j + k >= 3 ? inv : 1);}}cout << ans.val() << '\n';return 0;}