CS/알고리즘_문제풀이(파이썬)

offset

Jedy_Kim 2021. 5. 10. 15:13
728x90

문제

5x5 2차원 배열이 주어질 때 어떤 원소가 상하좌우에 있는 원소보다 작을 때 해당 위치에 * 을 표시하는 프로그램을 작성하시오. 경계선에 있는 수는 상하좌우 중 존재하는 원소만을 비교한다.

 

입력

5x5 행렬의 정보가 25 개의 수로 주어진다. 각 수는 0 에서 9 사이 수이다.

 

출력

*를 포함한 행렬을 출력예의 형식으로 출력한다.

 

예제 입력

 

3 4 1 4 9

2 9 4 5 8

9 0 8 2 1

7 0 2 8 4

2 7 2 1 4

예제 출력 

3 4 * 4 9

* 9 4 5 8

9 0 8 2 *

7 0 2 8 4

* 7 2 * 4

 

# 공부하기 전(후후...부끄러운 코드)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
 
import random
 
frame = [[random.randint(09for j in range(5)] for i in range(5)]
 
 
for i in range(len(frame)):
  for j in range(len(frame[i])):
    if i == 0# 좌, 우, 하
      if j == 0#우, 하
        if frame[i][j] < frame[i][j+1and frame[i][j] < frame[i+1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      elif j == (len(frame)-1): # 좌, 하 => 좌 가 별이 아닌경우
        if frame[i][j-1!= '*' and frame[i][j] < frame[i][j-1and frame[i][j] < frame[i+1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      else# 좌, 우, 하
        if frame[i][j-1!= '*' and frame[i][j] < frame[i][j+1and frame[i][j] < frame[i][j-1and frame[i][j] < frame[i+1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
    elif i == (len(frame)-1): # 상, 좌, 우
      if j == 0#우, 상
        if frame[i][j] < frame[i][j+1and frame[i][j] < frame[i-1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      elif j == (len(frame)-1): # 좌, 하 => 좌 가 별이 아닌경우
        if frame[i][j-1!= '*' and frame[i][j] < frame[i][j-1and frame[i][j] < frame[i-1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      else# 좌, 우, 상
        if frame[i][j-1!= '*' and frame[i][j] < frame[i][j+1and frame[i][j] < frame[i][j-1and frame[i][j] < frame[i-1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
    else#상, 하, 좌, 우
      if j == 0#상, 우, 하 => 상이 별이 아닌경우
        if frame[i-1][j] != '*' and frame[i][j] < frame[i-1][j] and frame[i][j] < frame[i][j+1and frame[i+1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      elif j == (len(frame)-1): # 상, 좌, 하 => 상, 좌 가 별이 아닌경우
        if frame[i-1][j] != '*' and frame[i][j-1!= '*' and frame[i][j] < frame[i-1][j] and frame[i][j] < frame[i][j-1and frame[i+1][j]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
      else# 상, 하, 좌, 우 => 상, 좌 별이 아닌경우
        if frame[i-1][j] != '*' and frame[i][j-1!= '*' and frame[i][j] < frame[i-1][j] and frame[i][j] < frame[i][j-1and frame[i+1][j] and frame[i][j] < frame[i][j+1]:
          frame[i][j] = '*'
        print(frame[i][j], end=' ')
  print('')  
  
cs

# 공부한 후

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
outline = [[10 for _ in range(10)] for _ in range(10)]
 
frame = list()
frame=[list(map(int, input().split())) for _ in range(5)]
 
for i in range(len(frame)):
  for j in range(len(frame[i])):
    outline[i+1][j+1= frame[i][j]
    
for i in range(len(frame)):
  for j in range(len(frame[i])):
    row = i+1
    col = j+1
    # 상, 하, 좌, 우
    if outline[row][col]<outline[row-1][col] and outline[row][col]<outline[row+1][col] and outline[row][col]<outline[row][col-1and outline[row][col]<outline[row][col+1]:
      frame[i][j] = -1
 
  
  
for i in frame:
  for j in i:
    if j == -1:
      print('*', end=' ')
    else:
      print(j, end=' ')
  print()
cs

# 복습

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
frame = [[10 for _ in range(7)] for _ in range(7)]
arr = []
for i in range(5):
  row = list(map(int, sys.stdin.readline().split()))
  arr.append(row)
  
for i in range(5):
  for j in range(5):
    frame[i+1][j+1= arr[i][j]
 
for i in range(16): # 행
  for j in range(16): # 열
    # 상, 하, 좌, 우 비교
    if frame[i][j] < frame[i-1][j] and frame[i][j] < frame[i+1][j] and frame[i][j] < frame[i][j-1and frame[i][j] < frame[i][j+1]:
      arr[i-1][j-1= '*' 
 
for i in arr:
  for j in i:
    print(j, end=' ')
  print()
cs

 

반응형

'CS > 알고리즘_문제풀이(파이썬)' 카테고리의 다른 글

GCD LCM  (0) 2021.05.10
상자 꾸미기  (0) 2021.05.10
백준 - 스택 수열 #1874  (0) 2021.03.30
백준 - 단어 뒤집기 #9093  (0) 2021.03.29
백준 - 스택 #10828  (0) 2021.03.29