자료구조 및 실습 - [3] 배열 실습 문제 (달팽이 문제, 대각선 방향 숫자출력) (2)

N x N (1≤N≤100) 크기의 행렬에 1 ~ N² 의 수를 아래 그림과 같이 차례로 위에서부터 → 방향과 ← 방향을 번갈아 가면서 채운 결과를 출력하시오.

아이디어

가장 먼저 1열에서 규칙을 찾으려고 노력했다. (1,1)에는 1 , (2,1)에는 8 , (3,1)에는 9, (4,1)에는 16 ...

N의 관점에서 보면 1, 2*N, 2*N+1, 4*N... 짝수 행과 홀수 행을 보면

1, 2N+1,4N+1,...

2N,4N,... 

으로 진행되므로 case를 나누었다.

 

코드

N = int(input())
k=1
for i in range(N):
  if i%2==0:
      print(f" {k} {k+1} {k+2} {k+3}")
      k= k+N+N-1
  
  if i%2==1:
    print(f" {k} {k-1} {k-2} {k-3}")
    k+=1

배열을 구성해서 풀면 k를 인덱스로 넣으면 된다.

 

---

N x M (1≤N, M≤100) 크기의 행렬에 1 ~ MN 의 수를 아래 그림과 같이 나선형으로 채운 결과를 출력하시오.

아이디어

과정을 하나하나 적어보았다. 몇가지 패턴이 보이는데

1. 방향이 우, 하, 좌, 상 을 반복하며 숫자가 입력된다.

2. 좌우, 상하가 짝을 이루며 한번 입력될수록 다음번에 입력되는 갯수가 1개씩 감소한다.

에 집중하여 코드를 구성했다.

 

코드

dir= [ 0,1,2,3] # 0은 우측, 1은 아래, 2는 좌측, 3은 위쪽
N,M = input().split()
N = int(N); M = int(M)
tot = N*M
a = []
b=[]
for a1 in range(N):
  a.append(list(range(0,M)))
k=1;i=0;j=-1
tmp = dir[0] #가장먼저 우측으로 돈다.
N=N-1
while k<=tot: # k가 N*M까지 돈다
  if tmp==0: #우측으로 가야할 순서면
    for _ in range(M): # 숫자 반복입력
      j=j+1
      #print(i,j)
      a[i][j]=k
      #print(f"a[{i}][{j}]: {k}")
      #print(a)
      k=k+1
    tmp = dir[1] #방향 전환
    M=M-1
    #print('------------')

  elif tmp ==1: #아래로 갈 순서
    for _ in range(N):
      i=i+1
      #print(i,j)
      a[i][j] = k
      #print(f"a[{i}][{j}]: {k}")
      #print(a)
      k=k+1
    N=N-1
    tmp = dir[2] #방향 변경
    #print('------------')
  elif tmp ==2: #좌로 갈 순서
    for _ in range(M):
      j=j-1
      #print(i,j)
      a[i][j] = k
      #print(f"a[{i}][{j}]: {k}")
      #print(a)
      k=k+1
    M=M-1
    tmp = dir[3] #방향 변경
    #print('------------')
  elif tmp == 3: #위로 갈 순서
    for _ in range(N):
      i=i-1
      #print(i,j)
      a[i][j] = k
      #print(f"a[{i}][{j}]: {k}")
      #print(a)
      k=k+1
    N=N-1
    tmp = dir[0] #다시 우측으로 방향전환
    #print('------------')

print(a[0])
print(a[1])
print(a[2])
print(a[3])

중간중간 출력을 확인하느라 print문을 많이 넣어놨다... 다행히 정상적으로 출력되었다.

---

N x M (1≤N, M≤100) 크기의 행렬에 1 ~ MN 의 수를 아래 그림과 같이 ↙ 대각선
방향으로 채운 결과를 출력하시오.

아이디어

 

 

코드

####배열 초기화####
N,M = input().split()
N = int(N); M = int(M)
tot = N*M
a = []
for a1 in range(N):
  b=[]
  for _ in range(M):
    b.append(0)
  a.append(b)
#----------------------------------
num=0
cnt=0
for i in range(N*M): #입력될 숫자 갯수
  if cnt>=N*M: #숫자 넘어가면 브레이크
    break
  else: # 숫자 안넘었으면
    for j in range(N): #최대 대각선 갯수가 N개이므로 N번 반복 / j는 0,1,2,3반복
      k=i-j #k는 0 / 1 0 / 2 1 0 / 3 2 1 0 / 4 3 2 1 ...식으로 증가함
      if k>=0 and k<M: # 경계조건 정의
        num+=1
        a[j][k] = num # 조건에 해당되면 배열에 입력
        cnt+=1 # 입력된 갯수 카운트

a

예를 들어 N=4, M=5라고 하자.

N x M에 숫자를 채울 때 대각선 방향으로 숫자가 1개,2개,...입력되는데 이때 대각선 방향으로 연속되게 적힐 수 있는 숫자의 갯수는 N개이다.

먼저 N*M번의 반복문을 돈다. 이후 최대 대각선 숫자 갯수는 N개 이므로 N번 반복문을 돌린다.

i가 포함된 반복문을 돌수록 i는  0,1,2,3... 증가하고 j는 0,1,2,3만 반복한다. k = i-j로 정의하면서 k는 대각선 갯수도 증가하면서 0 / 1 0 / 2 1 0 / 3 2 1 0 / 4 3 2 1 ...식으로 맨처음에 오는 값이 1씩 커진다. 경계조건을 설정해주면 최대 N개만 입력되면서 순차적으로 잘 입력되는 것을 확인할 수 있다.