Chapter 01

ABC201

Meg
Meg
2021.10.25に更新

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

A - Tiny Arithmetic Sequence

問題はこちら。

https://atcoder.jp/contests/abc201/tasks/abc201_a

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

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

参考

https://docs.python.org/ja/3/library/itertools.html

https://docs.python.org/ja/3/library/sys.html

B問題以降

誠意記事作成中です。