10-oesnuj

[oesnuj]

🔗 문제 링크

백준 | 상어초등학교

✔️ 소요된 시간

2시간

✨ 수도 코드

어렵다기 보다는 귀찮고 생각할게 많은 구현문제였습니다.
이번에는 c++말고 javascript로 풀어봤습니다.
자바스크립트 문제 풀이는 전체 입력을 txt파일로 받아온 후, 원하는대로 짤라 사용하는 방식을 주로 씁니다.

- 설계

1. 입력된 순서대로 앉힌다.

2. 학생을 앉힐 때 현재 교실(board)의 모든 빈자리에 대한 인접 정보를 받아서 저장

3. 인접 정보는 해당 빈자리의 좌표, 인접 빈자리 개수, 인접 좋아하는 학생의 수를 가진다.

4. 3으로 수집한 인접 정보들을 조건에 맞게 정렬하여 특정 학생의 최적의 자리 선정

5. 해당 자리에 학생을 앉힌다.

6. 모두 앉힌 이후 만족도를 조사한다.

- 변수

N : 학생수
students : 학생, 좋아하는 학생번호로 이루어진 배열
board : 자리배치를 표시하는 2차원 배열

- 구현 함수 설명

getSeatInfo(r, c, studentNum) : 특정 자리 좌표와 학생 번호를 매개변수로 받음

특정 자리 좌표를 기준으로 4방향 탐색 -> 빈자리 수, studentNum에 해당하는 좋아하는 학생 수 계산


choiceSeat( studentNum)
studentNum이 앉을 최적의 자리를 선정하여 앉힌다.
모든 자리 탐색 -> 빈자리에 대해서 getSeatInfo 수행 -> 인접 정보 배열(neighborInfos)에 저장

neighborInfos을 위 조건에 맞게 정렬시킨다. 정렬 후 가장 앞의 객체가 최적의 자리


calcSatisfy()
모든 자리에 대해서 getSeatInfo로 그 학생과 인접한 칸에 앉은 좋아하는 학생의 수를 받아오고 이에 따라 만족도 처리

❗주의할 점

result += Math.floor(10 ** (getSeatInfo(i, j ,board[i][j]).match-1));

만족도 처리 시에 match가 0인 경우 10의 -1승은 0.1 이므로 0으로 만들기 위해 Math.floor를 사용해야 함
math.floor(x) : x 이하의 가장 큰 정수 반환

📃 전체 코드

const fs = require('fs');
const filepath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const input = fs.readFileSync(filepath).toString().trim().split('\n');
//입력 데이터 정리
const N = +input.shift();
const students = input.map(x => {
    const [num, ...likes] = x.trim().split(' ').map(Number);
    return { num, likes };
});

let board = Array.from({ length: N }, () => Array(N).fill(0)); 

const dr = [-1, 1, 0, 0];
const dc = [0, 0, -1, 1];

main();

function main(){
    for(let i=0; i< N**2; i++){
        if(i == 0){
            board[1][1] = students[i].num; //첫학생은 1,1에 무조건 앉는다.
            continue;
        }
        choiceSeat(students[i].num); //학생을 조건에 맞게 앉히기
    }
    console.log(calcSatisfy()); //모두 앉은 후 만족도 계산하기
}

// 최적의 자리 선택 및 학생 배치 함수
function choiceSeat(studentNum){
    const neighborInfos = []; //인접 자리 정보를 모으는 배열
    for(let i = 0; i<N; i++){ 
        for(let j =0; j<N; j++){
            if(board[i][j] !== 0) continue; //이미 차있는 자리 패스
            neighborInfos.push(getSeatInfo(i, j, studentNum));
            }    
        }
        neighborInfos.sort((a, b) => {
            if (a.match !== b.match) {
                return b.match - a.match; // match 기준 내림차순 정렬
            } else if (a.empty !== b.empty) {
                return b.empty - a.empty; // empty 기준 내림차순 정렬
            } else if (a.r !== b.r) {
                return a.r - b.r; // r 기준 오름차순 정렬
            } else {
                return a.c - b.c; // c 기준 오름차순 정렬
            }
        });
        board[neighborInfos[0].r][neighborInfos[0].c] = studentNum; //최적의 위치에 앉히기
}

// 특정 자리의 인접 정보 계산 함수
function getSeatInfo(r, c, studentNum){
    let empty = 0;
    let match = 0;
    // 학생 번호에 맞는 좋아하는 학생들 찾기
    let studentLikes = students.find(student => student.num === studentNum)?.likes || [];
    for(let i = 0; i< 4; i++){
        nr = r + dr[i];
        nc= c + dc[i];
        if(nr < 0 || nc < 0 || nr >= N || nc >= N) continue;
        if (board[nr][nc] == 0) empty++;
        else if(studentLikes.includes(board[nr][nc])) match++
    }
    return { r: r, c: c, empty: empty, match: match };
}

//만족도 처리
function calcSatisfy(){
    let result = 0;
    for(let i = 0; i<N; i++){ //남아있는 모든 자리의 정보를 수집
        for(let j =0; j<N; j++){
            result += Math.floor(10 ** (getSeatInfo(i, j ,board[i][j]).match-1));
            }    
        }
    return result;
}

📚 새롭게 알게된 내용

1. 자바스크립트 임의 정렬 하는법 - 배열 메서드 사용
   자바 스크립트는 비교 조건에 따른 반환 값에 따라 a, b를 정렬한다.
   반환 값이 음수라면 a를 앞에 정렬
   반환 값이 양수라면 b를 앞에 정렬

배열.sort(비교함수);

배열.sort(function(a, b) {비교조건}); //함수사용
배열.sort((a, b) => {비교 조건}); 	//화살표 함수 사용

