AtCoder市にある3つの駅は、鉄道会社A,B のいずれかが管理しており、バスが運行することになる組み合わせが存在するかどうかを判定する問題です。
文字列Sの中に 「A」「B」の両方ともがあるかどうかを確認することで求めることができます。
片方しかない場合は、「No」となります。
提出
s = input()
if "A" in s and "B" in s:
print("Yes")
else:
print("No")
if文を用いて、文字列Sの中に、A,B が含まれているかどうかを判定し、含まれている場合は「Yes」、どちらか一方でも含まれていない場合は「No」を出力しています。
s = "ABB"
print("A" in s)
# > True
このほか、count()を用いる方法でも求めることができます。
提出
s = input()
a = s.count("A")
b = s.count("B")
if a == 0 or b == 0:
print("No")
else:
print("Yes")
count()を用いて、「A」「B」があるかどうか確認し、どちらか一方でも0の場合に「No」、どちらもある場合に「Yes」を返しています。