Linked List - 수열 합치기

for~ else문을 사용했다.
for ~ else문은 for문을 모두 돌 때까지 break가 되지 않으면 else 구문이 실행되는 형태이다.

  • 리스트 중간에 다른 리스트를 정상적으로 삽입하기 위해서는
    기존리스트[삽입할 위치:삽입할 위치] = 삽입할 리스트
    의 코드를 사용할 수 있다.
    insert 함수를 사용한다면
    [1,2,3,[4,5,6],7,8]와 같이 리스트 속에 리스트가 생성된다.
for tc in range(1,int(input())+1):
    N,M = map(int,input().split())
    arr=[list(map(int,input().split())) for _ in range(M)]
    res=arr.pop(0)
    
    for i in arr:
        chk=i[0]
        for j in range(len(res)):
            if chk<res[j]:
                res[j:j]=i
                break
        else:
            res.extend(i)
                
    if len(res)>10:
        res=res[-1:-10-1:-1]
    else:
        arr[::-1]
    
    print(f'#{tc}', end=' ')
    print(*res)

좋은 웹페이지 즐겨찾기