atcoder ABC122 B "ATcoder" 自分のコードの誤りがわからない(python)

Question

前提・実現したいこと

atcoder beginner contest122のB問題"ATcoder"を解いています。
ABC122 B ATcoder

他の人のコード等を見て解き方は理解できたのですが、自分のソースコードのどこが誤っていて動いていないのかが理解できません。
具体的には、li.append(count)をしているのに、リストにcountが追加されていないようです。

間違いを教えてくださるとありがたいです。

追記：
皆様のアドバイスを受けて、コードを変更しました。
具体的には、入力が1文字であった時にエラーが起こらないように修正いたしました。

修正後のコードを追記します

ソースコード（修正前）

S = list(input())
n = len(S)
count = 0
li = []
for i in range(n):
  if S[i] == "A" or "C" or "G" or "T":
    count += 1
  else:
    li.append(count)
    count = 0
print(max(li))

###ソースコード（修正後）

S = list(input())
n = len(S)
count = 0
li = []
for i in range(n):
  if i != n-1:
    if S[i] == "A" or S[i] == "C" or S[i] == "G" or S[i] == "T":
      count += 1
    else:
      li.append(count)
      count = 0
  if i == n-1:
      count += 1
      li.append(count)
if len(S) == 1:
  if S[0] == "A" or S[0] == "C" or S[0] == "G" or S[0] == "T":
    print(1)
  else:
    print(0)
else:
  print(max(li))

Answer 1

テストケースがatcoder_testcasesで公開されています。
SHINJUKUの正答は0ですが1となります。より単純な例ではZZも。
難しく処理しすぎていると思います。もう少し単純に判定できます。

Answer 2

他の人が書いてる条件のところは修正したとして、Sの末尾が　"A" or "C" or "G" or "T"　の文字列の場合どうなりますか？

Comments

[takuma1229]

回答ありがとうございます！
ご指摘の通り、Sの末尾がacgtだとlistにcountが追加されておらずエラーが出ていました！
修正後のコードを追記します。

しかし、まだWAが3つ出てしまっています。。。
何が問題か教えていただけると大変ありがたいです。

[satoshih]

if i == n-1:
count += 1
li.append(count)

の部分、末尾がなんの文字でも＋１してるので、Sが "AZ" とかだと２になりませんか？

Answer 3

S[i] == ("A" or "C" or "G" or "T")
では間違いでしょうね。

>>> print(("A" or "C" or "G" or "T"))
A
>>> print('T' == ("A" or "C" or "G" or "T"))
False

これを修正してもしなくても、入力が1文字の場合は実行時エラーになります。
自分の書いたコードにいろいろな入力をして結果を確認しましょう。

Comments

[takuma1229]

ありがとうございます！その通りだったようです。
ご指摘のケースに対応できるよう修正しました。

Answer 4

if S[i] == "A" or "C" or "G" or "T":

って、
if （S[i] == "A"） or "C" or "G" or "T":
って動作になるけど、そんでいいんでしょうか

Comments

[takuma1229]

なるほど！！！
いえ、求めている動作はif S[i] == ("A" or "C" or "G" or "T")です！
ちょっと変更してみます！

[takuma1229]

変更したらリストにcount自体は追加されたのですが、
サンプルケースの"ATCODER"を標準出力に設定すると、3になるはずの出力が1と返ってきてしまいます。。。

[y_waiwai]

求めてる動作ってのは
S[i] == "A" or S[i]=="C" or Ｓ［i］＝＝なんたら...
ってのなんでは。

[takuma1229]

そのように実装したのですが、テストケースでWAが2つ、REが3つ出てしまいました。。。どうやらどこか間違いがあるようです

[y_waiwai]

まあ、別のとこにもなんかあるんでしょうね

＃問題読んでない