どうもこんにちは、Megです。
このチャプターでは、PythonでAtCoder Beginner Contest 201 の解説を行っていきます。

A - Tiny Arithmetic Sequence

問題はこちら。

コードは以下です！自分の提出はこちら。

import itertools
import sys

As = list(map(int, input().split()))

for a in itertools.permutations(As, 3):
    if a[2] - a[1] == a[1] - a[0]:
        print('Yes')
        sys.exit()

print('No')

コード解説

A_1, A_2, A_3をAsという一つのリストで受け取ります。
itertools.permutationsでA_1, A_2, A_3を並べ替えた組を作ります。
そして、for~if文で等差数列の条件A_3 - A_2 = A_2 - A_1を満たしているか調べます。
条件を満たしていたらYesと出力し、sys.exit()で実行を止めます。
条件が満たされなかったら、Noを出力します。

参考

B問題以降

誠意記事作成中です。