oj.uz

Submission #1058224

# Submission time Handle Problem Language Result Execution time Memory
1058224 2024-08-14T09:00:19 Z Triseedot Beech Tree (IOI23_beechtree) C++17
0 / 100
 2000 ms 16112 KB 
// In honor of El Psy Congroo
#include <bits/stdc++.h>
using namespace std;

#ifdef LOCAL
#include <debug.h>
#else
#define debug(...)
#endif

using ll = long long;
#define all(v) v.begin(), v.end()
#define len(v) int(v.size())

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C) {
    vector<vector<int>> g(N);
    for (int u = 1; u < N; u++) {
        int v = P[u];
        g[v].push_back(u);
    }
    
    auto compress = [] (vector<int> &a) {
        int n = len(a);
        vector<pair<int, int>> b(n);
        for (int i = 0; i < n; i++) {
            b[i] = {a[i], i};
        }
        sort(all(b));
        int curr = 0;
        for (int i = 0; i < n; i++) {
            if (i != 0 && b[i].first != b[i - 1].first) curr++;
            a[b[i].second] = curr;
        }
    };
    compress(C);
    
    vector<bool> bad(N, false);
    for (int v = 0; v < N; v++) {
        vector<int> cnt(N, 0);
        for (int u : g[v]) {
            if (cnt[C[u]]) bad[v] = true;
            cnt[C[u]]++;
        }
    }
    
    auto calc = [&] (int root) {
        vector<int> cnt(N, 0), cnt_leaf(N, 0);
        vector<vector<int>> q(N + 1);
        int curr = 0;
        function<void(int)> calc_cnt = [&] (int v) {
            if (len(g[v]) == 0) {
                cnt_leaf[C[v]]++;
            }
            for (int u : g[v]) {
                if (cnt[C[u]] == 0) curr++;
                cnt[C[u]]++;
                calc_cnt(u);
            }
        };
        calc_cnt(root);
        for (int c = 0; c < N; c++) {
            if (cnt[c] - cnt_leaf[c] > 1) {
                return false;
            }
        }
        
        q[len(g[root])].push_back(root);
        while (!q[curr].empty() && curr > 0) {
            int v = q[curr].back();
            q[curr].pop_back();
            if (bad[v]) continue;
            
            for (int u : g[v]) {
                cnt[C[u]]--;
                if (cnt[C[u]] == 0) {
                    curr--;
                }
                q[len(g[u])].push_back(u);
            }
        }
        
        return curr == 0;
    };
    
    vector<int> res(N);
    for (int i = 0; i < N; i++) res[i] = calc(i);
    return res;
}

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 344 KB Output is correct

Subtask #1
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
4 Halted 0 ms 0 KB -

Subtask #2
0.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 0 ms 348 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
4 Halted 0 ms 0 KB -

Subtask #3
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 1 ms 348 KB Output is correct
11 Correct 3 ms 348 KB Output is correct
12 Correct 2 ms 344 KB Output is correct
13 Correct 3 ms 348 KB Output is correct
14 Correct 2 ms 348 KB Output is correct
15 Execution timed out 2094 ms 9860 KB Time limit exceeded
16 Halted 0 ms 0 KB -

Subtask #4
0.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Execution timed out 2092 ms 16112 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #5
0.0 / 14.0

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

Subtask #6
0.0 / 14.0

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

Subtask #7
0.0 / 12.0

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

Subtask #8
0.0 / 17.0

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

Subtask #9
0.0 / 12.0

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