백준 BAEKJOON 28289번 과 조사하기 [PYTHON/파이썬]

 

 

 

 

 

백준 BAEKJOON 28289번 과 조사하기 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅲ)

https://www.acmicpc.net/problem/28289

28289번: 과 조사하기

 

 

 

 

<풀이>

먼저 0이 4개가 채워진 리스트를 만들었다.

('소프트웨어개발과', '임베디드소프트웨어개발과', '인공지능소프트웨어개발과', '아무런 과에도 속하지 않는 1학년'의 수를 차례대로 담기 위함)

 

학생의 정보가 G, C, N 으로 주어지는데 문제를 푸는데 있어서는 사실상 N은 필요가 없는 정보이다.

오직 G와 C를 이용하여 if문으로 조건 처리해주어 풀었다.

 

 

 

 

 

<코드>

P = int(input())
result = [0] * 4

for _ in range(P):
    student_info = list(map(int, input().split()))

    # 1학년인 경우
    if student_info[0] == 1:
        result[3] += 1
        continue

    # 2, 3학년인 경우
    if student_info[1] == 1 or student_info[1] == 2:
        result[0] += 1
    elif student_info[1] == 3:
        result[1] += 1
    else:
        result[2] += 1

for r in result:
    print(r)