글 읽기 - 메모리 초과 어떻게 해결해야 할까요?

banana970 1년 전 0

메모리 초과가 발생하는데 어떻게 해결해야 할지 도무지 모르겠습니다...

방법이 있을까요? ㅜㅜ

#include <iostream>
#include <vector>
#include <map>
#include <utility>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;

vector <vector<int>> V(5001);
map <pair<int, int>, int> M;
queue <int> Que;
int N, Q;
int p, q, r, k, v;



int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> N >> Q;
	for (int i = 0; i < N - 1; i++) {
		cin >> p >> q >> r;
		V[p].push_back(q);
		V[q].push_back(p);
		M[make_pair(p, q)] = r;
		M[make_pair(q, p)] = r;
	}
	int cnt;
	for (int i = 0; i < Q; i++) {
		int visit[5001] = {};
		cin >> k >> v;
		Que.push(v);
		cnt = 0;
		int FIRST = v;
		while (Que.empty() == 0) {
			int temp = Que.front();
			visit[temp] = 1;
			Que.pop();
			for (int i = 0; i < V[temp].size(); i++) {
				if (visit[V[temp][i]] == 1) {
					continue;
				}
				if (temp == FIRST) {
					continue;
				}
				else if (M.find(make_pair(FIRST, V[temp][i])) != M.end()) {
					M[make_pair(FIRST, V[temp][i])] = min({ M[make_pair(FIRST,V[temp][i])],M[make_pair(FIRST, temp)], M[make_pair(temp, V[temp][i])] });
				}
				else {
					M[make_pair(FIRST, V[temp][i])] = min(M[make_pair(FIRST, temp)], M[make_pair(temp, V[temp][i])]);
				}
				if (M[make_pair(FIRST, V[temp][i])] >= k) {
					Que.push(V[temp][i]);
				}
			}
		}
		for (int i = 1; i <= N; i++) {
			if (i != v) {
				if (M[make_pair(v, i)] >= k) {
					cnt++;
				}
			}
		}
		cout << cnt << '\n';
	}
}

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

문제

문제

출처

ICPC

메모리 초과 어떻게 해결해야 할까요?

banana970 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말