pythonの競技プログラムの書き方
入力の基本
整数 1 つの場合
N = int(input())
input()ではstr型なのでint型に型変換している
整数が複数個存在する場合
N, K = map(int, input().split())
split()で分割したのをint型に(map()を使って変換している) 整数 N 個 (改行区切り)の場合
L = [int(input()) for i in range(N)]
整数 N 個 (スペース区切り)の場合
A = list(map(int, input().split()))
整数 (縦 H 横 W の行列)の場合
S = [list(map(int, input().split())) for i in range(H)]
処理を途中で終了させたい時
exit()を使う
より参照