[Algorithm] 합이 같은 부분 집합 (DFS) (javaScript)

swing·2023년 9월 6일
0

[Algorithm]

목록 보기
89/96

문제

N개의 원소로 구성된 자연수 집합이 주어지면, 이 집합을 두 개의 부분집합으로 나누었을 때 두 부분집합의 원소의 합이 서로 같은 경우가 존재하면 “YES"를 출력하고, 그렇지 않으면 ”NO"를 출력하는 프로그램을 작성하세요.
둘로 나뉘는 두 부분집합은 서로소 집합(Disjoint Set)이며, 두 부분집합을 합하면 입력으로 주 어진 원래의 집합이 되어야 합니다.
예를 들어 {1, 3, 5, 6, 7, 10}이 입력되면 {1, 3, 5, 7} = {6, 10} 으로 두 부분집합의 합이 16으로 같은 경우가 존재하는 것을 알 수 있다.

입력설명

첫 번째 줄에 집합의 원소 N(1<=N<=10)개가 주어진다.
각 원소는 중복되지 않으며, 그 크기는 1,000,000 이하입니다.

출력설명

첫 번째 줄에 “YES" 또는 ”NO"를 출력한다.

입출력예제

입력
1 3 5 6 7 10

출력
YES

문제 해결

const solution = (arr) => {
  let answer = "NO";
  let flag = false;
  const arrSum = arr.reduce((prev, curr) => prev + curr, 0);

  const dfs = (l, sum) => {
    if (l === arr.length) {
      if (flag) return;

      if (arrSum - sum === sum) {
        answer = "YES";
        flag = true;
      }
    } else {
      dfs(l + 1, sum + arr[l]);
      dfs(l + 1, sum);
    }
  };

  dfs(0, 0);
  return answer;
};

const tmp = solution([1, 3, 5, 6, 7, 10]);
console.log(tmp); // YES
profile
if(기록📝) 성장🌱

0개의 댓글