제출 #835134

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
835134		Hydrolyzed	악어의 지하 도시 (IOI11_crocodile)	C++14	100 / 100	399 ms	89168 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>

using namespace std;
using ll = long long;

const int MxN = 100010;

struct state_t {
	int v;
	ll w;
	bool operator < (const state_t &o) const {
		return w > o.w;
	}
	state_t(int _v, ll _w):
		v(_v), w(_w) {}
};

vector<pair<int, ll>> adj[MxN];
priority_queue<state_t> pq;
ll dist[MxN][2]; // [0] = best, [1] = secondary
bitset<MxN> visited;

int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
	for(int i=0; i<M; ++i) {
		adj[R[i][0]].emplace_back(R[i][1], L[i]);
		adj[R[i][1]].emplace_back(R[i][0], L[i]);
	}
	memset(dist, 0x3f, sizeof dist);
	for(int i=0; i<K; ++i) {
		pq.emplace(P[i], dist[P[i]][0] = 0ll);
	}
	while(!pq.empty()) {
		state_t now = pq.top(); pq.pop();
		if(visited[now.v]) {
			continue;
		}
		visited[now.v] = true;
		for(auto x: adj[now.v]) {
			ll nxt = now.w + x.second;
			if(dist[x.first][1] > nxt) {
				dist[x.first][1] = nxt;
				if(dist[x.first][1] < dist[x.first][0]) {
					swap(dist[x.first][1], dist[x.first][0]);
				}
				if(dist[x.first][1] <= 1e9 + 100) {
					pq.emplace(x.first, dist[x.first][1]);
				}
			}
		}
	}
	return dist[0][1];
}

• Subtask 1: 46 / 46
• Subtask 2: 43 / 43
• Subtask 3: 11 / 11