[백준 3023] 파이썬 : 마술사 이민혁

3023

https://www.acmicpc.net/problem/3023

3023번: 마술사 이민혁

[정답1]

import sys
R, C = map(int, sys.stdin.readline().split())
a = []
b = [[0 for j in range(2*C)] for i in range(2*R)]
for r in range(R):
    a.append(sys.stdin.readline().strip())
for i in range(R):
    for j in range(C):
        b[i][j] = a[i][j]
for i in range(R):
    for j in range(C, 2*C):
        b[i][j] = b[i][2*C-j-1]
for i in range(R, 2*R):
    for j in range(2*C):
        b[i][j] = b[2*R-i-1][j]
 
A, B = map(int, sys.stdin.readline().split())
if b[A-1][B-1] == '#':
    b[A-1][B-1] = '.'
else:
    b[A-1][B-1] = '#'
 
for r in range(2*R):
    for c in range(2*C):
        print(b[r][c], end = '')
    print()

.

.

.

[풀이]

for문을 통해서 기본, 오른쪽 대칭, 위아래 대칭순으로 각각 해주었다.

for문을 여러번 써서 지저분해보인다.

슬라이싱을 이용한 방법을 알아보았다.

[정답2]

import sys
R, C = map(int, sys.stdin.readline().split())
b = []
for r in range(R):
    a = sys.stdin.readline().strip()
    b += [a + a[::-1]]
b += b[::-1]
 
A, B = map(int, sys.stdin.readline().split())
if b[A-1][B-1] == '#':
    b[A-1] = b[A-1][:B-1] + '.' + b[A-1][B:]
else:
    b[A-1] = b[A-1][:B-1] + '#' + b[A-1][B:]
 
for r in range(2*R):
    print(b[r])

 

훨씬 깔끔해졌다.

 

처음 입력을 받음과 동시에 오른쪽 대칭(+a[::-1])을 추가해준다.

오른쪽 대칭을 한 상태에서 위아래 대칭이 곧 원점 대칭과 같음을 이용하여

b의 역순(+b[::-1])을 추가해준다.

에러 추가는 b[A-1]의 type이 str이기 때문에 b[A-1][B-1]을 바로 바꾸지 못하고

행 전체를 바꿔 입력해준다.