Submission #1354918

#TimeUsernameProblemLanguageResultExecution timeMemory
1354918scalifrastico_098World Map (IOI25_worldmap)C++20
0 / 100
0 ms344 KiB
#include "worldmap.h"
#include <bits/stdc++.h>
using namespace std; 
#define ll long long
#define F first
#define S second
vector<vector<int>> ans, kl, tr;
queue<int> h; vector<char> vis; vector<int> dist;
void dfs(int u, int p)
{
  vis[u]=1; h.push(u); 
  for(auto x: kl[u])
  {
    if(x==p)continue;if(vis[x])continue; dist[x]=dist[u]+1; 
    tr[u].push_back(x); dfs(x, u); h.push(u);
  }
}
vector<vector<int>> create_map(int n, int m, vector<int> a, vector<int> b) {
  kl.assign(n+1, vector<int>()); vector<int> o(n+1, 0), uw; 
  vis.assign(n+1, 0); tr.assign(n+1, vector<int> ()); dist.assign(n+1, 0);
  while(!h.empty())h.pop();
  for(ll i=0; i<m; i++){kl[a[i]].push_back(b[i]); kl[b[i]].push_back(a[i]);} 
  for(int i=1; i<=n; i++){sort(kl[i].begin(), kl[i].end());}
  for(int i=1; i<=n; i++){if(!vis[i]){dist[i]=0;dfs(i, 0);}}
  int y=(h.size())+2*n; ans.assign(3*n, vector<int>(3*n, -1)); int j=0;
  vector<vector<int>> di; vector<vector<pair<int, int>>> t(n+1);
  while(!h.empty())
  {
    int u=h.front(); h.pop(); vector<int> g(y, u); di.push_back(g);
    uw.push_back(u);
    if(o[u]==0)
    {
      vector<int> p(y, u);int i=1, xd=di.size(); 
      for(auto x: tr[u]){if(i>=y)break; t[u].push_back({xd, i}); i+=2;}
      di.push_back(p); uw.push_back(u);
      vector<int> l(y, u);di.push_back(l);uw.push_back(u);
    }
    o[u]++;
  }
  for(int i=1; i<=n; i++)
  {
    for(auto x: tr[i])
    {
      if(t[i].empty())continue;
      di[t[i].back().F][t[i].back().S]=x; t[i].pop_back();
    }
  }
  int us=0;
  for(int i=0; i<di.size(); i++)
  {
    if(us==3*n)break; if((dist[uw[i]]+i)%2!=0)continue;
    for(int j=0; j<3*n; j++){ans[j][(j+i)%(3*n)]=di[i][j];}
    us++;
  }
  if(us<3*n)
  {
    for(int i=0; i<di.size()&&us<3*n; i++)
    {
      if((dist[uw[i]]+i)%2==0)continue; 
      for(int j=0; j<3*n; j++){ans[j][(j+i)%(3*n)]=di[i][j];}
      us++;
    }
  }
  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...