oj.uz

Submission #1069082

# Submission time Handle Problem Language Result Execution time Memory
1069082 2024-08-21T15:43:15 Z bleahbleah Beech Tree (IOI23_beechtree) C++17
0 / 100
 85 ms 74832 KB 
#include "beechtree.h"
#include <bits/stdc++.h>
#define all(x) (x).begin(),(x).end()
using namespace std;

using ll = long long;
using ld = long double;

//#define int ll
#define sz(x) ((int)(x).size())

using pii = pair<int,int>;
using tii = tuple<int,int,int>;

const int nmax = 1e5 + 5;

vector<pii> g[nmax];
vector<int> P, C;

unordered_set<int> colors[nmax], play[nmax];
bool isanc(unordered_set<int>& A, unordered_set<int>& B) {
   if(sz(A) < sz(B)) return 0;
   for(auto &x : B)
      if(!A.count(x)) return 0;
   return 1;
}

vector<int> sol;
int h[nmax];

vector<int> dfs(int node) {
   cerr << node << '\n';
   vector<int> here;
   sol[node] = 1;
   h[node] = 0;
   for(auto [x, e] : g[node]) {
      auto T = dfs(x);
      
      //cerr << x << ": " << sol[x] << '\n';
      
      sol[node] &= sol[x];
      copy(all(T), back_inserter(here));
      h[node] = max(h[node], h[x] + 1);
   }
   here.emplace_back(node);
   
   if(sol[node] == 0 || sz(g[node]) != sz(colors[node])) { sol[node] = 0; return here; }
   vector<vector<int>> ts(h[node] + 1);
   for(auto x : here) ts[h[x]].emplace_back(x);
   
   for(auto x : here) play[x] = colors[x];
   
   for(int i = 0; i < h[node]; i++) {
         //if(node == 0) {
            //cerr << "niveaux " << i << '\n';
         //}
      for(auto x : here) {
         if(x == node) continue;
         if(h[x] < i) continue;
         
         //if(node == 0) {
            //cerr << x << '\n';
            //for(auto x : play)
         //}
         
         if(!isanc(play[P[x]], play[x])) { sol[node] = 0; return here; }
      }
      for(auto a : ts[i]) {
         if(a == node) continue;
         play[P[a]].erase(C[a]);
      }
   }
   return here;
}

std::vector<int> beechtree(int N, int M, std::vector<int> P_, std::vector<int> C_) {
   P = P_;
   C = C_;
   for(int i = 1; i < N; i++) {
      g[P[i]].emplace_back(i, C[i]);
      colors[P[i]].emplace(C[i]);
   }
   sol.assign(N, 0);
   dfs(0);
   
   
   return sol;
}

/**
      Töte es durch genaue Untersuchung\Töte es kann es nur noch schlimmer machen\Es lässt es irgendwie atmen
--
*/

Subtask #0
0.0 / 0.0

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

Subtask #1
0.0 / 9.0

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

Subtask #2
0.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 13912 KB Output is correct
2 Correct 3 ms 13916 KB Output is correct
3 Correct 3 ms 13940 KB Output is correct
4 Correct 3 ms 13916 KB Output is correct
5 Correct 4 ms 13916 KB Output is correct
6 Correct 3 ms 13916 KB Output is correct
7 Runtime error 85 ms 74832 KB Execution killed with signal 11
8 Halted 0 ms 0 KB -

Subtask #3
0.0 / 9.0

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

Subtask #4
0.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 13912 KB Output is correct
2 Correct 3 ms 13916 KB Output is correct
3 Correct 3 ms 13972 KB Output is correct
4 Correct 3 ms 13916 KB Output is correct
5 Runtime error 85 ms 74832 KB Execution killed with signal 11
6 Halted 0 ms 0 KB -

Subtask #5
0.0 / 14.0

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

Subtask #6
0.0 / 14.0

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

Subtask #7
0.0 / 12.0

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

Subtask #8
0.0 / 17.0

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

Subtask #9
0.0 / 12.0

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