인프런 커뮤니티 질문&답변

조태님의 프로필 이미지
조태

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

5-B : stack을 이용한 풀이

5-B 메모리초과

해결된 질문

작성

·

281

·

수정됨

0

안녕하세요 선생님

문제를 풀던 중 궁금한 내용이 생겨서 질문 드립니다.

#include<bits/stdc++.h>
using namespace std;

string str;
string bstr;
stack<char> st;
int flag;

int main() {
	cin >> str >> bstr;
	for (int i = 0; i < str.length(); i++) {
		st.push(str[i]);
		if (st.size() >= bstr.size() && st.top() == bstr[bstr.length() - 1]) { // 마지막이랑 똑같으면
			string istr = "";
			for (int a = 0; a < bstr.length(); a++) {
				istr += st.top();
				st.pop();
			}
			reverse(istr.begin(), istr.end());
			if (istr != bstr) {
				for (int b = 0; b < istr.length(); i++) {
					st.push(istr[b]);
				}
			}
		}
	}
	str = "";
	if (st.size() == 0) {
		cout << "FRULA" << "\n";
	}
	else {
		while (st.size()) {
			str += st.top(); st.pop();
		}
		reverse(str.begin(), str.end());
		cout << str << "\n";
	}
	return 0;
}

제가 처음으로 작성한 코드입니다.

for(char a : str) st.push(a) 같이 범위 기반 for문 대신

for (int i = 0; i < str.length(); i++) st.push(str[i]) 같이 원래의 for문 방식을 사용했다가

메모리 초과 오류를 겪었습니다.

두 방법이 어째서 메모리 차이가 많이 나는지 궁금합니다!

항상 감사합니다 좋은 하루 되세요 :)

답변 1

0

큰돌님의 프로필 이미지
큰돌
지식공유자

안녕하세요 조태님 ㅎㅎ

일단 이부분이요.

			for (int b = 0; b < istr.length(); i++) {
					st.push(istr[b]);

b++이 되어야 하지 않을까요?

제 생각에는 이부분 때문에 그런 거 같습니다.

두개의 코드 다 맞다고 뜹니다.

#include<bits/stdc++.h>
using namespace std;

string str;
string bstr;
stack<char> st;
int flag;

int main() {
	cin >> str >> bstr;
    for(int i = 0; i < str.size(); i++){
		st.push(str[i]);
		if (st.size() >= bstr.size() && st.top() == bstr[bstr.length() - 1]) { // 마지막이랑 똑같으면
			string istr = "";
			for (int a = 0; a < bstr.length(); a++) {
				istr += st.top();
				st.pop();
			}
			reverse(istr.begin(), istr.end());
			if (istr != bstr) {
				for (int b = 0; b < istr.length(); b++) {
					st.push(istr[b]);
				}
			}
		}
	}
	str = "";
	if (st.size() == 0) {
		cout << "FRULA" << "\n";
	}
	else {
		while (st.size()) {
			str += st.top(); st.pop();
		}
		reverse(str.begin(), str.end());
		cout << str << "\n";
	}
	return 0;
}

 

 

#include<bits/stdc++.h>
using namespace std;

string str;
string bstr;
stack<char> st;
int flag;

int main() {
	cin >> str >> bstr;
    for(char bb : str){
		st.push(bb);
		if (st.size() >= bstr.size() && st.top() == bstr[bstr.length() - 1]) { // 마지막이랑 똑같으면
			string istr = "";
			for (int a = 0; a < bstr.length(); a++) {
				istr += st.top();
				st.pop();
			}
			reverse(istr.begin(), istr.end());
			if (istr != bstr) {
				for (int b = 0; b < istr.length(); b++) {
					st.push(istr[b]);
				}
			}
		}
	}
	str = "";
	if (st.size() == 0) {
		cout << "FRULA" << "\n";
	}
	else {
		while (st.size()) {
			str += st.top(); st.pop();
		}
		reverse(str.begin(), str.end());
		cout << str << "\n";
	}
	return 0;
}

 

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.


조태님의 프로필 이미지
조태
질문자

아핰ㅋㅋㅋㅋㅋ 그런 오류를 겪었군요.. 감사합니다!

조태님의 프로필 이미지
조태

작성한 질문수

질문하기