[python] 백준 - 1309. 동물원

🤔문제 해결

1. S1 | DP

2. 사자가 없는 경우, 사자가 왼쪽에 있는 경우, 사자가 오른쪽에 있는 경우의 리스트를 만든다.

3. i번째에서 i-1번째에 사자의 상태에 따라 i번째의 경우의 수가 결정된다.

 (1) i번째에 사자가 없는 경우는 i-1번째에 사자가 왼쪽, 오른쪽 그리고 없어도 된다.

 (2) i번째에 사자가 왼쪽에 있는 경우는 i-1번째에 사자가 오른쪽에 있는 경우와 없는 경우만 가능하다.

 (i-1번째에 왼쪽에 있으면 i번째에서는 왼쪽에 있을 수 없다)

 (3) 오른쪽도 마찬가지

4. 경우의 수를 점차 더해나가면서 마지막 경우의 수를 다 더해준다.

 

💨 마지막에만 9901로 나누려고하면 메모리초과가 발생한다. 값을 쌓을 때 부터 나눠주면서 하자.

 

💻소스 코드

if __name__ == "__main__":
    N = int(input())
    dp_no_lion = [0]*(N+1)  # 사자가 없을 때
    dp_left_lion = [0]*(N+1)  # 왼쪽에 사자가 있을 떄
    dp_right_lion = [0]*(N+1)  # 오른쪽에 사자가 있을 때

    dp_no_lion[0] = 1
    for i in range(1, N+1):
        dp_no_lion[i] = (dp_no_lion[i-1] + dp_left_lion[i-1] + dp_right_lion[i-1]) % 9901
        dp_left_lion[i] = (dp_no_lion[i-1] + dp_right_lion[i-1]) % 9901
        dp_right_lion[i] = (dp_no_lion[i-1] + dp_left_lion[i-1]) % 9901

    answer = dp_no_lion[N] + dp_left_lion[N] + dp_right_lion[N]
    print(answer % 9901)
    

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE

링크: https://www.acmicpc.net/problem/1309

1309번: 동물원

---

문제

어떤 동물원에 가로로 두칸 세로로 N칸인 아래와 같은 우리가 있다.

이 동물원에는 사자들이 살고 있는데 사자들을 우리에 가둘 때, 가로로도 세로로도 붙어 있게 배치할 수는 없다. 이 동물원 조련사는 사자들의 배치 문제 때문에 골머리를 앓고 있다.

동물원 조련사의 머리가 아프지 않도록 우리가 2*N 배열에 사자를 배치하는 경우의 수가 몇 가지인지를 알아내는 프로그램을 작성해 주도록 하자. 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 친다고 가정한다.

입력

첫째 줄에 우리의 크기 N(1≤N≤100,000)이 주어진다.

출력

첫째 줄에 사자를 배치하는 경우의 수를 9901로 나눈 나머지를 출력하여라.