Submission #1252547

#TimeUsernameProblemLanguageResultExecution timeMemory
1252547Clan328World Map (IOI25_worldmap)C++20
0 / 100
4 ms832 KiB
#include "worldmap.h"
#include <bits/stdc++.h>

using namespace std;

#define pb push_back

vector<vector<int>> G;
vector<int> order, visited;

void dfs(int v) {
  order.pb(v);
  visited[v] = true;
  for (auto u : G[v]) {
    if (visited[u]) continue;
    dfs(u);
    order.pb(v);
  }
}

std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
  G = vector<vector<int>>(N);
  for (int i = 0; i < N; i++) {
    G[A[i]-1].push_back(B[i]-1);
    G[B[i]-1].push_back(A[i]-1);
  }

  order = vector<int>();
  visited = vector<int>(N);
  dfs(0);
  
  std::vector<std::vector<int>> ans(3*N + order.size() - N, std::vector<int>(3*N + order.size() - N, 1));

  visited = vector<int>(N);
  int idx = 0;
  for (auto x : order) {
    if (visited[x]) {
      for (int i = 0; i < ans.size(); i++) {
        ans[idx][i] = x+1;
      }
      idx++;
      continue;
    }

    visited[x] = true;

    for (int i = 0; i < ans.size(); i++) {
      ans[idx][i] = x+1;
      ans[idx+1][i] = x+1;
      ans[idx+2][i] = x+1;
    }

    int curr = 0;
    for (auto u : G[x]) {
      ans[idx+1][curr] = u+1;
      curr += 2;
    }

    idx += 3;
  }

  return ans;
}
#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...