abcコンテスト59回のC問題が分かりません(https://atcoder.jp/contests/abc059/tasks/arc072_a)。
長さNの数列があり、i番目の数はa_iです。あなたは1回の操作でどれか1つの項の値を1つだけ増やすか減らすことができます。
以下の条件を満たすために必要な操作回数の最小値を求めてください。
・すべてのi(1=<i=<n)に対し、第1項から第i項までの和は0ではない
・すべてのi(1=<i=<n-1)に対し、i項までの和とi+1項までの和の符号が異なる
入力
n
a_1 a_2 ・・・ a_n
出力
必要な操作回数の最小値を出力せよ
(例)
入力:
4
1 -3 1 0
出力:
4
以下に自分が考えたコードを示しましたが、間違いです。
どのように考えたらいいか教えていただけませんか。
初学者ゆえ、非常に見にくいコードとなっていますがご了承ください。
python
1n = input() 2a = [int(i) for i in input().split()] 3cou = len(a) ### cou:aの要素数の取得 4counter = 0 5X = a[0] ### X:i番目までの和 6ans = 0 ### ans:操作回数 7for i in a: 8 if X > 0: ### Xが正のとき、Xが負(-1)になるための操作回数を得る 9 b = -1 - X 10 ans += abs(b - a[i]) 11 X = -1 12 counter += 1 13 if counter == cou - 1: 14 break 15 else: ### Xが負のとき、Xが正(1)になるための操作回数を得る 16 b = 1 - X 17 ans += abs(b - a[i]) 18 X = 1 19 counter += 1 20 if counter == cou - 1: ### aの要素数-1番目になったらbreakしてansを出力 21 break 22print (ans)
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/03/05 13:19