解説つき!競ってわかる はじめてpythonプログラミング

競プロの情報発信と備忘録を兼ねている

ABC236C

ABC236C の解説

atcoder.jp


解答コード

https://atcoder.jp/contests/abc236/submissions/28779555

 

問題

急行列車が止まるかどうか判定する問題です.

 

解説

Sの全ての要素について集合Tに含まれるかどうかを判別すれば良いです.

標準入力を読み込みます.

N, M = map(int, input().split())
S = list(input().split())
T = set(input().split())

 

for文でSのすべての要素について判定します.
ifで含まれるかどうか条件分岐します.

for s in S:
    if s in T:
        print('Yes')
    else:
        print('No')

三項演算子(条件演算子)で1行で記述することもできます,

for s in S:
    print('Yes' if s in T else 'No')

 

以上,ABC236Cの解法でした.