inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

55. 기차운행 (스택 자료구조 응용)

제가 작성한 코드 한번 봐주실 수 있나요?

218

할루

작성한 질문수 7

0

먼저 풀어봤는데, 채점은 맞았습니다만 코드가 어떤지 여쭤보고 싶어서 올립니다

#include <stdio.h>
#include <stack>
#include <vector>
using namespace std;

int main() {
	int i, n, idx = 0, tmp = -1;
	stack<int> st;
	scanf("%d", &n);
	
	vector<char> res;
	vector<int> arr(n);
	
	for (i = 0; i < n; i++) {
		scanf("%d", &arr[i]);
	}
	i = 0;
	while (i < n) {
		if (st.empty()) {
			st.push(arr[i++]);
			res.push_back('P');
		}
		else {
			if (arr[i] < st.top()) {
				st.push(arr[i++]);
				res.push_back('P');
			}
			else if (st.top() > tmp) {
				tmp = st.top();
				st.pop();
				res.push_back('O');
			}
		}
	}
	while (!st.empty()) {
		if (st.top() < tmp) {
			printf("impossible");
			return 0;
		}
		tmp = st.top();
		st.pop();
		res.push_back('O');
	}

	for (i = 0; i<res.size(); i++) printf("%c", res[i]);
}

tmp 변수는 직전에 pop한 숫자를 담기 위한 용도이고,

pop하기 전에 이 tmp 값보다 큰지 확인하고자 했습니다.

괜히 뺑 돌아가서 문제를 푼 것 같은데 그래도 로직이 맞는지 궁금합니다. 감사합니다!

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

답변 1

1

김태원

안녕하세요^^

잘 하신 코드입니다.

이렇게 자신이 먼저 문제를 도전해 보고 제 영상과 비교해서 개선할 점은 스스로 개선하면서 발전하기시 바랍니다. 

테스트 케이스 질문

0

367

1

병합정렬 시간복잡도 질문

0

458

1

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

0

1339

2

질문드립니다.

0

371

1

질문드립니다!

0

425

1

dev 프로그램 질문

0

270

1

문제가 이해가 안되요

0

371

1

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

0

301

1

source file not compiled

0

1029

3

59번 질문드립니다.

0

366

1

25번 문제 질문

0

343

1

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

0

364

1

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

0

463

1

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

0

353

1

75번, 79번 priority_queue관련

1

351

1

75.최대 수입 스케줄

0

390

2

복면산 정답의 수

0

424

1

테스트 케이스에 대해서

0

438

1

수업 내용 질문입니다!

1

226

1

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

0

812

2

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

0

248

1

다른 풀이 방식

0

310

1

크루스칼 vs 프림

0

300

1

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

0

231

1