oj.uz

Submission #1046274

# Submission time Handle Problem Language Result Execution time Memory
1046274 2024-08-06T12:21:32 Z sleepntsheep File Paths (BOI15_fil) C
0 / 100
 104 ms 788 KB 
#include <stdio.h>
#include <stdlib.h>

#define N 6005

int n_, s, n, m, k, m_, par[N], *eh[N], eo[N], dep[N], chain[N], o;

int cc(const void *i, const void *j) {
  return *(int*)i-*(int*)j;
}

void pus(int i, int j) {
  int o = eo[i]++;
  if (!o) eh[i] = (int*)malloc(sizeof **eh * 2);
  else if (0 == (o & o - 1)) eh[i] = (int*)realloc(eh[i], 2 * o * sizeof **eh);
  eh[i][o] = j;
}

void efs(int u) {
  dep[u] = 1e9;
  for (int j = 0; j < eo[u]; j += 2) if (eh[u][j] != par[u]) {
    int v = eh[u][j], l = eh[u][j+1];
    dep[v] = l + dep[u];
    efs(v);
  }
}

void dfs(int u) {
  for (int j = 0; j < eo[u]; j += 2) if (eh[u][j] != par[u]) {
    int v = eh[u][j], l = eh[u][j+1];
    dep[v] = l + dep[u];
    if (dep[v] > k)
      efs(v);
    else
      dfs(v);
  }
}

int main(void) {
  scanf("%d%d%d%d", &n, &m, &k, &s);
  n_ = n;
  m_ = m;
  n = n + m;
  ++s;

  for (int p, l, i = 1; i <= n; ++i) {
    scanf("%d%d", &p, &l);
    pus(par[i] = p, i);
    pus(p, l + 1);
  }

  dfs(0);

  /* sub 2 */
  for (int i = n_ + 1; i <= n; ++i) {
    if (dep[i] == k) {
      puts("YES");
      continue;
    }

    o = 0;
    for (int cur = i; cur != 0; cur = par[cur])
      chain[o++] = dep[i] - dep[cur];
    chain[o++] = dep[i];

    qsort(chain, o, sizeof *chain, cc);

    for (int j = 1; j <= n_; ++j) {
      if (dep[j] + s > k)
        continue;
      int left = k - s - dep[j];

      int md, lb = 0, ub = o;
      while (ub - lb > 1)
        if (chain[md = lb + (ub - lb) / 2] <= left) lb = md;
        else ub = md;
      if (chain[lb] == left) {
        puts("YES");
        goto Next;
      }
    }


    puts("NO");
Next:;
  }
}

Compilation message

fil.c: In function 'pus':
fil.c:15:24: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   15 |   else if (0 == (o & o - 1)) eh[i] = (int*)realloc(eh[i], 2 * o * sizeof **eh);
      |                      ~~^~~
fil.c: In function 'main':
fil.c:40:3: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   40 |   scanf("%d%d%d%d", &n, &m, &k, &s);
      |   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
fil.c:47:5: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   47 |     scanf("%d%d", &p, &l);
      |     ^~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 33.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 604 KB Output is correct
2 Correct 21 ms 604 KB Output is correct
3 Correct 22 ms 616 KB Output is correct
4 Correct 22 ms 604 KB Output is correct
5 Incorrect 104 ms 788 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #3
0 / 34.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -