oj.uz

Submission #711867

# Submission time Handle Problem Language Result Execution time Memory
711867 2023-03-17T15:32:40 Z Pety Pipes (CEOI15_pipes) C++14
40 / 100
 4150 ms 30776 KB 
#include <vector>
#include <iostream>
#include <cassert>

using namespace std;

int n, m, x, y, sz[100002], p[100002], dp[20][100002], depth[100002];
vector<int>G[100002];

int find (int x) {
  if (p[x] == x)
    return x;
  return p[x] = find(p[x]);
}

void dfs2 (int x, int par) {
  dp[0][x] = par;
  depth[x] = depth[par] + 1;
  for (int i = 1; (1 << i) < n; i++)
    dp[i][x] = dp[i - 1][dp[i - 1][x]];
  for (auto it : G[x]) {
    if (it == par)
      continue;
    dfs2(it, x);
  }
}
int cnt;
void merge (int x, int y) {
  int px = find(x);
  int py = find(y);
  if (px == py)
    return;

  if (sz[px] > sz[py]) {
    swap(x, y);
    swap(px, py);
  }
  G[x].push_back(y);
  G[y].push_back(x);

  p[px] = py;
  sz[py] += sz[px];
  dfs2(x, y);
}

int lca (int x, int y) {
  if (depth[x] > depth[y])
    swap(x, y);
  int d = depth[y] - depth[x];
  for (int i = 0; (1 << i) <= d; i++)
    if (d & (1 << i))
      y = dp[i][y];
  if (x == y)
    return x;
  for (int i = 19; i >= 0; i--)
    if ((1 << i) != n && dp[i][x] != dp[i][y]) {
      x = dp[i][x];
      y = dp[i][y];
    }
  return dp[0][x];
}

int add[100002], viz[100002];

void dfs (int x, int par) {
  viz[x] = 1;
  for (auto it : G[x]) {
    if (it == par)
      continue;
    dfs(it, x);
    add[x] += add[it];
  }
  if (add[x] == 0 && par != 0)
    cout << x << " " << par << "\n";
}

void dfs3 (int x, pair<int, int> edge) {
  viz[x] = 1;
  int i = 0;
  for (auto it : G[x]) {
    if (edge == make_pair(x, i)) {
      i++;
      continue;
    }
    if (!viz[it])
       dfs3(it, edge);
    i++;
  }
}

int main () 
{
  ios_base::sync_with_stdio(false);
  cin.tie(0); cout.tie(0);
  cin >> n >> m;
  for (int i = 1; i <= n; i++) {
    p[i] = i;
    sz[i] = 1;
  }
  if (m <= 200) {
    for (int i = 1; i <= m; i++) {
      int x, y;
      cin >> x >> y;
      G[x].push_back(y);
      G[y].push_back(x);
    }
    for (int x = 1; x <=n; x++) {
      for (int j = 0; j < G[x].size(); j++) {
        int val = G[x][j];
        for (int k = 1; k <= n; k++)
          viz[k] = 0;
        dfs3(x, {x, j});
        if (!viz[val] && x < val) {
          cout << x << " " << val << "\n";
        }
      }
    }
    return 0;
  }
  for (int i = 1; i <= m; i++) {
    cin >> x >> y;
    if (find(x) == find(y)) {
      int L = lca(x, y);
      assert(L > 0);
      add[L] -= 2;
      add[x]++;
      add[y]++;
    }
    else {
      merge(x, y);
      //assert(abs(depth[x] - depth[y]) == 1);
    }
  }
  for (int i = 1; i <= n; i++)
    if (!viz[find(i)])
      dfs(find(i), 0);
  return 0;
}

Compilation message

pipes.cpp: In function 'int main()':
pipes.cpp:108:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  108 |       for (int j = 0; j < G[x].size(); j++) {
      |                       ~~^~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2644 KB Output is correct
2 Correct 2 ms 2644 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 3156 KB Output is correct
2 Correct 6 ms 3156 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 139 ms 3092 KB Output is correct
2 Correct 126 ms 3028 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 284 ms 3736 KB Output is correct
2 Correct 353 ms 3716 KB Output is correct

Subtask #5
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 464 ms 5352 KB Output is correct
2 Runtime error 428 ms 19796 KB Memory limit exceeded
3 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 887 ms 10944 KB Output is correct
2 Runtime error 789 ms 29152 KB Memory limit exceeded
3 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1586 ms 12124 KB Output is correct
2 Runtime error 1351 ms 16772 KB Memory limit exceeded
3 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2231 ms 14364 KB Output is correct
2 Runtime error 2092 ms 22020 KB Memory limit exceeded
3 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 3019 ms 18600 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #10
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 4150 ms 30776 KB Memory limit exceeded
2 Halted 0 ms 0 KB -