日本レジストリサービス(JPRS)プログラミングコンテスト2023(AtCoder Beginner Contest 324)の解説記事です。
ABC324 A – Same
問題
問題文の要約は以下の通りです。
問題の要約
\(N\) 個の整数 \(A_1, A_2, \cdots, A_N\) がすべて等しいか判定せよ。
制約
・\(2 \le N \le 100\)
・\(1 \le A_i \le 100\) (\(1 \le i \le 100\))
・入力は全て整数
入力
\(N\)
\(A_1\) \(A_2\) \(\cdots\) \(A_N\)
出力
すべて等しいなら Yes
を、そうでなければ No
を出力せよ。
解法
set
を使用して要素の個数を数える。
解説
\(N\) は使用しないのでinput()
だけ行い、変数には格納しません。
# A_iの個数(使用しない)
input()
\(A_i\) をset
型で受け取り要素の個数が1であればYes
を、そうでなければ No
を出力します。
# 出力
print('Yes' if len(set(input().split()))==1 else 'No')
解答
# A_iの個数(使用しない)
input()
# 出力
print('Yes' if len(set(input().split()))==1 else 'No')