방 배정 백준 13300번 배열

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

문제 정보 초등학교에서는 단체로 2박 3일 수학여행을 가기로 했다. 여러 학년이 같은 장소로 수학여행을 가려고 하는데 1학년부터 6학년까지 학생들이 묵을 방을 배정해야 한다. 남학생은 남학생끼리, 여학생은 여학생끼리 방을 배정해야 한다. 또한 한 방에는 같은 학년의 학생들을 배정해야 한다. 물론 한 방에 한 명만 배정하는 것도 가능하다.

한 방에 배정할 수 있는 최대 인원 수 K가 주어졌을 때, 조건에 맞게 모든 학생을 배정하기 위해 필요한 방의 최소 개수를 구하는 프로그램을 작성하시오.

예를 들어, 수학여행을 가는 학생이 다음과 같고 K = 2일 때 12개의 방이 필요하다. 왜냐하면 3학년 남학생을 배정하기 위해 방 두 개가 필요하고 4학년 여학생에는 방을 배정하지 않아도 되기 때문이다.

1학년	영희	동호, 동진
2학년	혜진, 상희	경수
3학년	경희	동수, 상철, 칠복
4학년		달호
5학년	정숙	호동, 건우
6학년	수지	동건

2차원 배열로 풀면 됩니다. 어차피 초등학교(6학년)과 성별(남,여)로 구분한다고 되어 있으니 6x2 2차원 행렬이 고정되어 있습니다. 그래서 해당 하는 배열에다가 값을 추가해주고 각 배열을 순회하면서 필요한 방의 개수를 구해주면 됩니다.

std_n,max_s = map(int,input().split())
room_n = [[0]*2 for x in range(6)]

for i in range(std_n):
    x,y = map(int,input().split())
    room_n[y-1][x] += 1
    
count = 0
for a in range(6):
    for b in range(2):
        roomd,rooms = divmod(room_n[a][b],max_s)
        if rooms != 0:
            roomd+=1
        count += roomd

print(count)

앞에서 보통 1차원 배열을 사용할때 [0]*길이 만큼 곱했는데 2차원 배열을 만들때 이렇게하면 얕은 복사 문제가 발생하니 for이나 append로 배열을 생성하셔야 합니다.

파이썬 리스트에서 2x2배열에서 얕은 복사가 일어나는 경우