Submission #1349399

#TimeUsernameProblemLanguageResultExecution timeMemory
1349399trimkusWorld Map (IOI25_worldmap)C++20
0 / 100
6 ms1464 KiB
#include "worldmap.h"
#include <bits/stdc++.h>
#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
using namespace std;
using ll = long long;


std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
  vector<vector<int>> adj(N + 1);
  vector<vector<int>> c(N + 1, vector<int>(N + 1));
  for (int i = 0; i < M; ++i) {
    adj[A[i]].push_back(B[i]);
    adj[B[i]].push_back(A[i]);
    c[A[i]][B[i]] = c[B[i]][A[i]] = 1;
  }
  vector<vector<int>> now;
  int mx = 0;
  auto dfs = [&](auto& dfs, int v) -> void {
    // cerr << "v=" << v << endl;
    vector<int> left;
    for (auto& u : adj[v]) {
      if (c[v][u]) {
        left.push_back(u);
        c[v][u] = c[u][v] = 0;
      }
    }
    if (sz(left) == 0) return;
    now.push_back({v});
    vector<int> add;
    add.push_back(v);      
    for (auto& u : left) {
      add.push_back(u);
      add.push_back(v);
    }
    mx = max(mx, sz(add));
    now.emplace_back(add);
    now.push_back({v});
    for (auto& u : left) {
      // cerr << "v="<<v << "u=" << u << endl;
      dfs(dfs, u);
    }
  };
  dfs(dfs, 1);
  // cerr << "a\n";
  // cerr << sz(now) << endl;
  mx = max(mx, sz(now));
  vector<vector<int>> ret(mx);
  for (int i = 0; i < sz(now); ++i) {
    for (int j = 0; j < sz(now[i]); ++j) {
      ret[i].push_back(now[i][j]);
    }
    for (int j = sz(now[i]); j < mx; ++j) {
      ret[i].push_back(now[i][0]);
    }
  }
  for (int i = 1; i <= N; ++i) {
    for (int j = 1; j <= N; ++j) {
      assert(c[i][j] == 0);
    }
  }
  return ret;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...