oj.uz

Submission #211368

# Submission time Handle Problem Language Result Execution time Memory
211368 2020-03-20T08:15:39 Z socho Crocodile's Underground City (IOI11_crocodile) C++14
0 / 100
 10 ms 8064 KB 
#include "crocodile.h"
#include "bits/stdc++.h"
using namespace std;
// #define endl '\n'
// #define int long long

int n, m;
const int MXN = 100005;
vector<pair<int, int> > adj[MXN];
bool ter[MXN], assign[MXN];
int dist[MXN];
int via[MXN];



int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
  
	n = N, m = M;
	for (int i=0; i<m; i++) {
		int a, b, w;
		a = R[i][0], b = R[i][1], w = L[i];
		adj[a].push_back(make_pair(b, w));
		adj[b].push_back(make_pair(a, w));
	}
	
	priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > pro;
	set<int> poss[MXN];
	bool done[MXN];
	
	for (int i=0; i<MXN; i++) dist[i] = INT_MAX;
	
	int k = K;
	for (int i=0; i<k; i++) {
		int x = P[i];
		ter[x] = true;
		dist[x] = 0;
		done[x] = true;
		pro.push(make_pair(0, x));
	}

	
	while (!pro.empty()) {
		
		int di = pro.top().first;
		int no = pro.top().second;
		pro.pop();
		
		if (dist[no] < di) continue; // do not relax

		if (poss[no].size() >= 2 && !done[no]) {
			// can assign second dist, first relaxing
			done[no] = true;
			dist[no] = *(++poss[no].begin());
		}
		
		if (done[no] && !assign[no]) {
			assign[no] = true;
			// can access adjs
			for (int i=0; i<adj[no].size(); i++) {
				// cout << "add option: " << no << " to " << adj[no][i].first << endl;
				int f = adj[no][i].second;
				poss[adj[no][i].first].insert(f + dist[no]);
				if (poss[adj[no][i].first].size() >= 2) {
					//sort(poss[adj[no][i].first].begin(), poss[adj[no][i].first].end());
					pro.push(make_pair(*(++poss[adj[no][i].first].begin()), adj[no][i].first));
				}
			}
		}

	}
	
	assert(dist[0] != INT_MAX);
	return dist[0];
  
}

Compilation message

crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:60:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for (int i=0; i<adj[no].size(); i++) {
                  ~^~~~~~~~~~~~~~~

Subtask #1
0 / 46.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 7808 KB Output is correct
2 Correct 9 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Incorrect 10 ms 8064 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #2
0 / 43.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 7808 KB Output is correct
2 Correct 9 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Incorrect 10 ms 8064 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 7808 KB Output is correct
2 Correct 9 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Incorrect 10 ms 8064 KB Output isn't correct
5 Halted 0 ms 0 KB -