oj.uz

답안 #1061941

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1061941 2024-08-16T15:41:19 Z mychecksedad 참나무 (IOI23_beechtree) C++17
0 / 100
 2000 ms 576188 KB 
#include "beechtree.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define all(x) x.begin(),x.end()
#define ll long long
#define ff first
#define ss second
#define vi vector<int>
const int N = 200005;
const ll INF = 1e18;

vector<int> g[N], S[N];
vi res;
bool isok[N];
int antidep[N], sz[N], col[N], n, m, par[N], dep[N];
bool ok = 1;
void dfs(int v){
  set<int> s;
  antidep[v] = 0;
  isok[v] = 1;
  int co = 0, node;
  sz[v] = 1;
  S[v].pb(v);
  for(int u: g[v]){
    dep[u] = dep[v] + 1;
    dfs(u);
    sz[v] += sz[u];
    par[u] = v;
    s.insert(col[u]);
    for(int x: S[u]) S[v].pb(x);
    antidep[v] = max(antidep[u] + 1, antidep[v]);
    if(g[u].size() > 0) co++, node = u;
    isok[v] = isok[v] & isok[u];
  }
  sort(all(g[v]), [&](const int &x, const int &y){
    return sz[x] > sz[y];
  });
  if(g[v].size() != s.size() || isok[v] == 0){
    isok[v] = 0;
    res[v] = 0;
    return;
  }

  // cout << v << ' ' << antidep[v] << '\n';
  if(g[v].size() == 0){
    res[v] = 1;
  }else if(s.size() == g[v].size() && antidep[v] == 1){
    res[v] = 1;
  }else{
    sort(all(S[v]), [&](const int &x, const int &y){
      if(dep[x] != dep[y]) return dep[x] < dep[y];
      return sz[x] > sz[y];
    });


    // cout << v << ' ';
    res[v] = 1;
    for(int i = 0; i + 1 < S[v].size(); ++i){
      set<int> s;
      int u = S[v][i];
      for(int j = 0; j < g[u].size(); ++j) s.insert(col[g[u][j]]); 
      int x = s.size();
      if(dep[u] != dep[S[v][i + 1]]){
        if(par[u] != S[v][i + 1] && par[S[v][i + 1]] != u){
          res[v] = 0;
        }
      }
      u = S[v][i + 1];
      for(int j = 0; j < g[u].size(); ++j) s.insert(col[g[u][j]]); 
      int y = s.size();
      if(x != y){
        res[v] = 0;
        break;
      }
      // cout << S[v][i] << ' ';
    }//cout << '\n';
    // cout << v << ' ' << res[v] << '\n';
  }
}
std::vector<int> beechtree(int nn, int mm, std::vector<int> P, std::vector<int> C)
{n=nn, m=mm;
  res.resize(n);
  for(int i = 1; i < n; ++i){
    g[P[i]].pb(i);
    col[i] = C[i];
  }
  // dfs(0);
  // bool ok = 1;
  // res[n - 1] = 1;
  // for(int i = n - 2; i >= 0; --i){
  //   if(ok == 0){
  //     res[i] = ok;
  //   }else if(C[i] != C[i + 1]){
  //     res[i] = 1;
  //     ok = 0;
  //   }else{
  //     res[i] = 1;
  //   }
  // }
  par[0] = n + 1;
  dep[0] = 0;
  dfs(0);
  return res;
}

Compilation message

beechtree.cpp: In function 'void dfs(int)':
beechtree.cpp:59:26: 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 = 0; i + 1 < S[v].size(); ++i){
      |                    ~~~~~~^~~~~~~~~~~~~
beechtree.cpp:62:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |       for(int j = 0; j < g[u].size(); ++j) s.insert(col[g[u][j]]);
      |                      ~~^~~~~~~~~~~~~
beechtree.cpp:70:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |       for(int j = 0; j < g[u].size(); ++j) s.insert(col[g[u][j]]);
      |                      ~~^~~~~~~~~~~~~
beechtree.cpp:22:15: warning: variable 'node' set but not used [-Wunused-but-set-variable]
   22 |   int co = 0, node;
      |               ^~~~

Subtask #0
0.0 / 0.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13656 KB Output is correct
2 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
3 Halted 0 ms 0 KB -

Subtask #1
0.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 2 ms 13660 KB Output is correct
3 Correct 2 ms 13748 KB Output is correct
4 Correct 2 ms 13660 KB Output is correct
5 Correct 2 ms 13660 KB Output is correct
6 Correct 2 ms 13660 KB Output is correct
7 Correct 2 ms 13660 KB Output is correct
8 Correct 2 ms 13660 KB Output is correct
9 Correct 2 ms 13660 KB Output is correct
10 Correct 3 ms 13660 KB Output is correct
11 Correct 2 ms 13660 KB Output is correct
12 Correct 2 ms 13660 KB Output is correct
13 Correct 2 ms 13660 KB Output is correct
14 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
15 Halted 0 ms 0 KB -

Subtask #2
0.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 2 ms 13660 KB Output is correct
3 Correct 2 ms 13748 KB Output is correct
4 Correct 2 ms 13660 KB Output is correct
5 Correct 2 ms 13660 KB Output is correct
6 Correct 2 ms 13660 KB Output is correct
7 Execution timed out 2070 ms 576188 KB Time limit exceeded
8 Halted 0 ms 0 KB -

Subtask #3
0.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 3 ms 13660 KB Output is correct
3 Correct 2 ms 13660 KB Output is correct
4 Correct 2 ms 13656 KB Output is correct
5 Correct 2 ms 13660 KB Output is correct
6 Correct 2 ms 13660 KB Output is correct
7 Correct 2 ms 13660 KB Output is correct
8 Correct 2 ms 13660 KB Output is correct
9 Incorrect 2 ms 13660 KB 2nd lines differ - on the 1st token, expected: '1', found: '0'
10 Halted 0 ms 0 KB -

Subtask #4
0.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 2 ms 13660 KB Output is correct
3 Correct 2 ms 13660 KB Output is correct
4 Correct 2 ms 13660 KB Output is correct
5 Execution timed out 2070 ms 576188 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #5
0.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13656 KB Output is correct
2 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
3 Halted 0 ms 0 KB -

Subtask #6
0.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 2 ms 13660 KB Output is correct
3 Correct 2 ms 13660 KB Output is correct
4 Correct 2 ms 13660 KB Output is correct
5 Correct 2 ms 13660 KB Output is correct
6 Correct 3 ms 13660 KB Output is correct
7 Correct 2 ms 13660 KB Output is correct
8 Correct 2 ms 13660 KB Output is correct
9 Correct 2 ms 13660 KB Output is correct
10 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
11 Halted 0 ms 0 KB -

Subtask #7
0.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13656 KB Output is correct
2 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
3 Halted 0 ms 0 KB -

Subtask #8
0.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13660 KB Output is correct
2 Correct 2 ms 13660 KB Output is correct
3 Correct 2 ms 13660 KB Output is correct
4 Correct 2 ms 13660 KB Output is correct
5 Correct 2 ms 13660 KB Output is correct
6 Correct 3 ms 13660 KB Output is correct
7 Correct 2 ms 13660 KB Output is correct
8 Correct 2 ms 13660 KB Output is correct
9 Correct 2 ms 13660 KB Output is correct
10 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
11 Halted 0 ms 0 KB -

Subtask #9
0.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 13656 KB Output is correct
2 Incorrect 2 ms 13660 KB 2nd lines differ - on the 2nd token, expected: '1', found: '0'
3 Halted 0 ms 0 KB -