백준 5557번 1학년: Python 풀이 및 정답 코드

문제 링크

https://www.acmicpc.net/problem/5557

5557번: 1학년

 

 

 

문제 설명

문제

상근이가 1학년 때, 덧셈, 뺄셈을 매우 좋아했다. 상근이는 숫자가 줄 지어있는 것을 보기만 하면, 마지막 두 숫자 사이에 '='을 넣고, 나머지 숫자 사이에는 '+' 또는 '-'를 넣어 등식을 만들며 놀고 있다. 예를 들어, "8 3 2 4 8 7 2 4 0 8 8"에서 등식 "8+3-2-4+8-7-2-4-0+8=8"을 만들 수 있다.

상근이는 올바른 등식을 만들려고 한다. 상근이는 아직 학교에서 음수를 배우지 않았고, 20을 넘는 수는 모른다. 따라서, 왼쪽부터 계산할 때, 중간에 나오는 수가 모두 0 이상 20 이하이어야 한다. 예를 들어, "8+3+2-4-8-7+2+4+0+8=8"은 올바른 등식이지만, 8+3+2-4-8-7이 음수이기 때문에, 상근이가 만들 수 없는 등식이다.

숫자가 주어졌을 때, 상근이가 만들 수 있는 올바른 등식의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 숫자의 개수 N이 주어진다. (3 ≤ N ≤ 100) 둘째 줄에는 0 이상 9 이하의 정수 N개가 공백으로 구분해 주어진다.

출력

첫째 줄에 상근이가 만들 수 있는 올바른 등식의 개수를 출력한다. 이 값은 263-1 이하이다.

 

 

 

문제 풀이

1. 각 연산에 나올 수 있는 숫자에 대한 범위 조건이 있고 범위가 작음(0이상 20이하)
2. 각 연산에서 가능한 분기는 두개임(빼거나, 더하거나)
-> dp 배열을 연속된 숫자(0-20)*연산의 수(N-1개), 즉 21*(N-1) 사이즈의 2차원 배열로 만들고 각 원소엔 numbers[i]까지 연산했을 떄 숫자 j를 만들 수 있는 경우의 수 를 저장한다.

12865번 평범한 배낭 문제와 똑같다.
https://he11owor1d.tistory.com/89

백준 12865번 평범한 배낭: Python 풀이 및 정답 코드

문제의 첫번째 예제를 돌려보면 dp는 이렇게 만들어진다.

코드는 다음과 같다.

 

 

 

코드

N = int(input())
numbers = list(map(int, input().split()))

dp = [[0] * 21 for _ in range(N - 1)]
dp[0][numbers[0]] = 1

for i in range(1, N - 1):
    for j in list(filter(lambda x: dp[i - 1][x] != 0, range(0, 21))):
        for x in [j - numbers[i], j + numbers[i]]:
            if 0 <= x <= 20:
                dp[i][x] = dp[i][x] + dp[i - 1][j]

print(dp[-1][numbers[-1]])

 

 

결과

 

 

다른 사람의 코드

 

 

 

리뷰

이전 문제랑 똑같아서 10분 정도 걸림