Submission #1057804

#TimeUsernameProblemLanguageResultExecution timeMemory
1057804mychecksedadThousands Islands (IOI22_islands)C++17
1.75 / 100
84 ms25172 KiB
#include "islands.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define en cout << '\n'
#define all(x) x.begin(),x.end()
#define pii pair<int,int>
#define vi vector<int>
#define ff first
#define ss second
const int N = 1e5+10;

int n, m;
vector<pair<int, int>> g[N];
vector<int> euler;
bool dfs(int v, int par, vector<bool> &vis){
  vis[v] = 1;
  // euler.pb(v);
  for(auto U: g[v]){
    int u = U.ss;
    if(u == 0 && par != 0){
      euler.pb(U.ff);
      return 1;
    }
    if(!vis[u]){
      euler.pb(U.ff);
      bool ok = dfs(u, v, vis);
      if(ok) return 1;
      euler.pop_back();
    }

  }
  return 0;
}

std::variant<bool, std::vector<int>> find_journey(int nn, int mm, std::vector<int> U, std::vector<int> V) { n=nn, m=mm;
  vector<vi> C(nn);
  map<pii, vi> A;
  for(int i = 0; i < mm; ++i){
    C[U[i]].pb(i);
    g[U[i]].pb({i, V[i]});
    A[{U[i], V[i]}].pb(i);
  }
  for(int i = 1; i < n; ++i){
    if(A[{0, i}].size() > 1 && A[{i, 0}].size() > 1){
      return vi{A[{0, i}][0], A[{i, 0}][0], A[{0, i}][1], A[{i, 0}][0], A[{0, i}][0]};
    }
  }
  // if(!(C[0].size() > 1 && C[1].size() > 0)) return false;
  vector<bool> vis(n);
  bool ok = dfs(0, 0, vis);
  if(!ok) return false;
  vi res;
  for(int i: euler){
    res.pb(i);
  }
  reverse(all(euler));
  for(int i: euler){
    res.pb(i^1);
  }

  // reverse(all(euler));
  for(int i: euler){
    res.pb(i);
  }

  reverse(all(euler));
  for(int i: euler){
    res.pb(i^1);
  }
  return res;
}
#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...