AtCoder-ABC220 C - Long Sequence【Python解答例】 - ebisukeプログラミング初心者脱出黙示録

AtCoder Beginner Contest220のC問題についてPythonの解答例を記事にしていきます。
AtCoder Beginner Contest 220 - AtCoder

AtCoder Beginner Contest220 C - Long Sequence

C - Long Sequence

問題文

長さ N の正整数のみからなる数列 A=(A 1 ,…,A N ) があります。
A を 10 ^ 100 回連結した数列を数列 B とします。

B の項を前から順に足したとき、和が初めて X を超えるのは何項目まで足したときですか？
すなわち、以下の式を満たす最小の整数 k を求めてください。

i=1
∑ B i >X
k

制約

・1≤N≤10 ^5
・1≤A i ≤10 ^9
・1≤X≤10 ^18
・入力は全て整数

解答例

n = int(input())

A = list(map(int, input().split()))

x = int(input())

sum_a = 0
for i in range(n):
    sum_a += A[i]

ans = 0
ans += n * (x // sum_a)
SUM = sum_a * (x // sum_a)
i = 0
while x >= SUM:
    SUM += A[i]
    ans += 1
    i += 1

print(ans)