글 읽기 - 15789번 질문 있습니다!

hjy3691 4년 전 0

이 친구 왜 런타임 에러가 자꾸 뜨는지 궁금합니다. 몇 번 검토해봤는데 어디서 배열초과가 나는지 모르겠습니다ㅜ

#include <iostream>
#include <algorithm>
#include <fstream>
#include <cstring>
#include <vector>
#include <map>
#include <stack>

using namespace std;
const int MAXN = 100002;

typedef vector<int> vi;
typedef vector <pair<int, int>> vpii;

int N, M, CTP, HAN, K;
int CTPs;
bool canAlly[MAXN];
vi ally[MAXN];
vpii sum;

void init() {
	ifstream input;
	input.open("Text.txt");

	cin >> N >> M;

	int x, y;
	for (int i = 0; i < M; i++) {
		cin >> x >> y;
		ally[x].push_back(y);
		ally[y].push_back(x);
	}

	cin >> CTP >> HAN >> K;

	//cout << "sizeof: " << sizeof(canAlly)<<" "<<endl;
	memset(canAlly, true, sizeof(canAlly));
	canAlly[HAN] = false;

	//cout << "allyHan의 size: " << ally[HAN].size() << endl;
	for (int i = 0; i< ally[HAN].size(); i++) {
		canAlly[ally[HAN][i]] = false;
	}

	//cout << "allyCTP의 size: " << ally[CTP].size() << endl;
	canAlly[CTP] = false;
	for (int i = 0; i<ally[CTP].size(); i++) {
		canAlly[ally[CTP][i]] = false;
	}

	CTPs = ally[CTP].size() + 1;
	//cout <<"CTPs 초기화: "<< CTPs << endl;
}

bool cmp(pair<int, int> a, pair<int , int> b){
	return a.first > b.first;
}

void alliance() {
	for (int i = 1; i <= N; i++) {
		if (canAlly[i]) {
			sum.push_back(make_pair(ally[i].size() + 1, i));
		}
	}

	sort(sum.begin(), sum.end(), cmp);

	for (int i = 0; i < K; i++) {
		if (sum.empty() || sum[0].first == -1) {
			break;
		}
		//cout << "i: " << i << endl;
		
		CTPs += sum[0].first;
		sum[0].first = -1;

		for (int j = 0; j < ally[sum[0].second].size(); j++) {
			int x = ally[sum[0].second][j];
			//cout << "j: " << j << ", sum: " << sum.size() << ", x :" << x << endl;

			for (int k = 0; k < sum.size(); k++) {

				if (sum[k].second == x) {
					//for (int i = 0; i < sum.size(); i++) {
					//	cout << sum[i].first << "x" << sum[i].second << "   ";
					//}
					//cout<< endl;
					sum[k].first = -1;
					break;
				}
			}
			//cout << "x: " <<x << endl;
		}
		sort(sum.begin(), sum.end(), cmp);
	}
}

int main() {
	init();
	alliance();
	cout << CTPs << endl;

	system("pause");
	return 0;
}

djm03178 4년 전 1

system("pause"); 를 하면 안 됩니다.

hjy3691 4년 전 0

아..... 정말 감사합니다...

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

문제

문제

출처

ICPC

15789번 질문 있습니다!

hjy3691 4년 전 0

djm03178 4년 전 1

hjy3691 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말