23か月前公開・23か月前更新・2 min read

Pythonで競技プログラミングをやるときに使う記法まとめ

はじめまして。
最近、AtCorderをやり始めた万年灰色のアラフォーエンジニアです。言語はPythonを使っています。

競プロって、特殊なコードが多いので慣れるまで大変です。そこで個人的に競プロ時によく使うPythonの記法をまとめていきます。(随時更新予定です。競プロを続けられている限り・・・💧)

Magicodeはコード実行できるので、短いコードを実行することが多い競プロと相性がいいのではと思っています。

入出力

a = int(input())
print(a)

b, c = map(int, input().split())
print(b)
print(c)

スペース区切り入力を配列にする場合 (listをsetにすれば、set型にもできる)

a_list:list = list((map(int, input().split())))
print(a_list)

[1, 2, 3]

別法 (上記の法がわかりやすいし応用もしやすいので、使うケースはないと思う)

a_list: list = [int(x) for x in input().split()]
print(a_list)

[1, 2, 3]

# 入力 ex.
# 1
# 3
# 5
a = [int(input()) for i in range(3)]
print(a)  # [1,3,5]

[1, 3, 5]

a = [1,3,5]
print(*a)

1 3 5

Mex(Minimum EXcluded Value)などの問題に使える。 (例題 AtCorder 250 B)

has_nums = [True, True, False, True, False]
print(has_nums.index(False))

Pythonの三項演算子は見づらいので、あまり使いたくないが、これぐらいはパッと使えるようにしたほうが良さそう。

is_ok = True
print("Yes" if is_ok else "No")
is_ok = False
print("Yes" if is_ok else "No")

Yes No

a = list(range(5))
print(a)

[0, 1, 2, 3, 4]

# 配列の最大値の値とindexを取得
# 最小値はmaxをminにする
l = [4, 9, 8, 19, 12]
m = max(l)
print(m)
print(l.index(m))

19 3

max, minは複数のintから最大値/最小値を得るのにも使用できる。

a, b = 2, 5

# いまいちな書き方
tmp = 0
if a < b:
  tmp = b
else:
  tmp = a
print(tmp)

# シンプルな書き方
print(max(a, b))

# index指定の削除
l = [4, 9, 8, 19, 12]
l.pop(1)
print(l)

# rangeは配列インデックスと同じ (start:0, end:引数-1)
for i in range(5):
  print(i)

print("------")

# start指定パターン
for i in range(2,5):
  print(i)

0 1 2 3 4 ------ 2 3 4

l = ["a","b","c"]
for i, v in enumerate(l):
  print(i, v)

0 a 1 b 2 c

Ads