CS/Algorithm

[Python] 함수 & 입출력

yujindonut 2022. 1. 31. 16:52
728x90

일반적으로 메소드 작성할때!

def add(a, b):
	return a + b
    
print(add(3,7))

코드를 줄이고자 할 때는 람다표현식을 이용할 수 있다.

print(lamda a,b : a + b)(3, 7))

일반적 입력예시 : 

더보기

5

65 90 75 34 99

각 데이터를 공백으로 구분하여 입력할 때는

더보기

# 데이터의 개수 입력

n = int(input())

 

# 각 데이터를 공백으로 구분하여 입력한다 - 데이터의 입력개수가 많을 경우

data = list(map(int, input().split()))

-input()으로 입력받은 문자열을 split()을 이용해 공백으로 나눈 리스트로 바꿈

-map을 이용하여 해당 리스트의 모든 원소에 int()함수를 적용한다

-최종적으로 그 결과를 list()로 다시 바꿔서 집어넣음

 

# 공백을 기준으로 구분하여 적은 수의 데이터 입력

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

하지만 파이썬의 기본 input() 함수는 동작 속도가 느려서 시간 초과로 오답 판정을 받을 확률이 높다!!

import sys

# 문자열 입력받기
data = sys.stdin.readline().rstrip()
print(data)

-readline()으로 입력하면 입력 후 엔터가 줄 바꿈 기호로 입력되는데, 이 공백문자를 제거 하기 위해서는 rstrip()함수를 사용해야한다. 

 

728x90