[JUNGOL 도형만들기2-1523]: 별삼각형1

JUNGOL 정보올림피아드&알고리즘에서 학습용으로 문제를 가져왔습니다. 문제가 될 시 수정, 삭제하겠습니다.

http://www.jungol.co.kr/

 

문제

삼각형의 높이 n과 종류 m을 입력 받은 후 다음과 같은 삼각형 형태로 출력하는 프로그램을 작성하시오.

입력 형식

삼각형의 크기 n(n의 범위는 100 이하의 자연수)과 종류 m(m은 1부터 3사이의 자연수)을 입력받는다.

출력 형식

위에서 언급한 3가지 종류를 입력에서 들어온 높이 n과 종류 m에 맞춰서 출력한다.
입력된 데이터가 주어진 범위를 벗어나면 "INPUT ERROR!"을 출력한다.

입력 예

3 2

출력 예

***
**
*

code

n,m=map(int,input().split())
if n<0 or n>100 or m<1 or m>3:
    print("INPUT ERROR!")
elif m==1:
    for i in range(1,n+1):
        print("*"*i)
elif m==2:
    for i in range(1,n+1):
        print("*"*(n+1-i))
elif m==3:
    for i in range(1,n+1):
        print(" "*(n-i)+"*" * (i*2-1))