[백준 - 5557번] 1학년 - Java //Wello Horld //

 

 

이번에는 BOJ의 5557번 문제 "1학년" 을 풀어보도록 하자

 

 

 

 

성공한 코드는 다음과 같다.

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        int[] arr = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for(int i = 0; i < N; i++){
            arr[i] = Integer.parseInt(st.nextToken());
        }
        long[][] dp = new long[N - 1][21];
        dp[0][arr[0]] = 1;
        for(int i = 1; i < N - 1; i++){
            for(int j = 0; j < 21; j++){
                if(dp[i - 1][j] != 0) {
                    if(j + arr[i] <= 20) dp[i][j + arr[i]] += dp[i - 1][j]; 
                    if(j - arr[i] >= 0) dp[i][j - arr[i]] += dp[i - 1][j];
                }
            }
        }
        bw.write(dp[N - 2][arr[N - 1]] + "\n");

        bw.flush();
        br.close();
        bw.close();
    }
}

 

 

 

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

5557번: 1학년

 

 

혹시 코드에 이상한 부분이나 틀린 부분이 있던지, 이해가 안가는 부분이 있다면 댓글로 알려주세요