AtCoder AGC029_A pythonについて

Question

概要

Atcoder AGC029_A(Irreversible operation)を提出したところ、TLE(TimeOut)になってしまうので解決方法について教えていただきたいです。

以下、コードの説明と実行したコードです。

説明

• Wより左にあるBの数を足していく操作をWの数だけ行う。
• その際に、一度足したBは一番左のWを含めてリストから削除する。削除したリストのBの数はtmp_1に追加していく。
• Wがなくなるまでwhileで回す。

変数の説明

• ans→最終的に出力するBの数。
• t→Wのインデックス番号(一番左のもの)
• tmp_1→削除したリストに含まれているBの数の和
• tmp_2→tより左にあるBの数

実行したコード

s=list(input())
ans=0
tmp_1=0

while "W" in s:
  t = s.index("W")
  tmp_2=s[:t].count("B")
  ans += tmp_1 + tmp_2
  tmp_1 += tmp_2
  s = s[t+1:]
print(ans)

Comments

[modieu]

print(t)は書かなければならないんですか？
printはだいぶ時間をとられると思います。

[kitsune_42]

print(t)はミスです。申し訳ないです。
実際に提出したコードには入っていないので修正いたします。

Answer 1

一行目のs=list(input())をs=input()にしてみたら?

Comments

[kitsune_42]

s=input()
にしたら上手くいきました！
listで処理すると重くなるんですね。。

Answer 2

while "W" in s:
  t = s.index("W")
  tmp_1 += s[:t].count("B")
  ans += tmp_1
  s = s[t+1:]