Error in memory efficient Dijkstra using set

Revision en2, by vaibhavmisra, 2022-05-27 17:24:09

Hello everyone, I have been trying to find shortest path using dijkstra with set by storing indices instead of index-distance pairs with the help of a custom comparator. It is failing on a very large test case and I am having a difficult time debugging it. Where am I doing wrong? Any type of hint would be appreciated.

The Problem My solution

vector<ll> d;
 
struct comp {
	bool operator()(const int& a, const int& b) const {
		return d[a] < d[b];
	}
};
 
void solve() {
	int n, m; cin >> n >> m;
	vector<vector<pii>> graph(n + 1);
	set<int, comp> q;											
	rep(m) {
		int a, b, c; cin >> a >> b >> c;
		graph[a].push_back(make_pair(b, c));
		graph[b].push_back(make_pair(a, c));
	}
	vector<int> p(n + 1, -1);
	d.resize(n + 1, LLONG_MAX);
	d[1] = 0;
	q.insert(1);
	while(!q.empty()) {
		int node = *q.begin();
		q.erase(q.begin());
		for(auto adj : graph[node]) {
			if(d[adj.ff] > d[node] + adj.ss) {
				if(q.find(adj.ff) != q.end()) q.erase(adj.ff); 
				d[adj.ff] = d[node] + adj.ss;
				p[adj.ff] = node;
				q.insert(adj.ff);
			}	
		}
	}
	if(d[n] == LLONG_MAX) cout << -1 << endl;
	else {
		vector<int> path;
		while(n != -1) {
			path.push_back(n);
			n = p[n];
		}
		rep(path.size()) cout << path[path.size() - i - 1] << ' '; cout << endl;
	}
}
Tags dijkstra

History

 
 
 
 
Revisions
 
 
  Rev. Lang. By When Δ Comment
en3 English vaibhavmisra 2022-05-27 17:28:07 7
en2 English vaibhavmisra 2022-05-27 17:24:09 1 Tiny change: '[a] < d[b]\n }\n};\n' -> '[a] < d[b];\n }\n};\n'
en1 English vaibhavmisra 2022-05-27 17:23:11 1481 Initial revision (published)