[JUNGOL 도형만들기1-2046]: 숫자사각형4

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

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

문제

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

< 처리조건 >
종류 2번의 경우 숫자의 진행 순서는 처음에 왼쪽에서 오른쪽으로 너비 n만큼 진행 한 후 방향을 바꾸어서 이를 반복한다.

입력 형식

정사각형 한 변의 길이 n(n의 범위는 100 이하의 정수)과 종류 m(m은 1부터 3사이의 정수)을 입력받는다.

출력 형식

위에서 언급한 3가지 종류를 입력에서 한 변의 길이 n과 종류 m에 맞춰서 출력한다. 숫자 사이는 공백으로 구분하여 출력한다.

입력 예

3 2

출력 예

1 2 3 
3 2 1 
1 2 3

code

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

if m==1:
    for i in range(1,n+1):
        for j in range(1,n+1):
            print(i,end=" ")          
        print()
elif m==2:
    for i in range(1,n+1):
        if i%2==1:
            for j in range(1,n+1):
                print(j,end=" ")          
            print()
        elif i%2==0:
            for j in range(n,0,-1):
                print(j,end=" ")
            print()
elif m==3:
    for i in range(1,n+1):

        for j in range(1,n+1):
            print(i*j,end=" ")         
        print()