JSで動的計画法を利用して部分和問題を解く

JSで動的計画法を利用して部分和問題を解く

2022-02-207 min read

目次

  1. 概要
  2. 動的計画法
  3. 部分和
  4. ソース
  5. おまけ-typescript用

概要

動的計画法を利用して部分和問題を計算した際のメモです。(AtCoder用)

動的計画法

動的計画法について

動的計画法 - Wikipedia

部分和

部分和問題について

部分和問題 - Wikipedia

ソース

動的計画法を行う関数

function solve(data, num) {
  const dp = Array.from(
    new Array(data.length + 1),
    () => new Array(num + 1).fill(0),
  );
  for (let col = 0; col < data.length; col++) {
    for (let row = 0; row < num + 1; row++) {
      if (row < data[col]) {
        dp[col + 1][row] = dp[col][row];
      } else {
        dp[col + 1][row] = Math.max(
          dp[col][row],
          dp[col][row - data[col]] + data[col],
        );
      }
    }
  }
  return dp[data.length][num] === num;
}

利用方法は以下です。

// [問題]
// 問: [3, 7, 8, 12, 13, 18] の部分和が 27 になる部分集合を求めよ。
// 答: 存在する。[7, 8, 12]

const arr = [3, 7, 8, 12, 13, 18];
console.log(solve(arr, 27) ? '存在する' : '存在しない');

おまけ: TypeScript用

TypeScript用のソースです。

const solve = (data: Array<number>, num: number) => {
  const dp = Array.from(
    new Array(data.length + 1),
    () => new Array(num + 1).fill(0),
  );
  for (let col = 0; col < data.length; col++) {
    for (let row = 0; row < num + 1; row++) {
      if (row < data[col]) {
        dp[col + 1][row] = dp[col][row];
      } else {
        dp[col + 1][row] = Math.max(
          dp[col][row],
          dp[col][row - data[col]] + data[col],
        );
      }
    }
  }
  return dp[data.length][num] === num;
};
Tags
javascript(109)
linux(54)
node.js(53)
amazon%20aws(47)
typescript(44)
%E3%82%A2%E3%83%AB%E3%82%B4%E3%83%AA%E3%82%BA%E3%83%A0(36)
%E7%94%BB%E5%83%8F%E5%87%A6%E7%90%86(30)
html5(29)
php(24)
centos(24)
python(22)
%E7%AB%B6%E6%8A%80%E3%83%97%E3%83%AD%E3%82%B0%E3%83%A9%E3%83%9F%E3%83%B3%E3%82%B0(21)
mac(21)
mysql(20)
canvas(19)
opencv(17)
%E9%9B%91%E8%AB%87(16)
docker(16)
wordpress(15)
atcoder(14)
apache(12)
%E6%A9%9F%E6%A2%B0%E5%AD%A6%E7%BF%92(12)
%E3%83%87%E3%83%BC%E3%82%BF%E3%83%99%E3%83%BC%E3%82%B9(12)
amazon%20s3(12)
red%20hat(12)
prisma(12)
ubuntu(11)
github(10)
git(10)
vue.js(10)
%E7%94%BB%E5%83%8F%E5%87%A6%E7%90%86100%E6%9C%AC%E3%83%8E%E3%83%83%E3%82%AF(10)
mariadb(10)
react(9)
aws%20cdk(9)
css3(8)
%E5%8F%AF%E8%A6%96%E5%8C%96(8)
%E5%B0%8F%E3%83%8D%E3%82%BF(8)
nestjs(8)
amazon%20lightsail(7)
next.js(7)
%E3%83%96%E3%83%AD%E3%82%B0(6)
cms(6)
oracle(6)
perl(6)
gitlab(6)
iam(5)
amazon%20ec2(5)
%E8%B3%87%E6%A0%BC%E8%A9%A6%E9%A8%93(5)
aws%20amplify(5)
curl(4)
Author
githubzennqiita
ただの備忘録です。

※外部送信に関する公表事項