Submission #211369

# Submission time Handle Problem Language Result Execution time Memory
211369 2020-03-20T08:16:38 Z socho Crocodile's Underground City (IOI11_crocodile) C++14
89 / 100
2000 ms 121344 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;
	multiset<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++) {
                  ~^~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 9 ms 7808 KB Output is correct
2 Correct 8 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Correct 9 ms 7936 KB Output is correct
5 Correct 10 ms 7936 KB Output is correct
6 Correct 10 ms 7936 KB Output is correct
7 Correct 10 ms 7936 KB Output is correct
8 Correct 9 ms 7936 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 9 ms 7808 KB Output is correct
2 Correct 8 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Correct 9 ms 7936 KB Output is correct
5 Correct 10 ms 7936 KB Output is correct
6 Correct 10 ms 7936 KB Output is correct
7 Correct 10 ms 7936 KB Output is correct
8 Correct 9 ms 7936 KB Output is correct
9 Correct 14 ms 8448 KB Output is correct
10 Correct 10 ms 7936 KB Output is correct
11 Correct 11 ms 8064 KB Output is correct
12 Correct 19 ms 9088 KB Output is correct
13 Correct 17 ms 9088 KB Output is correct
14 Correct 9 ms 7808 KB Output is correct
15 Correct 10 ms 8064 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 9 ms 7808 KB Output is correct
2 Correct 8 ms 7808 KB Output is correct
3 Correct 9 ms 7808 KB Output is correct
4 Correct 9 ms 7936 KB Output is correct
5 Correct 10 ms 7936 KB Output is correct
6 Correct 10 ms 7936 KB Output is correct
7 Correct 10 ms 7936 KB Output is correct
8 Correct 9 ms 7936 KB Output is correct
9 Correct 14 ms 8448 KB Output is correct
10 Correct 10 ms 7936 KB Output is correct
11 Correct 11 ms 8064 KB Output is correct
12 Correct 19 ms 9088 KB Output is correct
13 Correct 17 ms 9088 KB Output is correct
14 Correct 9 ms 7808 KB Output is correct
15 Correct 10 ms 8064 KB Output is correct
16 Execution timed out 2093 ms 121344 KB Time limit exceeded
17 Halted 0 ms 0 KB -