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 all(fl) fl.begin(),fl.end()
#define pb push_back
#define fi first
#define se second
#define for1(i,j,k) for(int i=j;i<=k;i++)
#define for2(i,j,k) for(int i=j;i>=k;i--)
#define for3(i,j,k,l) for(int i=j;i<=k;i+=l)
#define lb lower_bound
#define ub upper_bound
#define sz(a) (int)a.size()
#define pii pair<int,int>
#define pli pair<long long,int>
#define gcd __gcd
#define lcm(x,y) x*y/__gcd(x,y)
#define pil pair<int,long long>
const int maxn=3e5+9;
vector<pii>a[maxn];
int st[maxn][21],d[maxn];
pii pr[maxn];
void dfs(int u,int par){
for (auto v:a[u]){
if (v.fi==par)continue;
pr[v.fi]={u,v.se};
st[v.fi][0]=u;
for1(i,1,20)st[v.fi][i]=st[st[v.fi][i-1]][i-1];
d[v.fi]=d[u]+1;
dfs(v.fi,u);
}
}
pii b[maxn];
int lca(int u,int v){
if (d[u]<d[v])swap(u,v);
int k=d[u]-d[v];
for1(i,0,20)if (k>>i&1)u=st[u][i];
if (u==v)return u;
for2(i,20,0){
if (!st[u][i]||!st[v][i])continue;
if (st[u][i]!=st[v][i]){
u=st[u][i];
v=st[v][i];
}
}
return st[u][0];
}
bool use[maxn];
int ans[maxn];
signed main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
//freopen("temp.INP","r",stdin);
//freopen("temp.OUT","w",stdout);
int n,m;
cin>>n>>m;
for1(i,1,m)cin>>b[i].fi>>b[i].se;
for1(i,1,n-1){
int id;
cin>>id;
use[id]=1;
a[b[id].fi].pb({b[id].se,id});
a[b[id].se].pb({b[id].fi,id});
}
dfs(1,0);
vector<int>perm;
auto take=[&](int id){
int u=b[id].fi,v=b[id].se,t=lca(u,v),tu=u,tv=v;
vector<int>xd;
while (true){
if (!tu)break;
if (lca(tu,t)==tu)break;
if (pr[tu].se)xd.pb(pr[tu].se);
int lst=pr[tu].fi;
pr[tu]={0,0};
if (!lst)break;
tu=lst;
}
pr[u]={tu,0};
while (true){
if (!tv)break;
if (lca(tv,t)==tv)break;
if (pr[tv].se)xd.pb(pr[tv].se);
int lst=pr[tv].fi;
pr[tv]={0,0};
if (!lst)break;
tv=lst;
}
pr[v]={tv,0};
sort(all(xd));
for (auto v1:xd)perm.pb(v1);
if (!use[id])perm.pb(id);
};
for1(i,1,m){
take(i);
}
int cnt=0;
for (auto v:perm)ans[v]=++cnt;
for1(i,1,m)cout<<ans[i]<<" ";
}
# | 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... |