【Python】AtCoderで使える小ネタ・テクニック集

A = int(input())

N,M = map(int,input().split())
S = list(map(int,input().split()))
for _ in range(N):
  L.appnend(input())

from sys import stdin
N,M = map(int,stdin.readline().split())
S,T = [],[]
for n in range(N):
  S.append(stdin.readline())
for m in range(M):
  T.append(stdin.readline())
print(N,M)
print(S)
'''
4 4
['ab\n', 'cd\n', 'ef\n', 'gh\n']
改行が残る
'''

T.append(stdin.readline()[:-1])
['ab', 'cd', 'ef', 'gh']

l = [1,2,3,4,5]
print(*l)

'''出力
1 2 3 4 5
'''

C = [0,1,1,1,0,0]
column = "".join(map(str,C))

'''出力
011100
'''

A = list(range(100000))
c = ""
for a in A:
  c += str(a)
print(c)

A = list(range(100000))
c = "".join(map(str,A))
print(c)

a = set([1,3,53,2,3,2])
print(a)
#{1, 2, 3, 53}

#空のset型、list型の生成
list_example = list() #またはlist_example = []
set_example = set() #set_example = {}とすると辞書型で定義されてしまう。

#データの追加
list_example.append(5)
set_example.add(5)

#list型の要素削除
list_example.remove(5) #指定した要素に一致するものを削除。複数ある場合は、インデックスが小さい方が削除される。
list_example.remove(6) #指定した要素がリスト内に存在しない場合はエラー
a = list_example.pop() #末尾の要素を削除し、returnする。インデックスをしていすることも可能。削除する要素がないとエラー


#set型の要素の削除
set_example.remove(5) #指定した要素を削除する。指定する要素が集合にないとエラー
set_example.discard(5) #指定した要素を削除する。指定する要素が集合になくてもエラーは吐かない
a = set_exmaple.pop() #集合の中からランダムな要素を削除する。削除したものをreturnします。

if S[0] == 0:
  for sl in split_list:
    if sl in column:
      print("Yes")
      exit()
print("No")

D = {"Monday":5,
     "Tuesday":4,
     "Wednesday":3,
     "Thursday":2,
     "Friday":1}

import math
math.ceil(5/3)
'''出力
2
'''

math.ceil(-5/3)
'''出力
-1
'''

(x + y - 1) / y

import sys
sys.setrecursionlimit(2000000)

import sys
sys.setrecursionlimit(2000000)
N,M = map(int,sys.stdin.readline().split())
graph = {i+1 : [] for i in range(N)}
for _ in range(M):
    u,v = map(int,sys.stdin.readline().split())
    graph[u].append(v)
    graph[v].append(u)
 
result = 0
def dfs(v,visited):
  global result
  if result == 1000000:
    return
  result += 1
  visited[v-1] = 1
  for neibor in graph[v]:
    if not visited[neibor-1]:
      dfs(neibor,visited)
  visited[v-1] = 0
 
visited = [0] * N
dfs(1,visited)
print(result)

N,M = map(int,input().split())
V = {i+1:set() for i in range(N)}
for _ in range(M):
  u,v = map(int,input().split())
  V[u].add(v)
  V[v].add(u)

＃再帰で実装した例  
def dfs(v,visited,comp):
  visited.add(v)
  comp.add(v)
  for neibor in V[v]:
    if neibor not in visited:
      dfs(neibor,visited,comp)

#スタックを用いて実装した例
def dfs_stack(v,visited,comp):
  stack = [v]
  while stack:
    #一番最後に追加されたものを引っ張り出す。
    vertex = stack.pop() 
    comp.add(vertex)
    #訪れたことのない頂点ならば処理を実行
    if vertex not in visited: 
      visited.add(vertex)
      #接続する頂点が訪れたことのないものならスタックに追加
      for neibor in V[vertex]: 
        if neibor not in visited:
          stack.append(neibor)
      
visited = set() #set型で訪れたことのあるノード一覧を定義。
comps = [] #グラフの連結成分を格納するリスト
for v in V:
  comp = set()
  if v not in visited:
    dfs_stack(v,visited,comp)
    comps.append(comp)
print(len(comps))

from collections import deque
def bfs(v,visited,comp):
  queue = deque([v])
  comp.add(v)
  visited.add(v)
  while queue:
    vertex = queue.popleft()
    for neibor in V[vertex]:
      if neibor not in visited:
        visited.add(neibor)
        queue.append(neibor)
        comp.add(neibor)

def b_s(data,value,right): #binary_search
  left = 0
  right = right
  while left <= right:
    mid = (left + right) // 2
    if data[mid] == value:return True
    elif data[mid] < value:left = mid + 1
    else:right = mid - 1
  return False

import itertools
for i in itertools.product(range(1,6),repeat=3):
   print(i)

''' 出力
(1, 1, 1)
(1, 1, 2)
(1, 1, 3)
(1, 1, 4)
(1, 1, 5)
(1, 2, 1)
(1, 2, 2)
(1, 2, 3)
.
.
.
(5, 5, 3)
(5, 5, 4)
(5, 5, 5)
'''

import itertools
for i in itertools.permutations(range(1,6),3):
   print(i)

'''
(1, 2, 3)
(1, 2, 4)
(1, 2, 5)
(1, 3, 2)
(1, 3, 4)
(1, 3, 5)
(1, 4, 2)
.
.
.
(5, 4, 1)
(5, 4, 2)
(5, 4, 3)
'''

for i in itertools.combinations(range(1,6),3):
   print(i)
'''出力
(1, 2, 3)
(1, 2, 4)
(1, 2, 5)
(1, 3, 4)
(1, 3, 5)
(1, 4, 5)
(2, 3, 4)
(2, 3, 5)
(2, 4, 5)
(3, 4, 5)
'''

for i in itertools.combinations_with_replacement(range(1,6),3):
   print(i)

''' 出力
(1, 1, 1)
(1, 1, 2)
(1, 1, 3)
(1, 1, 4)
(1, 1, 5)
(1, 2, 2)
(1, 2, 3)
(1, 2, 4)
(1, 2, 5)
(1, 3, 3)
(1, 3, 4)
(1, 3, 5)
(1, 4, 4)
(1, 4, 5)
(1, 5, 5)
(2, 2, 2)
(2, 2, 3)
(2, 2, 4)
(2, 2, 5)
(2, 3, 3)
(2, 3, 4)
(2, 3, 5)
(2, 4, 4)
(2, 4, 5)
(2, 5, 5)
(3, 3, 3)
(3, 3, 4)
(3, 3, 5)
(3, 4, 4)
(3, 4, 5)
(3, 5, 5)
(4, 4, 4)
(4, 4, 5)
(4, 5, 5)
(5, 5, 5)
'''