글 읽기 - 정답 풀이랑 거의 같은데 왜 틀렸을까요? ㅠㅠ(풀이 둘다 올림)

yg04076 1년 전 1

테스트 예제 돌아가는 것도 둘이 똑같고 논리도 똑같은데... 뭐가 문제 일까요...?

그리고 제가 이상한 지점이라고 마킹한 지점....

if (st.size() != 0) {
for (int i = 0; i <= st.size(); i++) {
sum += st.top();
st.pop();
}
}

제가 초반에 쓴 풀인데... 이상하게 for문이 st.size()만큼 돌면 정답이 나와야 하는데, 이상하게 st.size()-1만큼만 돌더라구요

st.size()+1만큼 돌리니 예제 케이스가 잘 출력되던데.. 이건 왜 이런걸까요? (물론 이것도 예제 케이스는 출력이 잘되는데 틀렸다고 나오더라고요..)

#include <iostream> // 정답 처리된 블로그 풀이
#include <stack>
using namespace std;

int main() {
    int k, n;

    cin >> k;

    stack<int> s;

    for (int i = 0; i < k; i++) {
        cin >> n;
        if (n != 0) {
            s.push(n);
        }
        else {
            s.pop();
        }
    }

    int sum = 0;
    int size = s.size();

    for (int i = 0; i < size; i++) {
        sum += s.top();
        s.pop();
    }

    cout << sum;

    return 0;
}
/////////////////////////////////////////////////////////////////////
int main() {  // 제 풀이
    int N, input;
    int sum = 0;
    
    stack<int> st;

    cin >> N;

    while (N--) {
        cin >> input;
        if (input != 0) {
            st.push(input);
        }
        else {
            st.pop(); 
        }
    }
    
    if (st.size() != 0) { // size가 0이면 그냥 0이니 연산 필요없다
        for (int i = 0; i < st.size()-1; i++) { ////// 이상한 지점
            sum += st.top();  
            st.pop();
        }
        sum += st.top();
    }
   

    cout << sum;
}

seawon0808 1년 전 2

pop을 하면 size가 바뀌어서 틀립니다.

shs0911 1년 전 1

53번째 줄 전까지 스택에 n개의 원소가 남아있다고 해봅시다.

그러면 이 n개의 원소를 모두 더해야 답이겠죠. 남아있는 원소들은 모두 마지막 0이 나온 이후의 원소들일테니까요.

이 때, 스택의 사이즈는 n입니다.

매 반복문마다 i를 증가시키고, stack에서 pop을 하므로, 각 변수 및 함수 리턴값은 다음과 같습니다. 편의상 n을 6이라고 가정해보겠습니다.

i = 0, st.size() = 6

i = 1, st.size() = 5

i = 2, st.size() = 4

i = 3, st.size() = 3

.....

과연 n개의 원소를 모두 출력할 수 있을까요?

yg04076 1년 전 1

헉... 그걸 간과하고 있었네요ㅠㅠ 바로 해결했습니다 매우 감사합니다!!!!

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

문제

문제

출처

ICPC

정답 풀이랑 거의 같은데 왜 틀렸을까요? ㅠㅠ(풀이 둘다 올림)

yg04076 1년 전 1

seawon0808 1년 전 2

shs0911 1년 전 1

yg04076 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말