pythonで始めるatcoder [超絶初心者編]

超絶初心者がpythonでatcoderを始めた時に役立つコード集です。

スニペット

import sys
import math
import fractions
from functools import reduce

入力処理

1行 / 1列

s = input()
n = int(input())
x = float(input())

1行 / R列

l = list(map(int, input().split()))
print(sum(l))

n, m = input().split()
n, m = map(int, input().split())

C行 / 1列

n = int(input())
l = [input() for i in range(n)]
l = [int(input()) for i in range(n)]

C行 / R列

n = int(input())
l = [input().split() for i in range(n)]
l = [list(map(int, input().split())) for i in range(n)]

リストを連結して文字列として表示

l = ['python', 'list', 125]
lm = map(str, str_list)
lp = ''.join(lm)
    
print(mojiretu)

ソート

昇順

l.sort()
l.sort(key=lambda x:x[1])
l.sort(key=lambda x:(x[1],x[2]))

降順

l.sort(reverse=True)
l.sort(key=lambda x:x[1], reverse=True)
l.sort(key=lambda x:(x[1],x[2]) reverse=True)

数学系

切り上げ

math.ceil(num)

切り捨て

math.floor(num)

絶対値

abs(num)

最大値 / 最小値

max(l)
max(l, key=lambda x:x[1])

min(l)
max(l, key=lambda x:x[1])

整数の商と余りを取得

a, b = divmod(10, 3)

最大公約数

fractions.gcd(num1, num2)
reduce(fractions.gcd, numlist)

約数の個数

# @return [0]:約数の個数 [1]:約数リスト
def divisor(num):
    ret=[]
    L=[]
    for i in range(1,num+1):
        if (num%i==0):
            L.append(i)
    ret.append(len(L))
    ret.append(L)
    return ret

最小公倍数

(num1 * num2) // fractions.gcd(num1, num2)

配列

重複排除

L=list(set(numlist))

追加 / 削除

L.append(num1)

L.pop(0)

要素数カウント / 要素の存在有無

L.count()

if("aaa" in L)

if("aaa" not in L)

0パディング

a.zfill(3)