競プロをPythonでやるときのメモ

はじめに

AtcoderをPythonで戦って行く際、忘れまくるのでメモりまくります

標準入力

ただ受け取るだけ

入力：admin
s = input()
print(s)

admin 型[str]

入力：123
s = int(input())
print(s)

123 型[int]→文字列はエラーになる

リストにする

入力：admin
s = list(input())
print(s)

['a','d','m','i','n'] 型[list]

mapを使う（一度に複数の変数を受け取る）

入力：3 4 5
a, b, c = map(int, input().split())
print(a,b,c)

3 4 5 型[int]

改行されている値の取得

入力
1
2
3
4

# if 行数がわかる場合
a = [int(input()) for i in range(4)]

# if 行数が不明の場合
import sys

a = []
for i in sys.stdin:
    a.append(int(i))

[1, 2, 3, 4]

標準出力

複数の変数を出力

入力：[1, 2, 3, 4]
a = [1,2,3,4]
count = 1
print(a)
for i in a:
    print(count, i)
    print("{0}, {1}".format(count, i))
    count += 1

[1, 2, 3, 4]
(1, 1)
1, 1
(2, 2)
2, 2
(3, 3)
3, 3
(4, 4)
4, 4

文字列操作

逆順に出力

入力：123
a = input()
print(a)
print(a[::-1])

123 型[str]
321 型[str]

繰り返し処理

引数を複数に

a = [1,2,3]
b = ["a", "b", "c"]
for (i,j) in zip(a,b):
    print(i,j)

1 a 型[int str]
2 b 型[int str]
3 c 型[int str]

引数を複数に　その２

a = [11,12,13,14,15]
b = ["a", "b", "c", "d", "e"]


for (i,j) in zip(range(len(a)),b):
    print(a[i], b[i])

11 a 型[int str]
12 b 型[int str]
13 c 型[int str]
14 d 型[int str]
15 e 型[int str]