제출 #1058374

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1058374		fuad27	Thousands Islands (IOI22_islands)	C++17	10 / 100	35 ms	9304 KiB

islands

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


std::variant<bool, std::vector<int>> find_journey(
    int N, int M, std::vector<int> U, std::vector<int> V) {
	vector<vector<pair<int,int>>> g(N);
	for(int i = 0;i<M;i++) {
		g[U[i]].push_back({V[i],i});
	}
	bool subt2=1;
	{
		vector<vector<int>> a(N, vector<int>(N, -1));
		for(int i = 0;i<M;i++) {
			a[U[i]][V[i]] = i;
		}
		for(int i= 0;i<N;i++) {
			for(int j = 0;j<N;j++) {
				if(i==j)continue;
				if(a[i][j]==-1)subt2=false;
			}
		}
		if(M!=N*(N-1))subt2=false;
	}
	if(N == 2) {
		if(g[0].size() < 2)return false;
		if(g[1].size() == 0)return false;
		vector<int> v;
		v.push_back(g[0][0].second);
		v.push_back(g[1][0].second);
		v.push_back(g[0][1].second);
		v.push_back(g[0][0].second);
		v.push_back(g[1][0].second);
		v.push_back(g[0][1].second);
		return v;
	}
	else if(subt2) {
		if(N <= 2) {
			return false;
		}
		vector<vector<int>> a(N, vector<int>(N, -1));
		for(int i = 0;i<M;i++) {
			a[U[i]][V[i]] = i;
		}
		vector<int> ans={a[0][1], a[1][2], a[2][1], a[0][1], a[0][2], a[1][2], a[2][1], a[0][2]};
		return ans;
	}
	else {
		vector<int> topo;
		vector<bool> vis(N);
		auto dfs = [&](int at, auto &&dfs) -> void {
			if(vis[at])return;
			vis[at]=1;
			for(auto [to,w]:g[at]) {
				dfs(to, dfs);
			}
			topo.push_back(at);
		};
		for(int i= 0;i<N;i++) {
			dfs(i, dfs);
		}
		vis.assign(N, 0);
		reverse(topo.begin(), topo.end());
		for(int i:topo) 
		{
			for(auto [to,w]:g[i]) {
				if(vis[to]) {
					return true;
				}
			}
			vis[i] =1 ;
		}
	}
	return false;
}

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