[백준 - 17392번] 우울한 방학 - Java //Wello Horld //

 

이번에는 BOJ의 17392번 문제 "우울한 방학" 을 풀어보도록 하자

 

 

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

 

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));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[] arr = new int[N];
        st = new StringTokenizer(br.readLine());
        int sum = 0;
        for (int i = 0; i < N; i++) {
            int a = Integer.parseInt(st.nextToken());
            sum += a + 1;
            arr[i] = a;
        }
        int b = M - sum;
        int ans = 0;
        if (b > 0) {
            int zari = 1;
            while (b / (N + 1) != 0) {
                ans += Math.pow(zari, 2) * (N + 1);
                zari++;
                b -= (N + 1);
            }
            ans += Math.pow(zari, 2) * (b % (N + 1));
        }
        bw.write(ans + "\n");

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

 

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

17392번: 우울한 방학

 

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