Submission #1262681

#TimeUsernameProblemLanguageResultExecution timeMemory
1262681abdelhakim세계 지도 (IOI25_worldmap)C++20
15 / 100
1143 ms2162688 KiB
#include "worldmap.h"
#include <bits/stdc++.h>
#define ll long long
#define dbg(x) cerr << #x << ' ' << x << endl;
using namespace std;

vector<int> v;
vector<vector<ll>> adj;
void dfs(ll node, ll par)
{
  v.push_back(node+1);
  for (auto &&e : adj[node])
  {
    if(e!=par)
    {
      dfs(e,node);
      v.push_back(node+1);
    }
  }
}
std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) 
{
  v.clear();
  adj.assign(N,vector<ll>());
  for (int i=0;i<M;i++)
  {
    A[i]--;
    B[i]--;
    adj[A[i]].push_back(B[i]);
    adj[B[i]].push_back(A[i]);
  }
  dfs(0,0);
  vector<vector<int>> ans(v.size()+2*N,vector<int>(v.size()+2*N));
  ll sz=v.size()+2*N;
  vector<bool> vis(N);
  for (int i=0,j=0;i<ans.size()&&j<v.size();i++,j++)
  {
    vector<int> toadd(sz,v[j]);
    ans[i]=toadd;
    if(!vis[v[j]-1])
    {
      vis[v[j]-1]=1;
      i++;
      vector<int> one;
      for (int k=0;k<adj[v[j]-1].size();k++)
      {
        one.push_back(adj[v[j]-1][k]+1);
        one.push_back(v[j]);
      }
      while(one.size()<sz)one.push_back(v[j]);
      ans[i]=one;
      i++;
      vector<int> two(sz,v[j]);
      ans[i]=two;
    }
  }
  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...