[Python] Basic Data Types - Lists

Summary

파이썬의 리스트에 대해서 얼마나 이해하고 있는지 확인할 수 있고
리스트에서 사용할 수 있는 기본적인 함수들을 사용해서 풀 수 있는 문제였다.

문제에서 필요로 하는 옵션은 다음과 같았다.

• insert
• append
• remove
• pop
• sort
• reverse

이런 옵션을 사용해서 리스트에 추가, 삭제, 정렬 등의 기능을 수행하는 문제였다.

 

그리고 문자를 입력해 문자열을 나누어 명령어와 값 그리고 인덱스를 구분해서 사용해주어야 했다.
이 부분은 다음과 같이 split() 을 사용해서 문자열을 나누어 구분해줄 수 있었다.

cmd = input_str.split()

split 함수는 기본적으로 공백을 기준으로 문자열을 나누어준다.
원하는 문자로 나누고 싶으면 다음과 같이 괄호 () 안에 문자열을 나누어줄 특정 문자를 입력해주면 된다.

split([특정 문자 입력])

또한 나누어진 문자열은 배열 형태로 나누어져 아래와 같이 배열의 값을 가져오는 것처럼 사용할 수 있었다.

str = "Jay Log"

str_split = str.split()

print(str_split[0])
print(str_split[1])

# 결과
Jay
Log

 

이 외에도 split 을 다양하게 사용해볼 수 있다.

 

Problem

 

 

Submit Code

if __name__ == '__main__':
    N = int(input())
    
    arr = []
    
    for i in range(0, N):
        input_str = input()
        
        cmd = input_str.split()
    
        if cmd[0] == "insert":
            arr.insert(int(cmd[1]), int(cmd[2]))
        elif cmd[0] == "remove":
            arr.remove(int(cmd[1]))
        elif cmd[0] == "append":
            arr.append(int(cmd[1]))
        elif cmd[0] == "sort":
            arr.sort()
        elif cmd[0] == "pop":
            arr.pop()
        elif cmd[0] ==  "reverse":
            arr.reverse()
        elif cmd[0] ==  "print":
            print(arr)

 

 

Result