Algorithm/코드트리

[코드트리] 1차원 젠가 - 파이썬

potato_pizza 2024. 8. 4. 13:21
728x90

1차원 젠가

문제

https://www.codetree.ai/missions/2/problems/jenga-1d?&utm_source=clipboard&utm_medium=text

 

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

국가대표가 만든 코딩 공부의 가이드북 코딩 왕초보부터 꿈의 직장 코테 합격까지, 국가대표가 엄선한 커리큘럼으로 준비해보세요.

www.codetree.ai

 

풀이

  • 인덱스에 따라 결과 리스트에 추가하는 방식
n = int(input())
lst = []
for _ in range(n):
    a = int(input())
    lst.append(a)

s1, e1 = map(int, input().split())

s2, e2 = map(int, input().split())

temp = []
for i in range(n):
    if not (s1-1 <= i <= e1-1):  # s1-1부터 e1-1까지의 인덱스가 아닌 경우만 추가
        temp.append(lst[i])

result = []
for j in range(len(temp)):
    if not (s2-1 <= j <= e2-1):  # s2-1부터 e2-1까지의 인덱스가 아닌 경우만 추가
        result.append(temp[j])

print(len(result))
for k in result:
    print(k)
728x90
반응형