백준 10539 수빈이와 수열 (자바 Java)

백준 10539 수빈이와 수열

문제 정보

• 문제 출처 : 백준 Baekjoon
• 문제 번호 : 10539
• 문제 제목 : 수빈이와 수열
• 문제 난이도 : 브론즈2
• 풀이 언어 : Java

문제

수빈이는 심심해서 수열을 가지고 놀고 있다. 먼저, 정수 수열 A를 쓴다. 그리고 그 아래에 정수 수열 A의 해당 항까지의 평균값을 그 항으로 하는 정수 수열 B를 쓴다. 예를 들어, 수열 A가 1, 3, 2, 6, 8이라면, 수열 B는 1/1, (1+3)/2, (1+3+2)/3, (1+3+2+6)/4, (1+3+2+6+8)/5, 즉, 1, 2, 2, 3, 4가 된다.
수열 B가 주어질 때, 수빈이의 규칙에 따른 수열 A는 뭘까?

접근

반복문을 사용하여 문제를 풀었다.

풀이

• 수열의 길이 N을 입력 받는다.
• 길이 N인 배열 B와 A를 초기화한다.
• for 루프를 사용하여 B 배열에 N개의 값을 입력 받는다.
• A 배열의 첫 번째 요소(A[0])를 B 배열의 첫 번째 요소(B[0])와 같게 설정한다.
• 이후의 A 배열 요소들을 계산하기 위해 for 루프를 사용한다. A[i] = B[i] * (i + 1) – B[i – 1] * i 공식을 적용하여 계산한다.
• for 루프를 사용하여 계산된 A 배열의 각 요소를 공백으로 구분하여 출력한다.
• 입력이 모두 끝난 후 Scanner 객체를 닫는다.

후기

쉬운 문제였지만, 비슷한 유형의 문제들을 더 풀어볼 계획이다.

코드

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        int N = scanner.nextInt();
        int[] B = new int[N];
        int[] A = new int[N];

        for (int i = 0; i < N; i++) {
            B[i] = scanner.nextInt();
        }

        A[0] = B[0];
        for (int i = 1; i < N; i++) {
            A[i] = B[i] * (i + 1) - B[i - 1] * i;
        }

        for (int i = 0; i < N; i++) {
            System.out.print(A[i] + " ");
        }
        scanner.close();
	}
}