set（）で重複をなくす(AtCoder(競プロ)1日1問【26日目】【Python】)

AtCoder Beginner Contest 089 B問題

競技プログラミングで使える関数を実際の問題を通して紹介いたします。

今回は「set（）」です。

 set（）で重複をなくす

>|python|

n = int(input())
type = len(set([ i for i in input().split()]))
print(["Three","Four"][type>=4])

||<

n=input()を使ってないので無駄な作業なのですが、入力されるのだから仕方ない。

コードの改善 

>|python|

input()
print('Four' if 'Y' in input() else 'Three')

||<

>|python|

print('Four' if 'Y' in input() else 'Three')

||<

問題文によると

• S_i=P、S_j=W、S_k=G を満たす i,j,k が必ず存在する

制約条件までしっかり目を通すことで解くことが楽になる問題も多く存在しますので見逃すことがないようにしましょう。

独学プログラマー Python言語の基本から仕事のやり方まで