結果

問題 No.10 ＋か×か
コンテスト
ユーザー warashi
提出日時 2015-10-25 19:18:13
言語 Go
(1.23.4)
結果
AC  
実行時間 6 ms / 5,000 ms
コード長 1,293 bytes
コンパイル時間 15,004 ms
コンパイル使用メモリ 224,420 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-14 15:32:39
合計ジャッジ時間 11,648 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import (
	"fmt"
	"strings"
)

func main() {
	var N, TOTAL int
	var A []int
	fmt.Scan(&N, &TOTAL)
	A = make([]int, N)
	for i := 0; i < N; i++ {
		fmt.Scan(&A[i])
	}
	fmt.Println(solve(N, TOTAL, A))
}

func solve(N, total int, A []int) string {
	dp := make([]map[int]bool, N)
	for i := 0; i < N; i++ {
		dp[i] = make(map[int]bool)
	}
	dp[0][A[0]] = true
	dp[len(dp)-1][total] = true
	backcast(dp, A)
	return forcast(dp, A)
}
func backcast(dp []map[int]bool, A []int) {
	for i := len(dp) - 1; i > 0; i-- {
		dpn := dp[i-1]
		a := A[i]
		for b := range dp[i] {
			if b > a {
				dpn[b-a] = true
			}
			if b%a == 0 {
				dpn[b/a] = true
			}
		}
	}
}

func forcast(dp []map[int]bool, A []int) string {
	pos := 1
	val := A[0]
	path := make([]string, 0, len(dp)-1)
	dfs(val, pos, &path, dp, A)
	return strings.Join(path, "")
}

func dfs(val int, pos int, path *[]string, dp []map[int]bool, A []int) bool {
	if pos == len(dp) {
		return true
	}
	a := A[pos]
	if _, ok := dp[pos][val+a]; ok {
		*path = append(*path, "+")
		if dfs(val+a, pos+1, path, dp, A) {
			return true
		}
		*path = (*path)[:len(*path)-1]
	}
	if _, ok := dp[pos][val*a]; ok {
		*path = append(*path, "*")
		if dfs(val*a, pos+1, path, dp, A) {
			return true
		}
		*path = (*path)[:len(*path)-1]
	}
	return false
}
0