[Alogorithm] D-10 구현

치킨문제

# https://www.acmicpc.net/problem/15686
from itertools import combinations

n, m = map(int, input().split())
data = []
for i in range(n):
    data.append(list(map(int, input().split())))

two = []
one = []
for i in range(n):
    for j in range(n):
        if data[i][j] == 2:
            two.append([i, j])
        if data[i][j] == 1:
            one.append([i, j])

chicken = list(combinations(two, m))
result = [0] * len(chicken)
for i in one:
    for j in range(len(chicken)):
        a = 100
        for k in chicken[j]:
            temp = abs(k[0] - i[0]) + abs(k[1] - i[1])
            a = min(a, temp)
        result[j] += a

print(min(result))

1 집 2 치킨 좌표를 따로 받음
combinations(two, m) 치킨집중 m 개 조합

자물쇠와 열쇠

아무리 봐도 모르겠음...

제로

# https://www.acmicpc.net/problem/10773

n = int(input())
data = []
for i in range(n):
    a = int(input())
    if a == 0:
        data.pop()
    else:
        data.append(a)

print(sum(data))

실버4인데 아마 다른 언어는 스택을 구현해서 어려운건가..?
파이썬은 스택이 list로 내장함수가 존재해 편하게 풀었다.