[class 1]백준 X보다 작은 수(10871번) 풀이 (C++/Python)

문제 : X보다 작은 수(10871번)(Bronze V)

문제 링크 : https://www.acmicpc.net/problem/10871
출처 : Baekjoon Online Judge

 

저번 게시글에서 약간 수정하면 됩니다.

#include <bits/stdc++.h>
 
using namespace std;
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, x;
    cin >> n >> x;

    int a[n];
    
    for(int i = 0; i < n; i++) {
        int d;
        cin >> d;
        a[i] = d;
    }

    for(int i = 0; i < n; i++) {
        if(a[i] < x) cout << a[i] << " ";
    }
 
    return 0;
}

참고로 출력이 "1 4 2 3 "으로 맨 끝에 스페이스 바 한 개 있어도 정답 처리됩니다.

 

Python은 기본적으로 배열 대신 리스트로 구현합니다.(그게 더 편안합니다.)

import sys

input = sys.stdin.readline

N, X = map(int, input().split())
L:list = list(map(int, input().split()))

result:list = []
for i in L :
    if i < X :
        result.append(i)

print(*result)

여기서 *는 *args로 파이썬에서 여러 개의 인자로 함수를 호출하는 것으로 주로 함수 내부에 여러 개의 인자를 매개변수로 호출할 때 사용합니다.

여기서는 print()를 활용하여 여러 개의 인자를 스페이스 바 한 개로 띄워서 출력하게 만듭니다.

result = [1, 4, 2, 3]인 경우 "1 4 2 3"로 출력됩니다.