글 읽기 - 이 코드에서 시간 초과가 걸릴 만한 요소가 있을까요?

system_error_code_0 2년 전 0

각 정점마다 priority_queue를 이용한 다익스트라 코드를 짰습니다.

하지만 질문글과 인터넷을 뒤져봐도, 분명 알고리즘은 맞는데 계속해서 시간초과가 뜹니다...

혹시 제가 놓치고 있는 시간 초과가 발생하는 요소가 있을까요?

부탁드립니다...

#include <iostream>
#include <queue>
using namespace std;
typedef pair<int,int> p;
int n,m,k;
priority_queue<int> pq[1005];
vector<p> path[1005];
bool operator<(p a,p b) {
	return a.second>b.second;
}

void f() {
	priority_queue<p> dijk;
	dijk.push(p(1,0));
	pq[1].push(0);
	while(!dijk.empty()) {
		int x=dijk.top().first;
		int y=dijk.top().second;
		dijk.pop();
		for(int i=0;i<path[x].size();i++) {
			int xi=path[x][i].first;
			int yi=path[x][i].second+y;
			if(pq[xi].size()<k) {
				pq[xi].push(yi);
				dijk.push(p(xi,yi));
			} else if(pq[xi].top()>yi) {
				pq[xi].pop();
				pq[xi].push(yi);
				dijk.push(p(xi,yi));
			}
		}
	}
}

int main() {
	int a,b,c;
	cin>>n>>m>>k;
	for(int i=0;i<m;i++) {
		scanf("%d %d %d",&a,&b,&c);
		path[a].push_back(p(b,c));
	}
	f();
	for(int i=1;i<=n;i++) {
		if(pq[i].size()<k) printf("-1\n");
		else {
			printf("%d\n",pq[i].top());
		}
	}
}

system_error_code_0 2년 전 3

결국 혼자 해결했습니다.

더 다양한 시도를 해보고 질문을 올려야했었던 것 같습니다.

일단 operator 정의 부분을 없애

x를 -cost, y를 next_city로서 큐에 저장하면 되는 것으로 보아,

operator를 새로 정의한 것이 연산 시간을 훨씬 오래 걸리게 만드는 것이라고 생각되었습니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

이 코드에서 시간 초과가 걸릴 만한 요소가 있을까요?

system_error_code_0 2년 전 0

system_error_code_0 2년 전 3

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말