oj.uz

Submission #841658

# Submission time Handle Problem Language Result Execution time Memory
841658 2023-09-01T19:34:24 Z Batrr Beech Tree (IOI23_beechtree) C++17
0 / 100
 2000 ms 55388 KB 
#include "beechtree.h"
#include <bits/stdc++.h>

#define f first
#define s second
#define pb push_back
#define mp make_pair

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;

const int N = 300500, inf = 1e9, mod = 998244353;
const ll INF = 1e18;

vector<pii> g[N];
int n, m;
int ans[N];
int P[N], C[N];
int tin[N], timer;

vector<int> t[N];

int get(int v, int u) {
    if (g[v].size() > g[u].size())
        return -1;
    if (g[v].size() < g[u].size())
        return 1;
    for (int i = 0; i < g[v].size(); i++) {
        int x = get(g[v][i].s, g[u][i].s);
        if (x != 0)
            return x;
    }
    return 0;
}

bool cmp(int v, int u) {
    int x = get(v, u);
    if(x != 0)
        return x == -1;
    return tin[v] < tin[u];
}

void dfs(int v) {
    tin[v] = timer++;
    t[v].pb(v);
    for (auto it: g[v]) {
        int to = it.s;
        dfs(to);
        for (auto x: t[to])
            t[v].pb(x);
    }
    sort(t[v].begin() + 1, t[v].end(), cmp);

    bool ok = 1;
    ok &= (t[v][0] == v);
    for (int i = 1; i < t[v].size(); i++) {
        int cnt = 0;
        for (int j = 1; j < i; j++)
            if (C[t[v][i]] == C[t[v][j]])
                cnt++;
        ok &= (t[v][cnt] == P[t[v][i]]);
    }
    ans[v] |= ok;

}

vector<int> beechtree(int N, int M, vector<int> PP, vector<int> CC) {

    n = N;
    for (int i = 0; i < n; i++) {
        g[i].clear();
        t[i].clear();
        P[i] = PP[i];
        C[i] = CC[i];
        ans[i] = 0;
    }
    for (int i = 1; i < n; i++)
        g[P[i]].pb({C[i], i});
    for (int i = 0; i < n; i++)
        sort(g[i].begin(), g[i].end());

    dfs(0);


    return vector<int>(ans, ans + n);
}

Compilation message

beechtree.cpp: In function 'int get(int, int)':
beechtree.cpp:31:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |     for (int i = 0; i < g[v].size(); i++) {
      |                     ~~^~~~~~~~~~~~~
beechtree.cpp: In function 'void dfs(int)':
beechtree.cpp:59:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |     for (int i = 1; i < t[v].size(); i++) {
      |                     ~~^~~~~~~~~~~~~

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 3 ms 19036 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct

Subtask #1
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 3 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 3 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 4 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 3 ms 19032 KB Output is correct
16 Correct 3 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 5 ms 19032 KB Output is correct
20 Correct 3 ms 19032 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 3 ms 19032 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Correct 3 ms 19232 KB Output is correct
25 Correct 3 ms 19032 KB Output is correct
26 Correct 3 ms 19036 KB Output is correct
27 Correct 3 ms 19032 KB Output is correct
28 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
29 Halted 0 ms 0 KB -

Subtask #2
0.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 3 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Execution timed out 2041 ms 55388 KB Time limit exceeded
8 Halted 0 ms 0 KB -

Subtask #3
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 4 ms 19032 KB Output is correct
4 Correct 5 ms 19032 KB Output is correct
5 Correct 4 ms 19032 KB Output is correct
6 Correct 4 ms 19032 KB Output is correct
7 Correct 4 ms 19032 KB Output is correct
8 Correct 4 ms 19032 KB Output is correct
9 Incorrect 3 ms 19032 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
10 Halted 0 ms 0 KB -

Subtask #4
0.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Execution timed out 2041 ms 55388 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #5
0.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 3 ms 19036 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 4 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 3 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 3 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 3 ms 19032 KB Output is correct
16 Correct 4 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 3 ms 19032 KB Output is correct
20 Correct 3 ms 19032 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 5 ms 19032 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Correct 3 ms 19032 KB Output is correct
25 Correct 3 ms 19032 KB Output is correct
26 Correct 3 ms 19032 KB Output is correct
27 Correct 3 ms 19232 KB Output is correct
28 Correct 3 ms 19032 KB Output is correct
29 Correct 3 ms 19036 KB Output is correct
30 Correct 3 ms 19032 KB Output is correct
31 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
32 Halted 0 ms 0 KB -

Subtask #6
0.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 3 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 4 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 3 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 5 ms 19032 KB Output is correct
16 Correct 3 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 3 ms 19032 KB Output is correct
20 Correct 3 ms 19232 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 3 ms 19036 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
25 Halted 0 ms 0 KB -

Subtask #7
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 3 ms 19036 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 4 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 3 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 3 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 3 ms 19032 KB Output is correct
16 Correct 4 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 3 ms 19032 KB Output is correct
20 Correct 3 ms 19032 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 5 ms 19032 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Correct 3 ms 19032 KB Output is correct
25 Correct 3 ms 19032 KB Output is correct
26 Correct 3 ms 19032 KB Output is correct
27 Correct 3 ms 19232 KB Output is correct
28 Correct 3 ms 19032 KB Output is correct
29 Correct 3 ms 19036 KB Output is correct
30 Correct 3 ms 19032 KB Output is correct
31 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
32 Halted 0 ms 0 KB -

Subtask #8
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 4 ms 19032 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 3 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 4 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 3 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 5 ms 19032 KB Output is correct
16 Correct 3 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 3 ms 19032 KB Output is correct
20 Correct 3 ms 19232 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 3 ms 19036 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
25 Halted 0 ms 0 KB -

Subtask #9
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 19032 KB Output is correct
2 Correct 3 ms 19036 KB Output is correct
3 Correct 3 ms 19032 KB Output is correct
4 Correct 3 ms 19032 KB Output is correct
5 Correct 4 ms 19032 KB Output is correct
6 Correct 3 ms 19032 KB Output is correct
7 Correct 3 ms 19032 KB Output is correct
8 Correct 3 ms 19032 KB Output is correct
9 Correct 3 ms 19032 KB Output is correct
10 Correct 3 ms 19032 KB Output is correct
11 Correct 3 ms 19032 KB Output is correct
12 Correct 3 ms 19032 KB Output is correct
13 Correct 3 ms 19032 KB Output is correct
14 Correct 3 ms 19032 KB Output is correct
15 Correct 3 ms 19032 KB Output is correct
16 Correct 4 ms 19032 KB Output is correct
17 Correct 3 ms 19032 KB Output is correct
18 Correct 3 ms 19032 KB Output is correct
19 Correct 3 ms 19032 KB Output is correct
20 Correct 3 ms 19032 KB Output is correct
21 Correct 3 ms 19032 KB Output is correct
22 Correct 5 ms 19032 KB Output is correct
23 Correct 3 ms 19032 KB Output is correct
24 Correct 3 ms 19032 KB Output is correct
25 Correct 3 ms 19032 KB Output is correct
26 Correct 3 ms 19032 KB Output is correct
27 Correct 3 ms 19232 KB Output is correct
28 Correct 3 ms 19032 KB Output is correct
29 Correct 3 ms 19036 KB Output is correct
30 Correct 3 ms 19032 KB Output is correct
31 Incorrect 4 ms 19036 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
32 Halted 0 ms 0 KB -