[백준] 22971 증가하는 부분 수열의 개수

[백준] 22971 증가하는 부분 수열의 개수 - Dynamic Programming / Java

graycode 2023. 9. 26. 13:57

• 문제 링크

22971번: 증가하는 부분 수열의 개수

길이가 $N$인 수열 $A$가 주어진다. 수열의 $i$번째 원소($A_i$)로 끝나는 증가하는 부분 수열의 개수를 출력하는 프로그램을 작성하자. 단, 수가 너무 커질 수 있으니 $998\,244\,353$으로 나눈 나머지를

www.acmicpc.net

• 풀이 코드

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        int n = read();
        int[] arr = new int[n], dp = new int[n];

        for (int i = 0; i < n; i++) {
            arr[i] = read();
            dp[i] = 1;
            for (int j = 0; j < i; j++) if (arr[i] > arr[j]) dp[i] = (dp[i] + dp[j]) % 998244353;
        }

        for (int i : dp) sb.append(i).append(" ");

        bw.write(sb.toString());
        bw.flush();
    }

    private static int read() throws IOException {
        int c, n = System.in.read() & 15;
        while ((c = System.in.read()) > 32) n = (n << 3) + (n << 1) + (c & 15);

        return n;
    }

}