oj.uz

답안 #936210

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
936210 2024-03-01T12:10:45 Z winter0101 Rigged Roads (NOI19_riggedroads) C++14
19 / 100
 256 ms 67584 KB 
#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};
    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};
    tv=lst;
    }
    pr[v]={tv,0};
    sort(all(xd));
    //cout<<"TAKE "<<id<<'\n';
    for (auto v:xd)perm.pb(v);
    if (!use[id])perm.pb(id);
    //cout<<'\n';
    };
    for1(i,1,m){
    take(i);
    }
    int cnt=0;
    for (auto v:perm)ans[v]=++cnt;
    for1(i,1,m)cout<<ans[i]<<" ";
}

Subtask #1
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 14680 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 14680 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 55 ms 29700 KB Output is correct
2 Correct 103 ms 34424 KB Output is correct
3 Correct 89 ms 22568 KB Output is correct
4 Correct 173 ms 57408 KB Output is correct
5 Correct 208 ms 59284 KB Output is correct

Subtask #4
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 94 ms 38040 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 208 ms 60640 KB Output is correct
2 Correct 211 ms 67096 KB Output is correct
3 Correct 52 ms 30572 KB Output is correct
4 Correct 75 ms 36304 KB Output is correct
5 Correct 256 ms 67584 KB Output is correct

Subtask #6
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 169 ms 46284 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 32.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 14680 KB Output isn't correct
2 Halted 0 ms 0 KB -