結果
| 問題 |
No.2595 Parsing Challenge
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2023-12-23 19:24:17 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 26,729 bytes |
| コンパイル時間 | 10,656 ms |
| コンパイル使用メモリ | 465,740 KB |
| 最終ジャッジ日時 | 2025-02-18 14:00:30 |
|
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 5 |
| other | AC * 30 TLE * 1 -- * 24 |
ソースコード
#include <vector>
#include <string>
#include <atcoder/convolution>
const int DIGIT = 6;
const int BASE = 1000000;
struct positive_bigint {
std::vector<int> d;
positive_bigint() {
}
positive_bigint(long long X) {
while (X > 0) {
d.push_back(X % BASE);
X /= BASE;
}
}
positive_bigint(std::string S) {
if (S == "0") {
S = "";
}
int L = S.size();
d.resize((L + DIGIT - 1) / DIGIT, 0);
for (int i = L - 1; i >= 0; i -= 6) {
for (int j = std::max(i - 5, 0); j <= i; j++) {
d[i / DIGIT] *= 10;
d[i / DIGIT] += S[j] - '0';
}
}
std::reverse(d.begin(), d.end());
}
bool empty() const {
return d.empty();
}
int size() const {
return d.size();
}
int& operator [](int i) {
return d[i];
}
int operator [](int i) const {
return d[i];
}
};
std::string to_string(const positive_bigint& A) {
int N = A.size();
std::string ans;
for (int i = N - 1; i >= 0; i--) {
std::string tmp = std::to_string(A[i]);
if (i < N - 1) {
ans += std::string(DIGIT - tmp.size(), '0');
}
ans += tmp;
}
if (ans.empty()) {
ans = "0";
}
return ans;
}
std::istream& operator >>(std::istream& is, positive_bigint& A) {
std::string S;
is >> S;
A = positive_bigint(S);
return is;
}
std::ostream& operator <<(std::ostream& os, positive_bigint& A) {
os << to_string(A);
return os;
}
int cmp(const positive_bigint& A, const positive_bigint& B) {
int N = A.size();
int M = B.size();
if (N < M) {
return -1;
}
else if (N > M) {
return 1;
}
else {
for (int i = N - 1; i >= 0; i--) {
if (A[i] < B[i]) {
return -1;
}
if (A[i] > B[i]) {
return 1;
}
}
return 0;
}
}
bool operator ==(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) == 0;
}
bool operator !=(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) != 0;
}
bool operator <(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) < 0;
}
bool operator >(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) > 0;
}
bool operator <=(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) <= 0;
}
bool operator >=(const positive_bigint& A, const positive_bigint& B) {
return cmp(A, B) >= 0;
}
positive_bigint& operator +=(positive_bigint& A, const positive_bigint& B) {
int N = A.size();
int M = B.size();
while (N < M) {
A.d.push_back(0);
N++;
}
for (int i = 0; i < M; i++) {
A[i] += B[i];
}
for (int i = 0; i < N - 1; i++) {
if (A[i] >= BASE) {
A[i] -= BASE;
A[i + 1]++;
}
}
if (N > 0) {
if (A[N - 1] >= BASE) {
A.d.push_back(1);
A[N - 1] -= BASE;
}
}
return A;
}
positive_bigint operator +(const positive_bigint& A, const positive_bigint& B) {
positive_bigint A2 = A;
A2 += B;
return A2;
}
positive_bigint& operator -=(positive_bigint& A, const positive_bigint& B) {
int N = A.size();
int M = B.size();
for (int i = 0; i < M; i++) {
A[i] -= B[i];
}
for (int i = 0; i < N - 1; i++) {
if (A[i] < 0) {
A[i] += BASE;
A[i + 1]--;
}
}
while (!A.empty()) {
if (A.d.back() == 0) {
A.d.pop_back();
}
else {
break;
}
}
return A;
}
positive_bigint operator -(const positive_bigint& A, const positive_bigint& B) {
positive_bigint A2 = A;
A2 -= B;
return A2;
}
positive_bigint operator *(const positive_bigint& A, const positive_bigint& B) {
if (A.empty() || B.empty()) {
return 0;
}
int N = A.size();
int M = B.size();
std::vector<long long> a(N);
for (int i = 0; i < N; i++) {
a[i] = A[i];
}
std::vector<long long> b(M);
for (int i = 0; i < M; i++) {
b[i] = B[i];
}
std::vector<long long> C = atcoder::convolution_ll(a, b);
for (int i = 0; i < N + M - 2; i++) {
C[i + 1] += C[i] / BASE;
C[i] %= BASE;
}
if (C[N + M - 2] >= BASE) {
C.resize(N + M);
C[N + M - 1] += C[N + M - 2] / BASE;
C[N + M - 2] %= BASE;
}
positive_bigint ans;
ans.d.resize(C.size());
for (int i = 0; i < C.size(); i++) {
ans[i] = C[i];
}
return ans;
}
positive_bigint operator *=(positive_bigint& A, const positive_bigint& B) {
A = A * B;
return A;
}
struct bigint {
bool neg = false;
positive_bigint a;
bigint() {
}
bigint(long long X) : neg(X < 0), a(abs(X)) {
}
bigint(const positive_bigint& X, bool neg = false) : neg(neg), a(X) {
}
bigint(const std::string& s) {
if (!s.empty()) {
if (s[0] == '-') {
neg = true;
a = positive_bigint(s.substr(1, s.size() - 1));
}
else {
a = positive_bigint(s);
}
}
}
bool empty() const {
return a.empty();
}
int size() const {
return a.size();
}
int& operator [](int i) {
return a[i];
}
};
std::string to_string(const bigint& A) {
std::string ans;
if (A.neg) {
ans += '-';
}
ans += to_string(A.a);
return ans;
}
std::istream& operator >>(std::istream& is, bigint& A) {
std::string S;
is >> S;
if (S != "0") {
A = bigint(S);
}
return is;
}
std::ostream& operator <<(std::ostream& os, bigint A) {
os << to_string(A);
return os;
}
positive_bigint abs(const bigint& A) {
return A.a;
}
int cmp(const bigint& A, const bigint& B) {
if (!A.neg) {
if (!B.neg) {
return cmp(A.a, B.a);
}
else {
return 1;
}
}
else {
if (!B.neg) {
return -1;
}
else {
return cmp(B.a, A.a);
}
}
}
bool operator ==(const bigint& A, const bigint& B) {
return cmp(A, B) == 0;
}
bool operator !=(const bigint& A, const bigint& B) {
return cmp(A, B) != 0;
}
bool operator <(const bigint& A, const bigint& B) {
return cmp(A, B) < 0;
}
bool operator >(const bigint& A, const bigint& B) {
return cmp(A, B) > 0;
}
bool operator <=(const bigint& A, const bigint& B) {
return cmp(A, B) <= 0;
}
bool operator >=(const bigint& A, const bigint& B) {
return cmp(A, B) >= 0;
}
bigint operator +(const bigint& A) {
return A;
}
bigint operator -(const bigint& A) {
bigint A2 = A;
if (!A2.empty()) {
A2.neg = !A2.neg;
}
return A2;
}
bigint& operator +=(bigint& A, const bigint& B) {
if (A.neg == B.neg) {
A.a += B.a;
}
else {
int c = cmp(A.a, B.a);
if (c > 0) {
A.a -= B.a;
}
else if (c < 0) {
A.a = B.a - A.a;
A.neg = !A.neg;
}
else {
A = 0;
}
}
return A;
}
bigint operator +(const bigint& A, const bigint& B) {
bigint A2 = A;
A2 += B;
return A2;
}
bigint& operator -=(bigint& A, const bigint& B) {
if (A.neg != B.neg) {
A.a += B.a;
}
else {
int c = cmp(A.a, B.a);
if (c > 0) {
A.a -= B.a;
}
else if (c < 0) {
A.a = B.a - A.a;
A.neg = !A.neg;
}
else {
A = 0;
}
}
return A;
}
bigint operator -(const bigint& A, const bigint& B) {
bigint A2 = A;
A2 -= B;
return A2;
}
bigint operator *=(bigint& A, const bigint& B) {
if (A.empty() || B.empty()) {
A = 0;
}
else {
if (B.neg) {
A.neg = !A.neg;
}
A.a *= B.a;
}
return A;
}
bigint operator *(const bigint& A, const bigint& B) {
bigint A2 = A;
A2 *= B;
return A2;
}
#ifndef HIDDEN_IN_VS // 折りたたみ用
// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS
// ライブラリの読み込み
#include <bits/stdc++.h>
using namespace std;
// 型名の短縮
using ll = long long; using ull = unsigned long long; // -2^63 ~ 2^63 = 9 * 10^18(int は -2^31 ~ 2^31 = 2 * 10^9)
using pii = pair<int, int>; using pll = pair<ll, ll>; using pil = pair<int, ll>; using pli = pair<ll, int>;
using vi = vector<int>; using vvi = vector<vi>; using vvvi = vector<vvi>; using vvvvi = vector<vvvi>;
using vl = vector<ll>; using vvl = vector<vl>; using vvvl = vector<vvl>; using vvvvl = vector<vvvl>;
using vb = vector<bool>; using vvb = vector<vb>; using vvvb = vector<vvb>;
using vc = vector<char>; using vvc = vector<vc>; using vvvc = vector<vvc>;
using vd = vector<double>; using vvd = vector<vd>; using vvvd = vector<vvd>;
template <class T> using priority_queue_rev = priority_queue<T, vector<T>, greater<T>>;
using Graph = vvi;
// 定数の定義
const double PI = acos(-1);
const vi DX = { 1, 0, -1, 0 }; // 4 近傍(下,右,上,左)
const vi DY = { 0, 1, 0, -1 };
int INF = 1001001001; ll INFL = 4004004003104004004LL; // (int)INFL = 1010931620;
// 入出力高速化
struct fast_io { fast_io() { cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(18); } } fastIOtmp;
// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define lbpos(a, x) (int)distance((a).begin(), std::lower_bound(all(a), x))
#define ubpos(a, x) (int)distance((a).begin(), std::upper_bound(all(a), x))
#define Yes(b) {cout << ((b) ? "Yes\n" : "No\n");}
#define YES(b) {cout << ((b) ? "YES\n" : "NO\n");}
#define rep(i, n) for(int i = 0, i##_len = int(n); i < i##_len; ++i) // 0 から n-1 まで昇順
#define repi(i, s, t) for(int i = int(s), i##_end = int(t); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = int(s), i##_end = int(t); i >= i##_end; --i) // s から t まで降順
#define repe(v, a) for(const auto& v : (a)) // a の全要素(変更不可能)
#define repea(v, a) for(auto& v : (a)) // a の全要素(変更可能)
#define repb(set, d) for(int set = 0, set##_ub = 1 << int(d); set < set##_ub; ++set) // d ビット全探索(昇順)
#define repis(i, set) for(int i = lsb(set), bset##i = set; i >= 0; bset##i -= 1 << i, i = lsb(bset##i)) // set の全要素(昇順)
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て(昇順)
#define smod(n, m) ((((n) % (m)) + (m)) % (m)) // 非負mod
#define uniq(a) {sort(all(a)); (a).erase(unique(all(a)), (a).end());} // 重複除去
#define EXIT(a) {cout << (a) << endl; exit(0);} // 強制終了
#define inQ(x, y, u, l, d, r) ((u) <= (x) && (l) <= (y) && (x) < (d) && (y) < (r)) // 矩形内判定
// 汎用関数の定義
template <class T> inline ll pow(T n, int k) { ll v = 1; rep(i, k) v *= n; return v; }
template <class T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す)
template <class T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す)
template <class T> inline T get(T set, int i) { return (set >> i) & T(1); }
// 演算子オーバーロード
template <class T, class U> inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; }
template <class T> inline istream& operator>>(istream& is, vector<T>& v) { repea(x, v) is >> x; return is; }
template <class T> inline vector<T>& operator--(vector<T>& v) { repea(x, v) --x; return v; }
template <class T> inline vector<T>& operator++(vector<T>& v) { repea(x, v) ++x; return v; }
#endif // 折りたたみ用
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
#ifdef _MSC_VER
#include "localACL.hpp"
#endif
//using mint = modint1000000007;
using mint = modint998244353;
//using mint = modint; // mint::set_mod(m);
namespace atcoder {
inline istream& operator>>(istream& is, mint& x) { ll x_; is >> x_; x = x_; return is; }
inline ostream& operator<<(ostream& os, const mint& x) { os << x.val(); return os; }
}
using vm = vector<mint>; using vvm = vector<vm>; using vvvm = vector<vvm>; using vvvvm = vector<vvvm>;
#endif
#ifdef _MSC_VER // 手元環境(Visual Studio)
#include "local.hpp"
#else // 提出用(gcc)
inline int popcount(int n) { return __builtin_popcount(n); }
inline int popcount(ll n) { return __builtin_popcountll(n); }
inline int lsb(int n) { return n != 0 ? __builtin_ctz(n) : -1; }
inline int lsb(ll n) { return n != 0 ? __builtin_ctzll(n) : -1; }
inline int msb(int n) { return n != 0 ? (31 - __builtin_clz(n)) : -1; }
inline int msb(ll n) { return n != 0 ? (63 - __builtin_clzll(n)) : -1; }
#define dump(...)
#define dumpel(v)
#define dump_list(v)
#define dump_mat(v)
#define input_from_file(f)
#define output_to_file(f)
#define Assert(b) { if (!(b)) while (1) cout << "OLE"; }
#endif
#include <boost/multiprecision/cpp_int.hpp>
using Bint = boost::multiprecision::cpp_int; // クソデカ数が 1 個あるだけで TLE する
//using Bint = bigint;
string s; int i = 0;
Bint number() {
int sgn = 1;
while (s[i] == '-') {
i++;
sgn *= -1;
}
string num;
while (isdigit(s[i])) {
num += s[i];
i++;
}
return sgn * Bint(move(num));
}
Bint expression();
Bint factor() {
Bint res;
if (s[i] == '(') {
i++;
res = expression();
i++;
}
else {
res = number();
}
return res;
}
Bint term() {
vector<Bint> fs;
fs.emplace_back(factor());
while (1) {
if (s[i] == '*') {
i++;
fs.emplace_back(factor());
}
else break;
}
int m = sz(fs);
priority_queue_rev<pii> q;
// rep(i, m) q.push({ sz(fs[i].a.d), i });
rep(i, m) q.push({ 1, i });
while (sz(q) >= 2) {
auto [di, i] = q.top(); q.pop();
auto [dj, j] = q.top(); q.pop();
fs[i] *= fs[j];
q.push({ di + dj, i });
}
return fs[q.top().second];
}
Bint expression() {
Bint res = term();
while (1) {
if (s[i] == '+') {
i++;
res += term();
}
else if (s[i] == '-') {
i++;
res -= term();
}
else break;
}
return res;
}
void TLE() {
int n;
cin >> n >> s;
//s = "";
//rep(i, (int)1e6) s += '1';
//s += '1';
// コードテストで試してみたらこれで落ちる.
//s = "";
//rep(i, ((int)1e6 - 3) / 4) s += "9*(";
//s += "9*9";
//rep(i, ((int)1e6 - 3) / 4) s += ")";
//dump(sz(s));
// これでも落ちる.数式の木を構築して HLD せな・・・
//s = "";
//rep(i, ((int)1e6 - 1) / 7) s += "9*(9+(";
//s += "9";
//rep(i, ((int)1e6 - 1) / 7) s += ")";
s.push_back('$');
cout << expression() << endl;
}
vi l, r; vc op; vector<Bint> v; int pt = 0;
int number2() {
int sgn = 1;
while (s[i] == '-') {
i++;
sgn *= -1;
}
string num;
if (sgn == -1) num += "-";
while (isdigit(s[i])) {
num += s[i];
i++;
}
l.push_back(-1);
r.push_back(-1);
op.push_back('n');
Bint val;
val.assign(num);
v.emplace_back(val);
return pt++;
}
int expression2();
int factor2() {
int x = -1;
if (s[i] == '(') {
i++;
x = expression2();
i++;
}
else {
x = number2();
}
return x;
}
int term2() {
int x = factor2();
while (1) {
if (s[i] == '*') {
i++;
int y = factor2();
l.push_back(x);
r.push_back(y);
op.push_back('*');
v.emplace_back(0);
x = pt++;
}
else break;
}
return x;
}
int expression2() {
int x = term2();
while (1) {
if (s[i] == '+') {
i++;
int y = term2();
l.push_back(x);
r.push_back(y);
op.push_back('+');
v.emplace_back(0);
x = pt++;
}
else if (s[i] == '-') {
i++;
int y = term2();
l.push_back(x);
r.push_back(y);
op.push_back('-');
v.emplace_back(0);
x = pt++;
}
else break;
}
return x;
}
//【正方行列(固定サイズ)】
/*
* Fixed_matrix<T, n>() : O(n^2)
* T の要素を成分にもつ n×n 零行列で初期化する.
*
* Fixed_matrix<T, n>(bool identity = true) : O(n^2)
* T の要素を成分にもつ n×n 単位行列で初期化する.
*
* Fixed_matrix<T, n>(vvT a) : O(n^2)
* 二次元配列 a[0..n)[0..n) の要素で初期化する.
*
* A + B : O(n^2)
* n×n 行列 A, B の和を返す.+= も使用可.
*
* A - B : O(n^2)
* n×n 行列 A, B の差を返す.-= も使用可.
*
* c * A / A * c : O(n^2)
* n×n 行列 A とスカラー c のスカラー積を返す.*= も使用可.
*
* A * x : O(n^2)
* n×n 行列 A と n 次元列ベクトル array<T, n> x の積を返す.
*
* x * A : O(n^2)
* n 次元行ベクトル array<T, n> x と n×n 行列 A の積を返す.
*
* A * B : O(n^3)
* n×n 行列 A と n×n 行列 B の積を返す.
*
* Mat pow(ll d) : O(n^3 log d)
* 自身を d 乗した行列を返す.
*/
template <class T, int n>
struct Fixed_matrix {
array<array<T, n>, n> v; // 行列の成分
// n×n 零行列で初期化する.identity = true なら n×n 単位行列で初期化する.
Fixed_matrix(bool identity = false) {
rep(i, n) v[i].fill(T(0));
if (identity) rep(i, n) v[i][i] = T(1);
}
// 二次元配列 a[0..n)[0..n) の要素で初期化する.
Fixed_matrix(const vector<vector<T>>& a) {
// verify : https://yukicoder.me/problems/no/1000
Assert(sz(a) == n && sz(a[0]) == n);
rep(i, n) rep(j, n) v[i][j] = a[i][j];
}
// 代入
Fixed_matrix(const Fixed_matrix&) = default;
Fixed_matrix& operator=(const Fixed_matrix&) = default;
// アクセス
inline array<T, n> const& operator[](int i) const { return v[i]; }
inline array<T, n>& operator[](int i) { return v[i]; }
// 入力
friend istream& operator>>(istream& is, Fixed_matrix& a) {
rep(i, n) rep(j, n) is >> a[i][j];
return is;
}
// 比較
bool operator==(const Fixed_matrix& b) const { return v == b.v; }
bool operator!=(const Fixed_matrix& b) const { return !(*this == b); }
// 加算,減算,スカラー倍
Fixed_matrix& operator+=(const Fixed_matrix& b) {
rep(i, n) rep(j, n) v[i][j] += b[i][j];
return *this;
}
Fixed_matrix& operator-=(const Fixed_matrix& b) {
rep(i, n) rep(j, n) v[i][j] -= b[i][j];
return *this;
}
Fixed_matrix& operator*=(const T& c) {
rep(i, n) rep(j, n) v[i][j] *= c;
return *this;
}
Fixed_matrix operator+(const Fixed_matrix& b) const { return Fixed_matrix(*this) += b; }
Fixed_matrix operator-(const Fixed_matrix& b) const { return Fixed_matrix(*this) -= b; }
Fixed_matrix operator*(const T& c) const { return Fixed_matrix(*this) *= c; }
friend Fixed_matrix operator*(const T& c, const Fixed_matrix& a) { return a * c; }
Fixed_matrix operator-() const { return Fixed_matrix(*this) *= T(-1); }
// 行列ベクトル積 : O(n^2)
array<T, n> operator*(const array<T, n>& x) const {
array<T, n> y{ 0 };
rep(i, n) rep(j, n) y[i] += v[i][j] * x[j];
return y;
}
// ベクトル行列積 : O(n^2)
friend array<T, n> operator*(const array<T, n>& x, const Fixed_matrix& a) {
array<T, n> y{ 0 };
rep(i, n) rep(j, n) y[j] += x[i] * a[i][j];
return y;
}
// 積:O(n^3)
Fixed_matrix operator*(const Fixed_matrix& b) const {
// verify : https://yukicoder.me/problems/no/1000
Fixed_matrix res;
rep(i, n) rep(j, n) rep(k, n) res[i][j] += v[i][k] * b[k][j];
return res;
}
Fixed_matrix& operator*=(const Fixed_matrix& b) { *this = *this * b; return *this; }
// 累乗:O(n^3 log d)
Fixed_matrix pow(ll d) const {
Fixed_matrix res(true), pow2(*this);
while (d > 0) {
if (d & 1) res *= pow2;
pow2 *= pow2;
d /= 2;
}
return res;
}
#ifdef _MSC_VER
friend ostream& operator<<(ostream& os, const Fixed_matrix& a) {
rep(i, n) {
os << "[";
rep(j, n) os << a[i][j] << " ]"[j == n - 1];
if (i < n - 1) os << "\n";
}
return os;
}
#endif
};
using MAT = Fixed_matrix<Bint, 2>;
//【一次式の積の展開(基本対称式)】O(n (log n)^2)(の改変)
/*
* Πi∈[0..n) (z - x[i]) を返す.
*
* 戻り値の i 次の項の係数は,x[0..n) の符号付き n-i 次基本対称式になる.
*/
MAT expand(vector<MAT> f) {
// verify : https://judge.yosupo.jp/problem/factorial
int n = sz(f);
// 2 冪個ずつ掛けていく(分割統治法)
for (int k = 1; k < n; k *= 2) {
for (int i = 0; i + k < n; i += 2 * k) {
f[i] = f[i] * f[i + k];
}
}
return f[0];
}
//【貰う木 DP(森経由,多項式,mod 998244353)】O(n (log n)^3)(の改変)
/*
* 与えられた r を根とする根付き木に対し,r に対応する多項式を返す.
*
* 制約 :
* ある部分森に対応する多項式が f(z), ある部分木に対応する多項式が g(z) のとき,
* これらを直和した部分森に対応する多項式は積 f(z) g(z) である.
*
* MFPS leaf(int s) :
* 葉 s のみからなる部分木に対応する多項式を返す.
*
* pair<MFPS, MFPS> apply(int s) :
* ある部分森に対応する多項式が f(z) で,これらに共通の根 s を追加した部分木に
* 対応する多項式が a(z) f(z) + b(z) のとき,組 {a(z), b(z)} を返す.
*
* 利用:【多項式の積の展開】,【多項式の累積積の加重和】
*/
Bint tree_getDP_forest_MFPS(int rt) {
// 参考 : https://atcoder.jp/contests/abc269/editorial/4838
// verify : https://atcoder.jp/contests/abc269/tasks/abc269_h
int n = pt;
// 貰う木 DP で各部分木の重さを求め,重さ最大の頂点を最後になぞるよう順番を入れ替える.
// ついでに親へ戻る辺を削除し有向木にする.
vi w(n);
function<int(int)> dfs_w = [&](int s) {
w[s]++;
if (l[s] != -1) w[s] += dfs_w(l[s]);
if (r[s] != -1) w[s] += dfs_w(r[s]);
return w[s];
};
dfs_w(rt); dump(w);
function<MAT(int)> dfs_root;
function<void(int, vector<MAT>&)> dfs_path;
// heavy path の根である s に対応する多項式を返す.
dfs_root = [&](int s) {
// s が葉の場合は専用の答えを返す.
if (l[s] == -1) {
MAT mat;
mat[0][0] = v[s];
mat[1][0] = 1;
return mat;
}
// fh : s を根とする heavy path 上の頂点に対応する多項式を浅い順に並べたもの
// coef : fh の累積積に掛かる係数を浅い順に並べたもの
vector<MAT> fh;
// heavy path 上の頂点に対応する多項式を fh, coef に格納する.
if (w[l[s]] > w[r[s]]) {
MAT mat(true);
if (op[s] == '+') {
mat[0][1] = dfs_root(r[s])[0][0];
}
else if (op[s] == '-') {
mat[0][1] = -dfs_root(r[s])[0][0];
}
else {
mat[0][0] = dfs_root(r[s])[0][0];
}
fh.emplace_back(mat);
dfs_path(l[s], fh);
}
else {
MAT mat(true);
if (op[s] == '+') {
mat[0][1] = dfs_root(l[s])[0][0];
}
else if (op[s] == '-') {
mat[0][0] = -1;
mat[0][1] = dfs_root(l[s])[0][0];
}
else {
mat[0][0] = dfs_root(l[s])[0][0];
}
fh.emplace_back(mat);
dfs_path(r[s], fh);
}
// 分割統治法を用いて heavy path 上の多項式をまとめる計算を一括で行う.
auto mat = expand(fh);
dump("s:", s); dumpel(fh); dump(mat);
return mat;
};
// heavy path の根でない s に対応する多項式を格納する.
// fh : s が属する heavy path 上の頂点に対応する多項式を浅い順に並べたもの
// coef : fh の累積積に掛かる係数を浅い順に並べたもの
dfs_path = [&](int s, vector<MAT>& fh) {
// s が葉の場合は専用の答えを格納する.
if (l[s] == -1) {
MAT mat;
mat[0][0] = v[s];
mat[1][0] = 1;
fh.emplace_back(mat);
return;
}
if (w[l[s]] > w[r[s]]) {
MAT mat(true);
if (op[s] == '+') {
mat[0][1] = dfs_root(r[s])[0][0];
}
else if (op[s] == '-') {
mat[0][1] = -dfs_root(r[s])[0][0];
}
else {
mat[0][0] = dfs_root(r[s])[0][0];
}
fh.emplace_back(mat);
dfs_path(l[s], fh);
}
else {
MAT mat(true);
if (op[s] == '+') {
mat[0][1] = dfs_root(l[s])[0][0];
}
else if (op[s] == '-') {
mat[0][0] = -1;
mat[0][1] = dfs_root(l[s])[0][0];
}
else {
mat[0][0] = dfs_root(l[s])[0][0];
}
fh.emplace_back(mat);
dfs_path(r[s], fh);
}
};
return dfs_root(rt)[0][0];
};
int main() {
// input_from_file("input.txt");
// output_to_file("output.txt");
int n;
cin >> n >> s;
s.push_back('$');
int rt = expression2();
dump(l); dump(r); dump(op); dump(v); dump(rt); dump(pt);
//function<void(int)> dfs = [&](int s) {
// if (l[s] == -1) {
// cout << v[s];
// }
// else {
// cout << "(";
// dfs(l[s]);
// cout << op[s];
// dfs(r[s]);
// cout << ")";
// }
//};
//dfs(rt); cout << endl;
// これで TLE なら寝る
cout << tree_getDP_forest_MFPS(rt) << endl;
}