# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
836373 | SamAnd | Thousands Islands (IOI22_islands) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "islands.h"
#include <variant>
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define m_p make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
typedef long long ll;
const int N = 1003, M = 200005;
vector<pair<int, int> > g[N];
bool c[N];
vector<int> ans;
bool dfs(int x, vector<int>& v)
{
c[x] = true;
for (int i = 0; i < g[x].size(); ++i)
{
int h = g[x][i].fi;
int hi = g[x][i].se;
if (hi / 2 == v.back() / 2)
continue;
if (!c[h])
{
p[h] = x;
pi[h] = hi;
if (dfs(h))
return true;
}
else
{
return true;
}
}
return false;
}
std::variant<bool, std::vector<int> > find_journey(int N, int M, std::vector<int> U, std::vector<int> V)
{
for (int i = 0; i < M; ++i)
{
g[U[i]].push_back(m_p(V[i], i));
}
if (N == 2)
{
if (sz(g[0]) >= 2 && sz(g[1]) >= 1)
{
ans.push_back(g[0][0].se);
ans.push_back(g[1][0].se);
ans.push_back(g[0][1].se);
ans.push_back(g[0][0].se);
ans.push_back(g[1][0].se);
ans.push_back(g[0][1].se);
return ans;
}
return false;
}
int x = 0;
while (1)
{
vector<pair<int, int> > v;
for (int i = 0; i < g[x].size(); ++i)
{
int h = g[x][i].fi;
int hi = g[x][i].se;
if (!ans.empty() && ans.back() / 2 == hi / 2)
continue;
v.push_back(m_p(h, hi));
}
if (v.empty())
return false;
if (sz(v) == 1)
{
ans.push_back(v[0].se);
x = v[0].fi;
continue;
}
return true;
}
}
/*int main()
{
int n, m;
cin >> n >> m;
vector<int> u, v;
for (int i = 0; i < m; ++i)
{
int x, y;
cin >> x >> y;
u.push_back(x);
v.push_back(y);
}
cout << find_journey(n, m, u, v) << "\n";
return 0;
}*/
/*
5 6
0 1
1 0
1 3
3 1
0 4
4 0
3 6
0 1
1 0
1 2
2 1
2 0
0 2
*/