Python 백준 문제풀이(2차원 배열, 일반수학1)

2차원배열

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

2738번: 행렬 덧셈

n,m = map(int,input().split())
arr = [[0 for i in range(m)] for j in range(n)]
arr2 = [[0 for i in range(m)] for j in range(n)]
arr3 = [[0 for i in range(m)] for j in range(n)]
for i in range(n):
    l = list(map(int,input().split()))
    for j in range(m):
        arr[i][j]=l[j]
for i in range(n):
    k = list(map(int,input().split()))
    for j in range(m):
        arr2[i][j]=k[j]
for i in range(n):
    for j in range(m):
        arr3[i][j]=arr[i][j]+arr2[i][j]
        print(arr3[i][j],end=' ')
    print()

arr = [[0 for i in range(열)] for j in range(행)] 을 헷갈리지 말 것

다른이의 풀이:

A, B = [], []

N, M = map(int, input().split())

for row in range(N):
    row = list(map(int, input().split()))
    A.append(row)

for row in range(N):
    row = list(map(int, input().split()))
    B.append(row)
    
for row in range(N):
    for col in range(M):
        print(A[row][col] + B[row][col], end=' ')
    print()

아예 list를 통째로 넣어서 2차원 리스트를 만들어주고 있다.

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

2566번: 최댓값

table = [list(map(int, input().split())) for _ in range(9)]
max_value=0
max_col=0
max_row=0
for i in range(9):
    for j in range(9):
        if max_value<=table[i][j]:
            max_value=table[i][j]
            max_col=i+1
            max_row=j+1
print(max_value)
print(max_col,max_row)

table = [list(map(int, input().split())) for _ in range(9)] 이런 방식으로 2차원 테이블을 생성할 수도 있다.

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

arr = []
for i in range(5):
    l=list(input())
    arr.append(l)
    arr[i] += [''] * (15 - len(l))
for i in range(15):
    for j in range(5):
        print(arr[j][i],end='')

처음에는 arr[i] += [''] * (15 - len(l)) 이 부분을 arr[i].append( [''] * (15-len(l))) 이런식으로 작성했으나 후자의 방식은 원하던 대로 동작하지 않았다. arr[i]의 리스트에 ''를 추가하는 것이 아니라, ['']를 추가했다. ['c', 'd', 'e', 'f', ['', '', '', '', '', '', '', '', '', '', '']] 이런식으로. 또한 arr[i].append( '' * (15-len(l))) 이런식으로 작성하니 ['c', 'd', 'e', 'f', ''] 이런식으로 되어 원하던 결과를 도출하지 못했다. 이미 먼저 arr.append(l)로 리스트를 추가했으니, 해당 리스트에 나머지 ''를 추가하려면 arr[i] += [''] * (15 - len(l)) 이런식으로 작성해야했다. 표현법에 주의 요망

다른이의 풀이:

words = [input() for i in range(5)]

for j in range(15):
    for i in range(5):
        if j < len(words[i]):
            print(words[i][j], end='')

j가 len보다 짧을때만 출력하도록 작성했다.

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

arr = [[0 for  i in range(100)] for j in range(100)]
n = int(input())
sum=0
for _ in range(n):
    x, y = map(int,input().split())
    for i in range(10):
        for j in range(10):
            arr[x+i][y+j] = 1
for j in range(100):
    sum+=arr[j].count(1)
print(sum)

처음에

for i in range(10):
        for j in range(10):
            arr[x+i][y+j] = 1

이 자리에

for i in range(10):
        arr[x+i][y:y+10]=1

이런식의 코드나

arr[x:x+10][y:y+10]=1

이런식의 코드가 가능할까싶어 해봤으나 되지않았다. 슬라이싱 구문에 할당을 하려고 하면 안됨.

---

일반수학1

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

a, b= input().split()
b= int(b)
sum=0
k=0
for x in a:
    try:
        sum+=int(x)*(b**(len(a)-1-k))
        k+=1
    except:
        sum+=(ord(x)-55)*(b**(len(a)-1-k))
        k+=1
print(sum)

아스키코드를 이용해 풀었다. 

다른 풀이:

num_list = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"

n, b = input().split()
answer = 0
for i, num in enumerate(n[::-1]):
    answer += int(b) ** i * num_list.index(num)
print(answer)

enumerate를 사용해 풀었다. 인덱스와 값을 동시에 사용하고 싶을때 사용

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

n, m = map(int,input().split())
num_list = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
l = []
while n>0:
    l.append(n%m)
    n//=m
for i in l[::-1]:
    print(num_list[i],end='')

m진법을 십진법으로 만들기위해서 나눈값을 리스트에 넣고 몫을 계속 나눠주었다.

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

n = int(input())
l = [25,10,5,1]
for _ in range(n):
    m = int(input())
    for j in range(len(l)):
        print(m//l[j],end=' ')
        m%=l[j]
    print()

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

n = int(input())
m=1
while True:
    if n==1:
        break
    elif (3*(m-1)*(m-2)+1)<n<(3*(m)*(m-1)+2):
        break
    else:
        m+=1
print(m)

1, 2(2)~7(6개), 8(2+6)~19(12개), 20(2+6+6+6=2+6*3)~37(18개), 38(2+6+6+6+6+6+6=2+6*6)~61(24개)개),62(38+6*4=2+6*10)~91(30개)
→0/6/18(6*3)/36(6*6)/60(6*10)/90(6*15)
→ 6*(n*(n-1)/2)=3*n*(n-1)

이러한 식을 이용해 풀었으나 더 간단한 풀이가 있었다

num = int(input())
numbox = 1
cnt = 1

while num > numbox:
    numbox += 6 * cnt
    cnt += 1
print(cnt)

이런식으로 cnt를 하나씩 더해서 풀수도 있었다.

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

처음에는

a=int(input())
n=1
m=1
for _ in range(a-1):
    if (n+m)%2==0:
        if n==1:
            m+=1
        else:
            n-=1
            m+=1
    else:
        if m==1:
            n+=1
        else:
            n+=1
            m-=1
print(f"{n}/{m}")

이런식으로 작성했으나 시간초과가 떴다.

num = int(input())
line = 1

while num > line:
    num -= line
    line += 1
    
# 짝수일경우
if line % 2 == 0:
    a = num
    b = line - num + 1
# 홀수일경우
elif line % 2 == 1:
    a = line - num + 1
    b = num

print(f'{a}/{b}')

이런식으로 고쳤다. 문자열 포맷팅을 사용하였다. print(f'{변수}') 이런식으로 사용할 수 있다.

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

import math
up, dwn, total = map(int, input().split())
day = math.ceil((total - up) / (up - dwn)) + 1
print(day)

시간제한이 있으므로 for문을 쓸수없었다. math를 임포트하여 ceil을 사용해 올림을 해주었다.