[코드업_기초100제] #1098 문제

#1098 문제

P1098 : [기초-종합+배열] 설탕과자 뽑기

mat_n = list(map(int, input().split()))
print(mat_n)

5 5
[5, 5]

zero_mat = []
for i in range(mat_n[0]) :
    zero_mat.append([0]*mat_n[1])
zero_mat

[[0, 0, 0, 0, 0],
 [0, 0, 0, 0, 0],
 [0, 0, 0, 0, 0],
 [0, 0, 0, 0, 0],
 [0, 0, 0, 0, 0]]

n = int(input())

mat = []
for i in range(n) :
    mat.append(list(map(int, input().split())))
print(mat)

0 1 1
1 2 3
1 2 5
[[2, 0, 1, 1], [3, 1, 2, 3], [4, 1, 2, 5]]

length = []
for i in range(n) :
    length.append(mat[i][0]) 
print(length)

[2, 3, 4]

for i in range(n) :
    if mat[i][1] == 0 :
        for a in range(length[i]) :
            zero_mat[mat[i][2]-1][mat[i][3]-1+a] = 1
    elif mat[i][1] == 1 :
        for a in range(length[i]) :
            zero_mat[mat[i][2]-1+a][mat[i][3]-1] = 1
        

zero_mat

[[1, 1, 0, 0, 0],
 [0, 0, 1, 0, 1],
 [0, 0, 1, 0, 1],
 [0, 0, 1, 0, 1],
 [0, 0, 0, 0, 1]]

for i in range(mat_n[0]) :
    for j in range(mat_n[1]) :
        print(zero_mat[i][j], end = ' ')
    print('')

# 내 답안 정리
## 세로 가로 matrix 만들기
mat_n = list(map(int, input().split()))

## 0 matrix만들기
zero_mat = []
for i in range(mat_n[0]) :
    zero_mat.append([0]*mat_n[1])

## n개의 막대기 갯수
n = int(input())

## 막대기 길이, 방향, 좌표(x,y) matrix 만들기
mat = []
for i in range(n) :
    mat.append(list(map(int, input().split())))

## 길이만 따로 matrix 만들기
length = []
for i in range(n) :
    length.append(mat[i][0]) 

## if문으로 방향(0 or 1)을 for문으로 길이를 정함
for i in range(n) :
    if mat[i][1] == 0 :
        for a in range(length[i]) :
            zero_mat[mat[i][2]-1][mat[i][3]-1+a] = 1
    elif mat[i][1] == 1 :
        for a in range(length[i]) :
            zero_mat[mat[i][2]-1+a][mat[i][3]-1] = 1

## list를 숫자로만 입력해주기
for i in range(mat_n[0]) :
    for j in range(mat_n[1]) :
        print(zero_mat[i][j], end = ' ')
    print('')

# 코드업 답안
a,b=input().split()
h=int(a)
w=int(b)

m=[]
for i in range(h+1) :
    m.append([])
    for j in range(w+1) :
        m[i].append(0)

n=int(input())

for i in range(n) :
    l,d,x,y=input().split()
    for j in range(int(l)) :
        if int(d)==0 :
            m[int(x)][int(y)+j]=1
        else :
            m[int(x)+j][int(y)]=1


for i in range(1, h+1) :
    for j in range(1, w+1) :
        print(m[i][j], end=' ')
    print()

Share this post