This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int nx=2e5+5;
int n, a[nx], x, y, rt, l[nx], r[nx], dp[nx];
ll res;
stack<pair<int, int>> s;
void dfs(int u)
{
if (l[u]) dfs(l[u]), dp[u]=max(dp[u], dp[l[u]]+u-l[u]);
if (r[u]) dfs(r[u]), dp[u]=max(dp[u], dp[r[u]]+r[u]-u);
//cout<<"dp "<<u<<' '<<dp[u]<<'\n';
}
int main()
{
cin.tie(NULL)->sync_with_stdio(false);
cin>>n;
for (int i=1; i<=n; i++) cin>>a[i];
for (int i=1; i<n; i++) cin>>x>>y;
for (int i=1; i<=n; i++)
{
if (s.empty()) s.push({a[i], i});
else
{
while (!s.empty()&&s.top().first<a[i])
{
l[i]=s.top().second;
s.pop();
}
if (!s.empty()) r[s.top().second]=i;
else rt=i;
s.push({a[i], i});
}
}
//for (int i=1; i<=n; i++) cout<<"debug "<<i<<' '<<l[i]<<' '<<r[i]<<'\n';
dfs(rt);
cout<<dp[rt];
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |