[ 자바스크립트(JavaScript) ] section04 - 3 - 멘토링

📍 section04 - 3 - 멘토링(bruteForce)

조금 어려운 문제였다. bruteForce로 풀어야 하는 것은 알고 있었지만, 어떤 흐름으로 문제를 풀어야 할지 고민이 많았다. 강의를 봤는데도 이해가 잘 안 돼서 복습을 여러 번 했다.

 

결과적으로 이 문제의 핵심은 n명의 학생이 각각의 경우에 m번의 수학 성적 모두 mento와 mentee가 될 수 있는 조건이 맞는지 찾아야 하고, 그 안에서 수학 등수를 나타내는 idx를 고려해서 mentoIdx < menteeIdx인 조건을 찾을 수 있는지 물어보는 문제인 것 같다. 또, mento와 mentee는 같은 학생일 때는 성립하지 않음을 알아야 한다.

 

강의에서 4 중반 복문으로 풀었는데, 반복문의 개수가 많다 보니까 헷갈려서 수학 성적이 m개가 아닌 딱 1개만 주어졌을 때를 가정하고 풀어봤다. 다음의 사진처럼 경우의 수를 모두 구해보고 코드를 작성하니까 조금 이해가 됐다.

 

 

728x90

let arr = [3, 4, 1, 2];
let n = 4;
let m = 1;
let cnt = 0;

for (let x = 1; x <= n; x++) {
  for (let y = 1; y <= n; y++) {
    let mentoIdx = menteeIdx = 0;
    for (let i = 0; i < n; i++) {
      if (x === y) continue;
      if (arr[i] === x) mentoIdx = i;
      if (arr[i] === y) menteeIdx = i;
    }
    if (mentoIdx < menteeIdx) {
        cnt++; 
        console.log(x, y);
    }
  }
}
console.log(cnt);
👉🏽 
1 2
3 1
3 2
3 4
4 1
4 2

6

 

수학 성적에 2차원 배열이 들어갔을 때도 위에서 작성한 코드에서 크게 벗어나지 않는다. 다만, mento, mentee의 관계가 성립하려면 모든 수학 성적에서 mento가 더 앞서야 하므로 cnt가 m과 동일하다는 조건을 걸어줘야 한다. 전체 코드는 다음과 같다.

 

let arr = [[3, 4, 1, 2], [4, 3, 2, 1], [3, 1, 4, 2]];
let n = 4;
let m = 3;

console.log(solution(n, m, arr));

function solution(n, m, arr){
    let ans = 0;
    for (let x=1; x<=n; x++){
        for (let y=1; y<=n; y++){
            let cnt = 0;
            for (let i=0; i<m; i++){
                let mento = mentee = 0;
                for (let j=0; j<n; j++){
                    if (x===y) continue;
                    if (arr[i][j] === x) mento = j;
                    if (arr[i][j] === y) mentee = j;  
                }
                if (mento < mentee) cnt++;
            }
            if (cnt === m) ans++;
        }
    }
    return ans;
}