Submission #628664

#	Time	Username	Problem	Language	Result	Execution time	Memory
628664		dqhungdl	Thousands Islands (IOI22_islands)	C++17	5 / 100	32 ms	23952 KiB

islands

#include "islands.h"
#include <bits/stdc++.h>
#include <variant>
using namespace std;

const int MAX = 1e5 + 5;
int N;
bool mark[MAX];
vector<vector<vector<int>>> g;
vector<int> path;

bool dfs(int u, int p) {
    if (mark[u])
        return false;
    mark[u] = true;
    for (int v = 0; v < N; v++) {
        if (v != p && !g[u][v].empty()) {
            if (!v) {
                path.push_back(0);
                return true;
            }
            if (dfs(v, u)) {
                path.push_back(v);
                return true;
            }
        }
    }
    return false;
}

variant<bool, vector<int>> find_journey(int _N, int M, vector<int> U, vector<int> V) {
    N = _N;
    if (N == 2) {
        vector<int> g[2];
        for (int i = 0; i < M; i++)
            g[U[i]].push_back(i);
        if (g[0].size() < 2 || g[1].empty())
            return false;
        return vector<int>({g[0][0], g[1][0], g[0][1], g[0][0], g[1][0], g[0][1]});
    }
    g.resize(N, vector<vector<int>>(N));
    for (int i = 0; i < M; i++)
        g[U[i]][V[i]].push_back(i);
    bool isTwoSides = M % 2 == 0;
    for (int i = 1; i < M && isTwoSides; i += 2)
        if (U[i - 1] != V[i] || V[i - 1] != U[i])
            isTwoSides = false;
    dfs(0, 0);
    path.push_back(0);
    reverse(path.begin(), path.end());
    vector<int> rs;
    if (isTwoSides) {
        for (int i = 0; i < path.size() - 1; i++)
            rs.push_back(g[path[i]][path[i + 1]][0]);
        for (int i = path.size() - 1; i > 0; i--)
            rs.push_back(g[path[i]][path[i - 1]][0]);
        for (int i = path.size() - 2; i >= 0; i--)
            rs.push_back(g[path[i]][path[i + 1]][0]);
        for (int i = 1; i < path.size(); i++)
            rs.push_back(g[path[i]][path[i - 1]][0]);
        return rs;
    }
    return false;
}

//int main() {
//    freopen("../_input", "r", stdin);
//    int N, M;
//    assert(2 == scanf("%d %d", &N, &M));
//
//    vector<int> U(M), V(M);
//    for (int i = 0; i < M; ++i) {
//        assert(2 == scanf("%d %d", &U[i], &V[i]));
//    }
//
//    variant<bool, vector<int>> result = find_journey(N, M, U, V);
//    if (result.index() == 0) {
//        printf("0\n");
//        if (get<bool>(result)) {
//            printf("1\n");
//        } else {
//            printf("0\n");
//        }
//    } else {
//        printf("1\n");
//        vector<int> &canoes = get<vector<int>>(result);
//        printf("%d\n", static_cast<int>(canoes.size()));
//        for (int i = 0; i < static_cast<int>(canoes.size()); ++i) {
//            if (i > 0) {
//                printf(" ");
//            }
//            printf("%d", canoes[i]);
//        }
//        printf("\n");
//    }
//    return 0;
//}

Compilation message (stderr)

islands.cpp: In function 'std::variant<bool, std::vector<int, std::allocator<int> > > find_journey(int, int, std::vector<int>, std::vector<int>)':
islands.cpp:53:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |         for (int i = 0; i < path.size() - 1; i++)
      |                         ~~^~~~~~~~~~~~~~~~~
islands.cpp:59:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |         for (int i = 1; i < path.size(); i++)
      |                         ~~^~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 5
• Subtask 3: 0 / 21
• Subtask 4: 0 / 24
• Subtask 5: 0 / 45