백준(BOJ) #20125 쿠키의 신체측정(Python) 이중 List 다루기 재활

알고리즘 문제 풀이시 기본이 되는 것은 이중 List를 다루는 것이다.

입력받는 것부터 기억나지 않아 Python 기본을 다시 시작해보자면

3
1 2 3
4 5 6
7 8 9

3
123
456
789

name = list(map(int, input().split()))

을 통해서 int형태로 받을 수 있다. string이 이어져있는 경우는 split 없이 받을 수 있다.

list에서 index와 for문이 같이 필요한 경우에는 enumerate를 사용하면 된다.

⇒ tuple 형태로 변환

⇒ start를 사용해서 index를 0 이 아닌 1부터 사용가능하다.

nums = ["A", "B", "C"]
for i, num in enumerate (nums): # 0 A 1 B 2 C
    print(i, num, end=" ")

for i, num in enumerate (nums, start=1): # 1 A 2 B 3 C
    print(i, num, end=" ")

for  num in enumerate (nums): #(0, 'A') (1, 'B') (2, 'C')
    print(num,end=" ")

특정한 열을 기준으로 가져오기

zip 사용하기

A = [
    [1,2,3],[4,5,6],[7,8,9],[10,11,12]
     ]

b =  list(zip(*A))
c =  list(zip(*A))[1][0]
print(b) #[(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]
print(c) #2

Uploaded by N2T