#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pll pair<ll,ll>
#define pii pair<int,int>
#define fs first
#define sc second
#define tlll tuple<ll,ll,ll>
const int mxn = 3e5+10;
pii edges[mxn];
int N,M;
vector<int> paths[mxn*30];
int ppp;
bitset<mxn> need;
int newnode(){
return ++ppp;
}
namespace TREE{
vector<pii> tree[mxn];
int par[mxn][20],vid[mxn][20],dep[mxn];
void dfs(int now){
for(int i = 1;i<20;i++){
par[now][i] = par[par[now][i-1]][i-1];
vid[now][i] = newnode();
paths[vid[now][i]].push_back(vid[now][i-1]);
paths[vid[now][i]].push_back(vid[par[now][i-1]][i-1]);
}
for(auto nxt:tree[now]){
if(nxt.fs == par[now][0])continue;
par[nxt.fs][0] = now;
vid[nxt.fs][0] = nxt.sc;
dep[nxt.fs] = dep[now]+1;
dfs(nxt.fs);
}
return;
}
void GO(){
par[1][0] = 1;
dfs(1);
}
}
namespace DAG{
int deg[mxn*30];
int ans[mxn];
void add_edge(int a,int b,int from){
if(TREE::dep[a]<TREE::dep[b])swap(a,b);
int d = 0;
for(int i = 19;i>=0;i--){
if(d&(1<<i)){
paths[from].push_back(TREE::vid[a][i]);
a = TREE::par[a][i];
}
}
for(int i = 19;i>=0;i--){
if(TREE::par[a][i] != TREE::par[b][i]){
paths[from].push_back(TREE::vid[a][i]);
paths[from].push_back(TREE::vid[b][i]);
a = TREE::par[a][i];
b = TREE::par[b][i];
}
}
if(a != b){
paths[from].push_back(TREE::vid[a][0]);
paths[from].push_back(TREE::vid[b][0]);
}
return;
}
void TOPO(){
priority_queue<pii,vector<pii>,less<pii>> pq;
int C = M;
for(int i = 0;i<=ppp;i++){
for(auto nxt:paths[i]){
deg[nxt]++;
}
}
for(int i = 0;i<=ppp;i++){
if(!deg[i]){
if(i>=1&&i<=M)pq.push(make_pair(i,i));
else pq.push(make_pair(mxn,i));
}
}
while(!pq.empty()){
auto now = pq.top().sc;
pq.pop();
if(now>=1&&now<=M){
ans[now] = C--;
}
for(auto nxt:paths[now]){
deg[nxt]--;
if(!deg[nxt]){
if(nxt>=1&&nxt<=M)pq.push({nxt,nxt});
else pq.push({mxn,nxt});
}
}
}
return;
}
void GET(){
for(int i = 1;i<=M;i++)cout<<ans[i]<<' ';cout<<'\n';
return;
}
}
int main(){
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
cin>>N>>M;
ppp = M+1;
for(int i = 1;i<=M;i++){
cin>>edges[i].fs>>edges[i].sc;
}
for(int i = 1;i<N;i++){
int id;
cin>>id;
need[id] = true;
TREE::tree[edges[id].fs].push_back({edges[id].sc,id});
TREE::tree[edges[id].sc].push_back({edges[id].fs,id});
}
TREE::GO();
for(int i = 1;i<=M;i++){
if(!need[i])DAG::add_edge(edges[i].fs,edges[i].sc,i);
}
DAG::TOPO();
DAG::GET();
}
Compilation message
riggedroads.cpp: In function 'void DAG::GET()':
riggedroads.cpp:114:2: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
114 | for(int i = 1;i<=M;i++)cout<<ans[i]<<' ';cout<<'\n';
| ^~~
riggedroads.cpp:114:43: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
114 | for(int i = 1;i<=M;i++)cout<<ans[i]<<' ';cout<<'\n';
| ^~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
226116 KB |
Output is correct |
2 |
Correct |
48 ms |
226128 KB |
Output is correct |
3 |
Correct |
49 ms |
226184 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
226116 KB |
Output is correct |
2 |
Correct |
48 ms |
226128 KB |
Output is correct |
3 |
Correct |
49 ms |
226184 KB |
Output is correct |
4 |
Correct |
50 ms |
226640 KB |
Output is correct |
5 |
Incorrect |
52 ms |
226596 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
106 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
114 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
159 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
112 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
226116 KB |
Output is correct |
2 |
Correct |
48 ms |
226128 KB |
Output is correct |
3 |
Correct |
49 ms |
226184 KB |
Output is correct |
4 |
Correct |
50 ms |
226640 KB |
Output is correct |
5 |
Incorrect |
52 ms |
226596 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |