[Baekjoon] 17550_Inquiry I

문제(출처: https://www.acmicpc.net/problem/17550)

---

< Inquiry I >

 

문제 풀이 

 

정수의 합과 정수 제곱의 합을 미리 구해 최댓값을 구한다. 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class _17550_ { // Inquiry I

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));

		int n = Integer.parseInt(bf.readLine());

		int arr[] = new int[n + 1];
		for (int i = 1; i < n + 1; i++) {
			arr[i] = Integer.parseInt(bf.readLine());
		}

		long square[] = new long[n + 1];
		long sum[] = new long[n + 1];

		for (int i = 1; i < n + 1; i++) {
			square[i] = (arr[i] * arr[i]) + square[i - 1];
			sum[i] = arr[i] + sum[i - 1];
		}

		long result = 0;
		for (int i = 1; i < n; i++) {
			result = Math.max(result, square[i] * (sum[n] - sum[i]));
		}

		System.out.println(result);

	}
}

변수)
n : 정수 개수
arr : 정수
square : 제곱의 합
sum : 합
result : 최댓값 

 

정수 개수를 입력받는다. 정수 개수만큼 정수를 입력받아 arr에 저장한다. 미리 제곱의 합과 합을 구해 각각 square와 sum 배열에 저장한다. 앞에서부터 k개의 제곱의 합과 k+1부터 n까지의 합을 곱한 값의 최댓값을 구한다. 

---

---