정렬 알고리즘 문제 (Feat : 백준 브론즈)

성찬홍·2024년 9월 28일

정렬

자료구조

목록 보기

16/29

약수 구하기

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

문제

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다.

6을 예로 들면

6 ÷ 1 = 6 … 0
6 ÷ 2 = 3 … 0
6 ÷ 3 = 2 … 0
6 ÷ 4 = 1 … 2
6 ÷ 5 = 1 … 1
6 ÷ 6 = 1 … 0
그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.

두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을 작성하시오.

입력
첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

출력
첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 0을 출력하시오.

내 풀이

1부터 N까지 모든 수를 확인하기에 ,최악의 경우 O(N)의 시간이 걸린다.

function solution(N, K) {
  let count = 0;
  
  for (let i = 1; i <= N; i++) {
    if (N % i === 0) {
      count++;
      if (count === K) {
        return i;  // K번째 약수인 i를 반환
      }
    }
  }
  
  // 만약 K번째 약수가 없다면 0을 반환
  return 0;
}

GPT 풀이

약수를 효율적으로 구하려면 sqrt(N)까지의 수만 확인하고 , 그 때의 짝이 되는 약수도 동시에 확인하는 방식이 최적화된 방식이다.
Math.sqrt() ⇒ 제곱근

function solution(N, K) {
  let divisors = [];

  for (let i = 1; i <= Math.sqrt(N); i++) {
    if (N % i === 0) {
      divisors.push(i);
      if (i !== N / i) {
        divisors.push(N / i);
      }
    }
  }

  divisors.sort((a, b) => a - b);  // 약수를 정렬

  return divisors[K - 1] || 0;  // K번째 약수 반환, 없으면 0
}

이진수

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

내 풀이

function solution9(T, testCases) {
  function matchTwo(num) {
    let number = num;
    let arr = [];
    while (number !== 0) {
      const div = Math.floor(number / 2); // 몫
      const a = number % 2; // 나머지
      if (div === 0) {
        arr.push(a === 0 ? 0 : 1);
        number = 0;
      } else {
        arr.push(a === 0 ? 0 : 1);
        number = div;
      }
    }

    let ans = "";

    arr.reverse().forEach((item, index) => {
      if (item === 1) {
        ans.length === 0
          ? (ans += String(index))
          : (ans += ` ${String(index)}`);
      }
    });

    return ans;
  }

  let answer = [];
  for (let i = 0; i < T; i++) {
    const result = matchTwo(testCases[i]);
    result !== "" ? answer.push(result) : "";
  }

  return answer.join("\n");
}
console.log(solution9(1, [13]));

GPT 풀이

내 풀이에서 좀 더 불필요한 과정을 줄여준 풀이이다.

function solution9(T, testCases) {
  function matchTwo(num) {
    let result = [];
    let position = 0;

    while (num > 0) {
      if (num % 2 === 1) {
        result.push(position);
      }
      num = Math.floor(num / 2);
      position++;
    }

    return result.join(' ');  // 공백으로 구분하여 위치 출력
  }

  let answers = [];
  for (let i = 0; i < T; i++) {
    answers.push(matchTwo(testCases[i]));
  }

  return answers.join('\n');  // 각 테스트 케이스 결과를 줄바꿈으로 구분하여 반환
}

파보나치 수 5

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

내 풀이

// 피보나치 수 5
function sol8(n) {
  let arr = [0, 1, 1];
  let answer = 0;
  if (n === 0) {
    return 0;
  } else if (n === 1 || n === 2) {
    return 1;
  }

  let init = 3; // 3번쨰거부터 구하기
  while (true) {
    const addNum = arr[init - 1] + arr[init - 2];
    console.log("addNum", addNum);
    arr.push(addNum);
    if (arr.length === n + 1) {
      console.log("확인");
      console.log("addNum", addNum);
      answer = addNum;
      break;
    }
    init++;
  }
  console.log("answer", answer);

  return;
}
sol8(10);

GPT 풀이

function sol8(n) {
  if (n === 0) return 0;
  if (n === 1) return 1;

  let prev = 0, curr = 1;

  for (let i = 2; i <= n; i++) {
    let next = prev + curr;
    prev = curr;
    curr = next;
  }

  return curr;
}

N번 째 큰 수

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

내 풀이

해당 풀이는 내 풀이와 GPT와는 거의 동일했다.

function sols(T, array) {
  let arr = [];
  for (let i = 0; i < T; i++) {
    const b = array[i].sort((a, b) => {
      return b - a;
    });

    // 3번째 큰 수
    arr.push(b[2]);
  }
  console.log(arr.join(`\n`));
}
sols(4, [
  [1, 2, 3, 4, 5, 6, 7, 8, 9, 1000],
  [338, 304, 619, 95, 343, 496, 489, 116, 98, 127],
  [931, 240, 986, 894, 826, 640, 965, 833, 136, 138],
  [940, 955, 364, 188, 133, 254, 501, 122, 768, 408],
]);