[python] 실수 많이한 문제

https://www.codetree.ai/missions/4/problems/play-with-array?&utm_source=clipboard&utm_medium=text 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

n,q = map(int,input().split()) #n : 원소의 개수 / q: 질의의 수 

arr = list(map(int,input().split())) #원소리스트  # split() 안빼먹게 조심 

for _ in range(q): 
    #b=0
    #action,a,b =map(int,input().split())
    li = list(map(int,input().split()))

    action = li[0]
    a=li[1] #숫자의 범위 잘보기... 
    
    if action ==1: #a번째 원소를 출력
        print(arr[a-1])
    
    elif action ==2: #숫자 a가 있는지를 판단 / 해당 원소가 몇 번째 원소인지를 출력 / 
                     #a가 2개 이상 있다면 index가 더 작은 원소를 출력 / a가 없다면 0을 출력
        if a not in arr:
            print(0)
        else:
            print(arr.index(a)+1) # 값을 1증가시켜야함... 
    else: #a번째 원소부터 b번째 원소까지 순서대로 공백을 사이에 두고 출력
        b= li[2]
        for i in range(a-1,b):
            print(arr[i],end=" ") #원소의 위치 i 가아니라 i 위치에 있는 해당원소 arr[i] 를 출력해야함 
        print()