This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
//by szh
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pii pair<int,int>
#define pll pair<long long,long long>
#define pb push_back
#define debug(x) cerr<<#x<<"="<<x<<endl
#define pq priority_queue
#define inf 0x3f
#define rep(i,a,b) for (int i=a;i<(b);i++)
#define MP make_pair
#define SZ(x) (int(x.size()))
#define ll long long
#define mod 1000000007
#define ALL(x) x.begin(),x.end()
void inc(int &a,int b) {a=(a+b)%mod;}
void dec(int &a,int b) {a=(a-b+mod)%mod;}
int lowbit(int x) {return x&(-x);}
ll p0w(ll base,ll p) {ll ret=1;while(p>0){if (p%2ll==1ll) ret=ret*base%mod;base=base*base%mod;p/=2ll;}return ret;}
const int maxn = 1e5+10;
int n;
int tree[maxn*4];
int dfn[maxn];
vector <int> edge[maxn];
int w[maxn],ww[maxn];
int st[maxn][17];
pii q[maxn];
int tot = 1;
map <int,int> mp;
int bit[maxn];
int sz[maxn];
int id[maxn];
void dfs(int u) {
dfn[u] = tot++;
sz[u] = 1;
rep(i,1,17) {
if (st[u][i-1]==-1) break;
st[u][i] = st[st[u][i-1]][i-1];
}
for (int v:edge[u]) dfs(v),sz[u] += sz[v];
}
void update(int c,int cl,int cr,int pos,int val) {
if (cl==cr) tree[c] = val;
else {
int mid=cl+cr>>1;
if (pos<=mid) update(c<<1,cl,mid,pos,val);
else update(c<<1|1,mid+1,cr,pos,val);
tree[c] = max(tree[c<<1],tree[c<<1|1]);
}
return;
}
void update(int pos,int val) {
while (pos<maxn) {
bit[pos] += val;
pos += lowbit(pos);
}
return;
}
int query(int c,int cl,int cr,int l,int r) {
if (l<=cl and cr<=r) return tree[c];
int mid=cl+cr>>1;
int ret = -1;
if (l<=mid) ret = query(c<<1,cl,mid,l,r);
if (r>mid) ret = max(ret,query(c<<1|1,mid+1,cr,l,r));
return ret;
}
int query(int pos) {
int ret = 0;
while (pos>0) {
ret += bit[pos];
pos -= lowbit(pos);
}
return ret;
}
int main() {
// freopen("input.txt","r",stdin);
std::ios::sync_with_stdio(false);cin.tie(0);
memset(st,-1,sizeof(st));
memset(tree,-1,sizeof(tree));
cin>>n;
rep(i,1,n+1) cin>>w[i],mp[w[i]]=1;
for (auto &it:mp) it.se=tot++;
tot = 1;
rep(i,1,n+1) ww[i] = mp[w[i]];
id[1] = tot++;
rep(i,1,n) {
int u,v;
cin>>u>>v;
id[v] = tot++;
u = id[u];
v = id[v];
// debug(u),debug(v);
q[i] = {u,v};
edge[u].pb(v);
st[v][0] = u;
}
rep(i,1,n+1) w[id[i]] = ww[i];
tot = 1;
dfs(1);
update(1,1,n,dfn[1],1);
rep(qaq,1,n) {
int u=q[qaq].fi;
// debug(u);
int val=query(1,1,n,dfn[u],dfn[u]+sz[u]-1);
// debug(val);
vector<pii> tmp;
while (1) {
int cnt = 1;
for (int j =16;j>=0;j--) if (st[u][j]!=-1 and query(1,1,n,dfn[st[u][j]],dfn[st[u][j]]+sz[st[u][j]]-1)==val) u = st[u][j],cnt+=(1<<j);
tmp.pb({w[val],cnt});
// debug(w[val]);debug(cnt);debug(u);debug(st[u][0]);
if (st[u][0]==-1) break;
u = st[u][0],val = query(1,1,n,dfn[u],dfn[u]+sz[u]-1);
}
// debug(SZ(tmp));
ll ans = 0;
if (SZ(tmp)>1) {
int cnt = 0;
for (int i=SZ(tmp)-1;i>=0;i--) {
ans += 1ll*(cnt-query(tmp[i].fi))*tmp[i].se;
cnt += tmp[i].se;
update(tmp[i].fi,tmp[i].se);
}
for (auto it:tmp) update(it.fi,-it.se);
}
cout<<ans<<"\n";
update(1,1,n,dfn[q[qaq].se],q[qaq].se);
}
return 0;
}
Compilation message (stderr)
construction.cpp: In function 'void update(int, int, int, int, int)':
construction.cpp:52:13: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
52 | int mid=cl+cr>>1;
| ~~^~~
construction.cpp: In function 'int query(int, int, int, int, int)':
construction.cpp:70:12: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
70 | int mid=cl+cr>>1;
| ~~^~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |