集合setが最強だなと思ったのでメモ。
ARC006A
Difficulty:238
問題自体は簡単。
あとはいかに早くスマートに記述できるかが競プロで高パフォーマンスを出すための鍵！
※スマートに記述できた方が、バグもコード量も減ると思う！
別にsetなんか使わなくても解けるんだけど、今回はsetを使って解いてみる。
集合setの予備知識
高校数学のベン図みたいなことができる！
A|B:和集合
A&B:積集合
A^B:排他的論理和(どっちか一方のみtrue ビット演算のXOR)
A-B:差集合(Aのみにあるもの)
具体的には・・・
test.py
A={1,2,3,5,6}
B={2,4,5,7}
print(A|B)
print(A&B)
print(A^B)
print(A-B)
出力結果
set強い(確信)
いろいろ使いどころありそう
実際にARC006Aをsetを使ってとくとこんな感じ
test.py
def I(): return int(input())
def LI(): return list(map(int,input().split()))
E = set(LI()) #集合として受け取る！
B = I()
L = set(LI()) #集合として受け取る！
ans = 0
count = len(E&L) #積集合！！！
if count==5 and B in L:
ans = 2
else:
ans = {6:1,5:3,4:4,3:5}[count] if count>=3 else 0
print(ans)
この集合の知識が頭にはいっていればこの問題は3分以内でコード描けるはず！
こういうの知らなくても解けるけど、知ってると早く解ける=強い人になれるとなんとなく思ってる。
おわり!