hello-algo/ja/codes/go/chapter_dynamic_programming/knapsack_test.go

// File: knapsack_test.go
// Created Time: 2023-07-23
// Author: Reanon ([email protected])

package chapter_dynamic_programming

import (
	"fmt"
	"testing"
)

func TestKnapsack(t *testing.T) {
	wgt := []int{10, 20, 30, 40, 50}
	val := []int{50, 120, 150, 210, 240}
	c := 50
	n := len(wgt)

	// 全探索
	res := knapsackDFS(wgt, val, n, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)

	// メモ化探索
	mem := make([][]int, n+1)
	for i := 0; i <= n; i++ {
		mem[i] = make([]int, c+1)
		for j := 0; j <= c; j++ {
			mem[i][j] = -1
		}
	}
	res = knapsackDFSMem(wgt, val, mem, n, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)

	// 動的計画法
	res = knapsackDP(wgt, val, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)

	// 空間を最適化した動的計画法
	res = knapsackDPComp(wgt, val, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)
}

func TestUnboundedKnapsack(t *testing.T) {
	wgt := []int{1, 2, 3}
	val := []int{5, 11, 15}
	c := 4

	// 動的計画法
	res := unboundedKnapsackDP(wgt, val, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)

	// 空間を最適化した動的計画法
	res = unboundedKnapsackDPComp(wgt, val, c)
	fmt.Printf("ナップサックの容量を超えない品物の最大価値は %d\n", res)
}