Submission #944002

#	Time	Username	Problem	Language	Result	Execution time	Memory
944002		Amaarsaa	Crocodile's Underground City (IOI11_crocodile)	C++14	100 / 100	402 ms	75420 KiB

crocodile

#include<bits/stdc++.h>
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
using ll = long long;
using pll = pair < ll, ll >;
const ll INF = 1e18;
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]){
	ll i, x, c, y, s, p, X;
	vector < pair < ll,ll > > adj[N + 2];
	
	for (i = 0; i < M; i ++) {
		x = R[i][0];
		y = R[i][1];
		adj[x].push_back({y, L[i]});
		adj[y].push_back({x, L[i]});
	}
	priority_queue < pll, vector < pll > , greater < pll > > pq;
	ll D[N + 2][3];
	for (i = 0; i < N; i ++) D[i][1] = INF, D[i][0] = INF;
	for (i = 0; i < K; i ++) {
		pq.push({0, P[i]});
		D[P[i]][0] = D[P[i]][1] = 0;
	}
	
	while(!pq.empty()) {
		x = pq.top().second;
		c = pq.top().first;
		pq.pop();
		if ( D[x][1] != c) continue;
		
		for ( pair < ll, ll > pa : adj[x]) {
			X = pa.first;
			s = pa.second;
			p = D[X][1];
			if ( s + c < D[X][0]) {
				if ( D[X][1] != D[X][0]) {
					D[X][1]= D[X][0];
					pq.push({D[X][1], X});
				}
				D[X][0] = s + c;
			}
			else {
				if ( s + c< D[X][1]) {
					D[X][1] = s + c;
					pq.push({D[X][1], X});
				}
			}
		}
	}
  	return D[0][1];
}

Compilation message (stderr)

crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:11:20: warning: variable 'p' set but not used [-Wunused-but-set-variable]
   11 |  ll i, x, c, y, s, p, X;
      |                    ^

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