From ab16224b192fdbd005cb231c83d2ed816715f82a Mon Sep 17 00:00:00 2001 From: Ta180m Date: Wed, 16 Oct 2019 20:00:06 -0500 Subject: Create btree.cpp --- 2012/November/Gold/btree.cpp | 80 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 80 insertions(+) create mode 100644 2012/November/Gold/btree.cpp diff --git a/2012/November/Gold/btree.cpp b/2012/November/Gold/btree.cpp new file mode 100644 index 0000000..16462ab --- /dev/null +++ b/2012/November/Gold/btree.cpp @@ -0,0 +1,80 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#define init_io(pname) ifstream cin((string)pname+".in"); ofstream cout((string)pname+".out"); ios_base::sync_with_stdio(false); cin.tie(NULL) +#define FOR(i, a, b, in) for (int i = (a); i < (b); i += in) +#define REP(i, a, b) for (int i = (a); i < (b); i++) +#define RFOR(i, a, b, in) for (int i = (a) - 1; i >= (b); i -= in) +#define RREP(i, a, b) for (int i = (a) - 1; i >= (b); i--) +#define trav(a, x) for (auto& a : x) +#define mp make_pair +#define pb push_back +#define f first +#define s second +#define lb lower_bound +#define ub upper_bound +#define sz(x) (int)x.size() +#define all(x) begin(x), end(x) +#define rsz resize +#define mem(a, b) memset(a, (b), sizeof(a)) +using namespace std; +typedef string str; +typedef long long ll; +typedef long double ld; +typedef pair ii; typedef pair pl; typedef pair pd; +typedef vector vi; typedef vector vl; typedef vector vd; +typedef vector vii; typedef vector vpl; typedef vector vpd; +constexpr auto INF = (int)1e9; +constexpr auto LINF = (ll)1e18; + +int L[40005]; +vi G[40005]; +map A[40005]; +map> B[40005]; + +int dfs(int u, int d) { + int ret = 0; + L[u] == 1 ? A[u][d] = 1 : B[u][d] = 1; + for (auto& v : G[u]) { + ret = max(dfs(v, L[v] + d), ret); + for (auto& x : A[v]) { + if (B[u].find(2 * d - L[u] - x.first) != B[u].end()) { // Match left side with corresponding right side + ret = max(max(x.second, B[u][2 * d - L[u] - x.first]), ret); + } + } + for (auto& x : B[v]) { + if (A[u].find(2 * d - L[u] - x.first) != A[u].end()) { // Match right side with corresponding left side + ret = max(max(x.second, A[u][2 * d - L[u] - x.first]), ret); + } + } + for (auto& x : A[v]) A[u][x.first] = max(max(x.first - d, x.second), A[u][x.first]); // Merge A[v] to A[u], adjust max nesting depth accordingly + for (auto& x : B[v]) B[u][x.first] = max(max(d - x.first, x.second), B[u][x.first]); // Merge B[v] to B[u], adjust max nesting depth accordingly + A[v].clear(), B[v].clear(); + } + while (!A[u].empty() && A[u].begin()->first < d) A[u].erase(A[u].begin()); + while (!B[u].empty() && B[u].begin()->first > d) B[u].erase(B[u].begin()); + return ret; +} + +int main() { + init_io("btree"); + int N; cin >> N; + for (int i = 1; i < N; i++) { + int p; cin >> p; + G[--p].push_back(i); + } + for (int i = 0; i < N; i++) { + char c; cin >> c; + L[i] = (c == '(' ? 1 : -1); + } + cout << dfs(0, L[0]); +} -- cgit v1.2.3-70-g09d2