This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,avx,avx2,avx512f")
#include<bits/stdc++.h>
//#pragma GCC optimize("Ofast,unroll-loops")
using namespace std;
#define ll long long
#define ull unsigned long long
#define ld long double
#define rep(a) rep1(i,a)
#define rep1(i,a) rep2(i,0,a)
#define rep2(i,b,a) for(int i=(b); i<((int)(a)); i++)
#define rep3(i,b,a) for(int i=(b); i>=((int)(a)); i--)
#define all(a) a.begin(),a.end()
#define pii pair<int,int>
#define pb push_back
#define inf 1010000000
//#define inf 4000000000000000000
#define eps 1e-9
#define sz(a) ((int)a.size())
#define pow2(x) (1ll<<(x))
#define ceiling(a,b) (((a)+(b)-1)/(b))
#ifdef i_am_noob
#define bug(...) cerr << "#" << __LINE__ << ' ' << #__VA_ARGS__ << "- ", _do(__VA_ARGS__)
template<typename T> void _do(T && x) {static int cnt=0;cerr << x << endl;cnt++;if(cnt>1000) exit(0);}
template<typename T, typename ...S> void _do(T && x, S&&...y) {cerr << x << ", "; _do(y...);}
#else
#define bug(...) 826
#endif
inline char readchar(){
const int maxn=1000000;
static char buf[maxn],*p=buf,*q=buf;
if(p==q&&(q=(p=buf)+fread(buf,1,maxn,stdin))==buf) return EOF;
else return *p++;
}
inline int readint(){
int c=readchar(),x=0,neg=0;
while((c<'0'||c>'9')&&c!='-'&&c!=EOF) c=readchar();
if(c=='-') neg=1,c=readchar();
while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^'0'),c=readchar();
return neg?-x:x;
}
const int Mod=1000000007,Mod2=998244353;
const int MOD=Mod;
const int maxn=100005;
//i_am_noob
int n,a[maxn],e[maxn][2],fa[maxn],dep[maxn],siz[maxn],tour[maxn],cur,val[2*maxn][18],id[maxn],to[maxn],head[maxn],bit[maxn],tail[4*maxn],tag[4*maxn];
vector<int> vec,adj[maxn],child[maxn];
vector<pii> opers;
void dfs(int u, int par){
fa[u]=par;
dep[u]=par==-1?0:dep[par]+1;
siz[u]=1;
tour[u]=cur;
val[cur++][0]=u;
for(auto v: adj[u]) if(v!=par){
dfs(v,u);
val[cur++][0]=u;
siz[u]+=siz[v];
}
}
void dfs2(int u, int par, int h){
to[u]=-1,head[u]=h,id[u]=cur++;
if(par>=0) child[par].pb(u);
for(auto v: adj[u]) if(v!=par) if(to[u]==-1||siz[v]>siz[to[u]]) to[u]=v;
if(to[u]>=0) dfs2(to[u],u,h);
for(auto v: adj[u]) if(v!=par&&v!=to[u]) dfs2(v,u,v);
}
bool cmp(int i, int j){return dep[i]<dep[j];}
int querylca(int u, int v){
int l=tour[u],r=tour[v];
if(l>r) swap(l,r);
int x=__lg(r-l+1);
return min(val[l][x],val[r-pow2(x)+1][x],cmp);
}
void modifybit(int p, int x){for(p++; p<maxn; p+=p&-p) bit[p]+=x;}
int querybit(int l, int r){
int res=0;
for(r++; r; r-=r&-r) res+=bit[r];
for(; l; l-=l&-l) res-=bit[l];
return res;
}
void push(int k, int l, int r){
if(l!=r){
if(tag[k]!=-1){
tag[k<<1]=tag[k],tag[k<<1|1]=tag[k];
tail[k<<1]=tag[k],tail[k<<1|1]=tag[k];
}
}
tag[k]=-1;
}
void build(int k, int l, int r){
if(l==r){
tail[k]=0;
return;
}
int mid=l+r>>1;
build(k<<1,l,mid),build(k<<1|1,mid+1,r);
}
void modifyseg(int k, int l, int r, int ql, int qr, int x){
if(l>qr||r<ql) return;
if(ql<=l&&qr>=r){
tail[k]=tag[k]=x;
return;
}
int mid=l+r>>1;
push(k,l,r);
modifyseg(k<<1,l,mid,ql,qr,x),modifyseg(k<<1|1,mid+1,r,ql,qr,x);
}
int queryseg(int k, int l, int r, int p){
if(l==r) return tail[k];
int mid=l+r>>1;
push(k,l,r);
if(p<=mid) return queryseg(k<<1,l,mid,p);
return queryseg(k<<1|1,mid+1,r,p);
}
void modify(int u, int x){
int v;
while(1){
v=head[u];
modifyseg(1,0,maxn-1,id[v],id[u],x);
if(v==0) break;
u=fa[v];
}
}
signed main(){
ios_base::sync_with_stdio(0),cin.tie(0);
#ifdef i_am_noob
freopen("input1.txt","r",stdin);
freopen("output1.txt","w",stdout);
freopen("output2.txt","w",stderr);
#endif
n=readint();
rep(n) a[i]=readint();
rep(n) vec.pb(a[i]);
sort(all(vec));
vec.resize(unique(all(vec))-vec.begin());
rep(n) a[i]=lower_bound(all(vec),a[i])-vec.begin();
rep(n-1) e[i][0]=readint(),e[i][1]=readint();
rep(n-1) e[i][0]--,e[i][1]--;
rep(n-1) adj[e[i][0]].pb(e[i][1]),adj[e[i][1]].pb(e[i][0]);
cur=0;
dfs(0,-1);
rep2(j,1,18) rep(2*n-pow2(j)) val[i][j]=min(val[i][j-1],val[i+pow2(j-1)][j-1],cmp);
cur=0;
dfs2(0,-1,0);
rep(4*maxn) tag[i]=-1;
build(1,0,maxn-1);
rep(n-1){
int u=e[i][0],v=e[i][1],x=0,y,z,t,tot=0;
ll ans=0;
opers.clear();
while(1){
y=queryseg(1,0,maxn-1,id[x]);
z=querylca(u,y);
ans+=1ll*(tot-querybit(0,a[y]))*(dep[z]-dep[x]+1);
if(z==u) break;
modifybit(a[y],dep[z]-dep[x]+1);
opers.pb({a[y],-dep[z]+dep[x]-1});
tot+=dep[z]-dep[x]+1;
x=*--upper_bound(all(child[z]),u,[](int i, int j){return id[i]<id[j];});
}
for(auto& [p,q]: opers) modifybit(p,q);
modify(v,v);
cout << ans << "\n";
}
return 0;
}
Compilation message (stderr)
construction.cpp: In function 'void build(int, int, int)':
construction.cpp:106:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
106 | int mid=l+r>>1;
| ~^~
construction.cpp: In function 'void modifyseg(int, int, int, int, int, int)':
construction.cpp:116:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
116 | int mid=l+r>>1;
| ~^~
construction.cpp: In function 'int queryseg(int, int, int, int)':
construction.cpp:123:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
123 | int mid=l+r>>1;
| ~^~
construction.cpp: In function 'int main()':
construction.cpp:176:19: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
176 | for(auto& [p,q]: opers) modifybit(p,q);
| ^
construction.cpp:163:41: warning: unused variable 't' [-Wunused-variable]
163 | int u=e[i][0],v=e[i][1],x=0,y,z,t,tot=0;
| ^
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |