본문 바로가기
Algorithm(CodeTree, Python)/완전탐색1

[코드트리 자리 마다 숫자를 정하는 완전탐색] 한 가지로 열리는 자물쇠python

by kurooru 2022. 12. 11.
# n 입력
n = int(input())
# password 입력
password = list(map(int, input().split()))

# 함수들
# is_open(a, b, c)
def is_open(a, b, c):
    # 하나라도 차가 2 이하면 열림
    return abs(a - password[0]) <= 2 or abs(b - password[1]) <= 2 or abs(c - password[2]) <= 2

# 설계
# cnt
cnt = 0

# 완전 탐색 시작
for i in range(1, n+1):
    for j in range(1, n+1):
        for k in range(1, n+1):
            # 열리면,
            if is_open(i, j, k):
                # cnt에 추가
                cnt += 1

# 출력
print(cnt)