ex) a == 1, b == 3 a<b인 상황이라면
오름차순 정렬 
return a - b; (음수 반환)

내림차순 정렬
return b - a; (양수 반환)

2. 2차원 배열 초기화 시키기

let board = Array.from({ length: N }, () => Array(N).fill(0));

[pu2rile]

오 자바스크립트...!! 각 변수, 함수마다 따로 설명을 적어 주셔서 좀 더 이해하기가 수월했어요!!
저는 파이썬으로 한번 풀어봤는데, 딕셔너리 like 를 사용해서 각 학생의 좋아하는 학생 목록을 저장하니 간편하더라고요!

• 키: 학생 번호
• 값: 해당 학생이 좋아하는 4명의 학생 번호 목록

또, 학생 배치 로직에서는 리스트 seats_candidates 를 사용해서 모든 좌석을 순회하고, 빈 좌석일 경우

1. 인접한 칸에서 좋아하는 학생 수 like_seat 와 빈칸 수 empty_seat 계산
2. 후보 리스트에 (like_seat, empty_seat, y, x) 추가
3. 후보 리스트를 정렬하여 최적의 자리 선택
4. 학생을 해당 자리(seats_candidates[0])에 배치

이렇게 구현해 봤습니다! 준서님 말대로 정말... 고민할 게 많은 문제였던 것 같아요 헉헉 pr 수고하셨습니다!

코드

N = int(input())
seats = [[0] * N for _ in range(N)]
like = {}  # 각 학생의 좋아하는 학생 목록 딕셔너리
students = []  # 학생 배치 순서 리스트

dy = [-1, 1, 0, 0] # 방향 벡터
dx = [0, 0, -1, 1]

for _ in range(N * N):
    data = list(map(int, input().split()))  # 학생 번호, 좋아하는 학생 4명 입력
    student = data[0]  # 학생 번호
    students.append(student)  # 학생 배치 순서 리스트에 추가
    like[student] = data[1:]  # 좋아하는 학생 목록 딕셔너리에 저장

# 배치
for student in students:
    seats_candidates = []  # 가능한 좌석 후보 목록 초기화
    for y in range(N):
        for x in range(N):
            if seats[y][x] != 0:  # 이미 학생이 앉아 있다면 패스
                continue
            like_seat = 0  # 좋아하는 학생 수
            empty_seat = 0  # 빈 자리 수
            for dir in range(4):  # 인접한 칸 확인
                ny, nx = y + dy[dir], x + dx[dir]  # 새로운 좌표 계산
                if 0 <= ny < N and 0 <= nx < N:  # 좌표가 유효한지 확인
                    if seats[ny][nx] in like[student]:  # 인접 칸에 좋아하는 학생이 있는지 확인
                        like_seat += 1
                    if seats[ny][nx] == 0:  # 인접 칸이 비어 있는지 확인
                        empty_seat += 1
            # 가능한 좌석 후보에 추가
            seats_candidates.append((like_seat, empty_seat, y, x))
    # 조건에 맞는 자리 정렬
    seats_candidates.sort(key=lambda x: (-x[0], -x[1], x[2], x[3]))
    y, x = seats_candidates[0][2], seats_candidates[0][3]  # 최적의 자리 선택
    seats[y][x] = student  # 학생 배치

result = 0  # 최종 만족도 점수 초기화
# 만족도 계산
for y in range(N):
    for x in range(N):
        student = seats[y][x]
        like_seat = 0  # 인접한 좋아하는 학생 수
        for dir in range(4):  # 인접한 칸 확인
            ny, nx = y + dy[dir], x + dx[dir]  # 새로운 좌표 계산
            if 0 <= ny < N and 0 <= nx < N and seats[ny][nx] in like[student]:  # 유효한 좌표인지 확인, 좋아하는 학생인지 확인
                like_seat += 1
        if like_seat > 0:  # 좋아하는 학생이 하나 이상 있을 경우 만족도 계산
            result += 10 ** (like_seat - 1)  # 만족도 계산 공식

print(result)  # 최종 만족도 출력

[suhyun113]

이번에도 구현 문제군요!! 문제 이름이 어떻게 상어초등학교?!🦈 이름이 귀엽군요,,
늘 하던 c++이 아닌 javascript로 하셔셔 신기했습니당. 자스..개인적으로 어려워하는 언어 중 하나ㅜㅜ입니다.

준서님이 엄청 꼼꼼하게 설명해주셔서 문제와 코드를 이해하는데 도움이 많이 되었어요!
자바스크립트는 입력 데이터를 처리하는 부분이 엄청 길던데,, 다른 언어와 비슷한 것 같으면서 조금 다른 ..신기했습니다ㅏ
학생들 자리를 배치하는 알고리즘도 따로 함수를 이용해 구현하여 보기 좋았어요! choiceSeat함수에서는 getSeatInfo함수를 호출하여 현재 자리의 인접 자리 정보를 계싼해 빈 자리 수와 좋아하는 학생 수를 반환한 결과를 바탕으로 정렬해 가장 좋은 자리에 학생을 배치하는 함수를 구현한 것이 너무 깔끔했습니다. 주석도 많이 달아주셔서 이해도 잘 되었습니당

마지막으로 calcSatisfy함수를 통해 만족도를 계산하는 함수까지 코드가 잘 짜여있는 것 같아요! 현재 코드도 좋지만, 학생들을 자리에 배치할 때, 모든 빈 자리를 탐색하고 있는 현재 알고리즘을 우선순위 큐를 사용한다면 최적의 자리를 더 빨리 찾을 수 있어 시간 복잡도를 줄일 수 있을 것 같아요! 이해도 잘 되구 좋은 PR 작성해주셔서 감사합니다🤓