[파이썬 익숙해지기] no5341:피라미드

문제

주어진 숫자 N부터 시작하여, 순차적으로 1씩 적게 쌓아 최종적으로 1까지 쌓는 피라미드의 벽돌 수를 구하시오.

즉, 1~N까지의 합 구하기 

 

문제 링크 : https://www.acmicpc.net/problem/5341

---

원리

1. 단순 반복문

for문을 이용해 1부터 N까지 순차적으로 합산하기

 

2. 합 공식

주어진 숫자 N에 대해서,

∑N = N * (N+1) // 2

---

학습 내용 

1. Python방식 while 반복문 (무한루프)

while 1 :

#또는

while true :

 

2. Python방식 for 문

# 시작 지정 방식
for i in range (시작숫자, 마지막 다음숫자) :

# 0부터 시작 방식
for i in range (마지막 다음숫자) :

 

3. Python방식 메소드 선언

# 메소드 선언
def MethodName(parameter):
    return parameter
    
# if문을 이용한 클래스 이름 매칭 선언
if __name__ == "__main__" :

 

4. Python방식 타입변환(예시 : int)

N = input()

int intN = int(N)

 

5. Python방식 입/출력

# 입력 
A = input()

# 출력
print(A + "입니다.")

---

나의 코드

1. 반복문 

# 반복문을 사용한 합산
while 1 :
    N = int(input())
    if N == 0 : 
        break
    Result = 0
    for i in range(1, N+1):
        Result += i
    print(Result)

 

2. 합 공식

#  N * (N+1) // 2 공식을 이용한 합산 (N과 N+1의 곱을 2로 나눈 몫)
def Pyramid(num):
    return num * (num+1) // 2

while 1:
    N = int(input())
    if N == 0 : 
        break
    print(Pyramid(num = N))

---

레퍼런스

반복문 방식 참고글 : https://codeomni.tistory.com/823

합공식 방식 참고글 : https://somjang.tistory.com/entry/BaekJoon-5341%EB%B2%88-Pyramids-Python