글 읽기 - 이 코드가 맞는 이유가 궁금합니다

dhsg0717 2년 전 0

어느 분의 코드를 참고해서 작성한 코드입니다

이 방식이면 마지막 수열을 입력 받을 때 뒷부분이 입력이 안 받아질 수도 있는데

입력이 버퍼에 들어가는 방식이 엔터를 기준으로 한 줄이 통째로 들어가서

답을 내기 위해 남은 개수 만큼만 버퍼에서 가져오고 나머지 뒷부분은 버퍼에 남아있게 되어서

잘 작동하는 것인가요?

#include<iostream>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<sstream>
#include<limits>
#include<algorithm>
#include<stack>
#include<list>
#include<queue>
#include<map>
#include<numeric>

using namespace std;

int n;
int m;
int A[100001];
int B[100001];
vector<int> ret;

void solve() {
    int count = 0;
    for(int i = n - 1; i >= 0; i--) {
        if(!A[i] && m) {
            ret.push_back(B[i]);
            m--;
        }
    }
    while(m--) {
        int temp;
        cin >> temp;
        ret.push_back(temp);
    }
}

int main(void)
{
    cin.tie(NULL);
    ios_base::sync_with_stdio(false);
  
    cin >> n;
    for(int i = 0; i < n; i++) 
        cin >> A[i];
    for(int i = 0; i < n; i++) 
        cin >> B[i];
    cin >> m;

    solve();
        
    for(int i = 0; i < ret.size(); i++) 
        cout << ret[i] << ' ';  

    return 0;
}

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

문제

문제

출처

ICPC

이 코드가 맞는 이유가 궁금합니다

dhsg0717 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말