inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

it 취업을 위한 알고리즘 문제풀이 입문 (with C/C++) : 코딩테스트 대비

39. 두 배열 합치기(병합정렬 예비학습)

배열이 아니라 벡터를 이용해봤는데, 시간 제한 초과가 나는 이유가 궁금합니다.

315

jarchive

작성한 질문수 5

1

안녕하세요 선생님.  

선생님께서는 39번 문제 풀때 배열을 이용하셨지만 제가 먼저 풀어볼때는 벡터를 썼고 문제의 입출력예시는 알맞게 동작을 했었습니다. 그러나 채점폴더에서는 테스트케이스 5개 전부 시간초과가 났습니다. 

제 코드는 아래와 같습니다. 

어떤 부분이 잘못된 것이고 어떻게 고쳐야 할지 궁금합니다.

감사합니다. 

int main(int argc, char** argv){
	
	// 배열의 크기 저장, 인덱스를 저장하는 변수 3개 (p) 셋 다 1로 시작함.  
	int a_size, b_size, arr_size, i, p1=1, p2=1, p3=1; 
	
	//freopen("input.txt", "rt", stdin);
	
	// 필요한 배열 3개 : 입력받는 배열 arr_a, arr_b, 결과배열 res 
	 
	// 입력받기 시작  
	scanf("%d", &a_size);
	vector<int> arr_a(a_size);
		
	for(i=1; i<=a_size; i++){  // 벡터에 숫자를 입력받는 인덱스를  1 부터 하고 있음.  
		scanf("%d", &arr_a[i]);
	}
	
	scanf("%d", &b_size);
	vector<int> arr_b(b_size);
		
	for(i=1; i<=b_size; i++){ 
		scanf("%d", &arr_b[i]);
	}// 입력받기 끝  
	
	// 결과 저장할 벡터  
	vector<int> res(a_size+b_size);

	// 인덱스 변수 2개를 가지고 작은 수가 나올 때 마다 p1 또는 p2를 이동시킴. 
	while(p1 <= a_size && p2 <= b_size){ // 어느 한 쪽이 자기 크기보다 커질 때까지. 
		
		if(arr_a[p1] < arr_b[p2]){
			res[p3++] = arr_a[p1++]; 
			// p3를 대입한 후에, 1증가 해준다. res[p3]; p3+=1; 과 같다.  
		}else{
			res[p3++] = arr_b[p2++]; 
		}
	}
	
	// 한 쪽은 다 res에 들어가고, 다른 한 쪽은 res에 못들어가고 남았을 것. 
	while(p1 <= a_size) res[p3++] = arr_a[p1++];
	while(p2 <= b_size) res[p3++] = arr_b[p2++];

	for(i=1; i<=a_size+b_size; i++){ // 출력 할 때도 인덱스 1부터 시작. 
		printf("%d ", res[i]);
	}
	
	return 0;
}

C++ 코테 준비 같이 해요!

답변 1

2

김태원

안녕하세요^^

인덱스 번호를 1번부터 사용하려면 벡터선언할 때 크기를   vector<int> arr_a(a_size+1); 와 같이 하세요. 다른 벡터도 마찬가지입니다.                        

테스트 케이스 질문

0

373

1

병합정렬 시간복잡도 질문

0

462

1

41.연속된 자연수의 합 문제풀이에서 수학적인 원리를 모르고 있습니다.

0

1345

2

질문드립니다.

0

376

1

질문드립니다!

0

430

1

dev 프로그램 질문

0

275

1

문제가 이해가 안되요

0

376

1

4번 나이차이 문제 접근법 질문 드립니다.

0

307

1

source file not compiled

0

1047

3

59번 질문드립니다.

0

372

1

25번 문제 질문

0

349

1

4. 나이차이 문제 질문입니다.

0

372

1

90번 라이언 킹 심바 1번 테스트 케이스

0

470

1

71번 문제 전역 변수 질문 있습니다

0

365

1

75번, 79번 priority_queue관련

1

355

1

75.최대 수입 스케줄

0

400

2

복면산 정답의 수

0

431

1

테스트 케이스에 대해서

0

445

1

수업 내용 질문입니다!

1

232

1

풀어보면 좋은 문제 목록 - 2580 스토쿠 DFS 질문입니다!!

0

822

2

12. 플로이드-와샬(그래프 최단거리) . 27:25초

0

254

1

다른 풀이 방식

0

317

1

크루스칼 vs 프림

0

306

1

숫자 총개수 small 질문있습니다.

0

242

1