#include "worldmap.h"
#include <cassert>
#include <queue>
#include <iostream>
std::vector<std::pair<int, int>> spanning_tree(std::vector<std::vector<int>>& adj) {
std::vector<bool> vis(adj.size(), false);
std::vector<std::pair<int, int>> edges;
std::queue<int> q;
q.push(1);
vis[1] = true;
while (!q.empty()) {
int node = q.front();
q.pop();
for (int dest : adj[node]) {
if (!vis[dest]) {
vis[dest] = true;
edges.push_back({node, dest});
q.push(dest);
}
}
}
return edges;
}
void dfs(int node, std::vector<std::vector<int>>& adj, std::vector<bool>& vis, std::vector<int>& order) {
vis[node] = true;
order.push_back(node);
for (int dest : adj[node]) {
if (!vis[dest]) {
dfs(dest, adj, vis, order);
order.push_back(node);
}
}
}
std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
assert(M == N-1);
std::vector<std::vector<int>> adj(N+1, std::vector<int>());
std::vector<std::pair<int, int>> tree_edges = spanning_tree(adj);
for (auto &edge : tree_edges) {
adj[edge.first].push_back(edge.second);
adj[edge.second].push_back(edge.first);
}
std::vector<bool> vis(N+1, false);
std::vector<int> order;
dfs(1, adj, vis, order);
int sz = order.size();
std::vector<std::vector<int>> ans(sz, std::vector<int>(sz));
for (int i = 0; i < sz; i++) {
for (int j = 0; j < sz; j++) {
ans[i][j] = order[i];
}
}
return ans;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |