Thieves and Prisons

In file included from /usr/include/c++/7/cassert:44:0,
                 from /usr/include/x86_64-linux-gnu/c++/7/bits/stdc++.h:33,
                 from input/code.cpp:1:
input/code.cpp: In function 'void decode()':
input/code.cpp:66:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     assert(p.size()==n);
            ~~~~~~~~^~~

#include <bits/stdc++.h>
using namespace std;

#define rep(i, a, b) for(ll i = a; i < ll(b); ++i)
#define trav(a, x) for(auto& a : x)
#define all(x) x.begin(), x.end()
#define sz(x) (ll)(x).size()
typedef long long ll;
typedef pair<ll, ll> pii;
typedef vector<ll> vi;

ll n;
vector<vi> e;


void getPostorder(ll v, ll l){
    rep(i,0,e[v].size())
        if(e[v][i]!=l) getPostorder(e[v][i],v);
    cout<<v+1<<" ";
}

void encode(){
    e.resize(n);
    rep(i,0,n-1){
        ll a,b;
        cin>>a>>b;
        --a; --b;
        e[a].push_back(b);
        e[b].push_back(a);
    }
    getPostorder(0,-1);
    cout<<endl;
}

vi p;
vector<vi> c;

ll parent(ll v){
    if(p[v]==-1){
        p[v] = p.size();
        p.push_back(-1);
        c.push_back(vi(1,v));
    }
    return p[v];
}

ll children(ll v,ll d){
    if(d==0) return v;
    ll newChild = p.size();
    c[v].push_back(newChild);
    p.push_back(v);
    c.push_back(vi(0));
    return children(newChild, d-1);
}

void decode(){
    ll v = 0;
    p.push_back(-1);
    c.push_back(vi(0));
    rep(i,0,n-1){
        ll dist; cin>>dist;
        v = parent(v);
        v = children(v,dist-1);
    }

    assert(p.size()==n);

    rep(i,0,n){
        if(p[i]!=-1){
            cout<<i+1<<" "<<p[i]+1<<endl;
        }
    }
}

int main() {
    cin.sync_with_stdio(0); cin.tie(0);
    cin.exceptions(cin.failbit);
    ll t;
    cin>>t>>n;
    if(t==1) encode();
    else decode();
}

Test 1

Group: 2, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 3

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 4

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 5

Group: 2, 4, 5

Verdict: WRONG ANSWER

Test 6

Group: 4, 5

Verdict: WRONG ANSWER

Test 7

Group: 4, 5

Verdict: WRONG ANSWER

Test 8

Group: 4, 5

Verdict: WRONG ANSWER

Test 9

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 10

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 11

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 12

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 13

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 14

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 15

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 16

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 17

Group: 1, 2, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 18

Group: 1, 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 19

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 20

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 21

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 22

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 23

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 24

Group: 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 25

Group: 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 26

Group: 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 27

Group: 3, 4, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 28

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 29

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 30

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 31

Group: 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 32

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 33

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 34

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 35

Group: 2, 4, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 36

Group: 3, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 37

Group: 3, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 38

Group: 3, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 39

Group: 3, 5

Verdict: RUNTIME ERROR

code: input/code.cpp:66: void decode(): Assertion `p.size()==n' failed.

Test 40

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 41

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 42

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 43

Group: 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 44

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 45

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 46

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...

Test 47

Group: 2, 5

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::__ios_failure'
  what():  basic_ios::...