본문 바로가기
알고리즘/백준 solved.ac PS

[백준][solved.ac][Bronze 3][Python] 10539번 : 수빈이와 수열

by 빛밤하늘 2021. 6. 18.
반응형

밑의 링크는 백준에서의 문제 링크입니다.

 

10539번: 수빈이와 수열

수빈이는 심심해서 수열을 가지고 놀고 있다. 먼저, 정수 수열 A를 쓴다. 그리고 그 아래에 정수 수열 A의 해당 항까지의 평균값을 그 항으로 하는 정수 수열 B를 쓴다.  예를 들어, 수열 A가 1, 3,

www.acmicpc.net

 

 

 

 

 

 

 

 

파이썬 코드와 결과입니다.

# readline을 사용하기 위해 import합니다.
from sys import stdin


# 첫째 줄에는 수열 B의 길이 N을 입력합니다.
# 1 <= N <= 100
# 정수형으로 변환합니다.
N = int(stdin.readline())
# 둘째 줄에는 수열 Bi를 이루는 N개의 정수를 공백으로 구분해 입력합니다.
# 1 <= Bi <= 10^9
# 각각 정수형으로 변환하고 리스트 변수에 넣어줍니다.
B = list(map(int, stdin.readline().split(' ')))
# 길이가 N인 수열 A를 리스트 변수로 선언합니다.
# 모든 값을 0으로 초기화합니다.
A = [0] * N

# 수열의 길이 N만큼 반복합니다.
for idx in range(N):
    # Ai의 값을 저장하는 변수를 선언합니다.
    # (수열 B에서 현재 인덱스의 값) X (현재 인덱스 + 1) - (수열 A의 첫 번째부터 현재 인덱스+1까지의 합)
    Ai = B[idx] * (idx + 1) - sum(A[:idx+1])
    # 수열 A의 현재 인덱스에 Ai의 값을 초기화합니다.
    A[idx] = Ai

# 출력 형식에 맞게 Ai들을 출력합니다.
for Ai in A:
    print(Ai, end=' ')

결과

 

 

 

※ 궁금한 부분, 이상한 점 및 오타는 댓글에 부탁드립니다.

※ 더 효율적이고 빠른 정답을 환영합니다.

반응형

댓글