[Gold III] Title: 훈련소로 가는 날, Time: 348 ms, Memory: 86512 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/18858. 훈련소로 가는 날/README.md

@@ -0,0 +1,46 @@
+# [Gold III] 훈련소로 가는 날 - 18858 
+
+[문제 링크](https://www.acmicpc.net/problem/18858) 
+
+### 성능 요약
+
+메모리: 86512 KB, 시간: 348 ms
+
+### 분류
+
+다이나믹 프로그래밍
+
+### 제출 일자
+
+2026년 2월 22일 16:38:37
+
+### 문제 설명
+
+<p style="text-align: justify;">훈련소로 가는 날 욱제는 문득 떠올렸다. 훈련소가 논산에 있는 이유는 무엇일까? 왜 why?</p>
+
+<p>그것은 바로…</p>
+
+<p> </p>
+
+<p> </p>
+
+<p> </p>
+
+<p> </p>
+
+<p> </p>
+
+<p> </p>
+
+<p>논산(non-산)은 산이 아니기 때문이다. 길이가 3인 수열 a<sub>1</sub>, a<sub>2</sub>, a<sub>3</sub>가 산임은 a<sub>1</sub> < a<sub>2</sub> > a<sub>3</sub>임을 의미한다. 어떤 수열이 논산임은 수열의 인접한 세 항이 산인 경우가 없음을 의미한다. 다시 말해, 길이 N의 수열 a에 대해 2 ≤ i < N 이고 a<sub>i-1</sub> < a<sub>i</sub> > a<sub>i+1</sub>인 경우가 없다.</p>
+
+<p>논산인 수열이 몇 개가 있는지 알아보자.</p>
+
+### 입력 
+
+ <p>첫째 줄에 N과 M이 주어진다.</p>
+
+### 출력 
+
+ <p>1 이상 M 이하의 정수로 이루어진 길이 N의 수열 중 논산인 것의 개수를 998,244,353으로 나눈 나머지를 출력한다.</p>
+

백준/Gold/18858. 훈련소로 가는 날/훈련소로 가는 날.swift

@@ -0,0 +1,54 @@
+import Foundation
+
+let MOD = 998244353
+
+let NM = readLine()!.split(separator: " ").map { Int($0)! },
+    N = NM[0],
+    M = NM[1]
+
+if N == 1 {
+    print(M)
+    exit(0)
+}
+
+var dp = Array(
+    repeating: Array(
+        repeating: Array(repeating: 0, count: 2),
+        count: M + 1
+    ),
+    count: N + 1
+)
+
+for v in 1...M {
+    dp[1][v][0] = 1
+}
+
+for i in 2...N {
+    for u in 1...M {
+        for s in 0...1 {
+            let current = dp[i-1][u][s]
+            if current == 0 { continue }
+            
+            for v in 1...M {
+                if v > u {
+                    dp[i][v][1] = (dp[i][v][1] + current) % MOD
+                } else if v == u {
+                    dp[i][v][0] = (dp[i][v][0] + current) % MOD
+                } else {
+                    if s == 0 {
+                        dp[i][v][0] = (dp[i][v][0] + current) % MOD
+                    }
+                }
+            }
+        }
+    }
+}
+
+var answer = 0
+for v in 1...M {
+    for s in 0...1 {
+        answer = (answer + dp[N][v][s]) % MOD
+    }
+}
+
+print(answer)