[2023-01-18 👳‍♂️ TIL ]

오늘은 간단한 백준 1문제 풀었습니다.

5587번 문제
1~30 까지의 번호를 가진 전체학생 중에 과제를 제출하지 않은 학생들의 번호를 찾아 출력하는 문제입니다.

처음에는 아래와 같은 방식으로 코드를 작성하였습니다.

import sys

input = sys.stdin.readline
good_student = []
bad_student = []
total = [int(n) for n in range(1, 31)]
1~30까지의 숫자를 배열로 선언

for num in range(28):
    good_student.append(int(input()))
제출한 학생들을 배열로 저장

for student in total:
    if student not in good_student:
        print(student)
전체 학생중에서 제출하지 않은 학생만 출력

통과는 되었지만 코드가 맘에 안들어서 다시 구현했는데

import sys

input = sys.stdin.readline

total = [int(n) for n in range(1, 31)]
1~30까지의 숫자를 배열로 선언

for num in range(28):
    good_student = int(input())
    index = total.index(good_student)
    total[index] = -1

for find in total:
    if find != -1:
        print(find)

위에 코드도 통과는 되었지만 아마도 학생의 수가 30으로 한정이 되어있다보니 통과가 된것같습니다.
아마 주어진 학생 수가 엄청 많았다면...🙀
과제를 제출한 학생들의 번호를 입력받는 반복문에서 index()함수를 사용하였는데
나중에 찾아보니 시간 복잡도가 O(n) 인 함수였습니다.. 이중 반복문을 쓴 꼴..
코드는 맨위에보다 짧지만 오히려 성능은 좋지 않다는 것