AtCoder Beginner Contest 072 C - Together 【Python】

https://atcoder.jp/contests/abc072/tasks/arc082_a

AtCoder ProblemsのRecommendationで Difficulty: 407、Solve Probability: 52%でした。

3つの操作後の数列をループさせて整数を辞書にカウントしていき、そのカウント数の最大となる整数が答えとなりました。

from collections import defaultdict

N = int(input())
A = list(map(int, input().split()))
Ap1 = [i+1 for i in A]
Am1 = [i-1 for i in A]

d = defaultdict(int)

for i in A:
    d[i] += 1
for i in Ap1:
    d[i] += 1
for i in Am1:
    d[i] += 1

print(d[max(d, key=d.get)])