BOJ / 14606번 / 피자 (Small) [Go][Python3]

👩🏻‍💻 문제

14606번: 피자 (Small)

---

✍🏻 풀이

🎨 Go

// https://www.acmicpc.net/problem/14606
package main

import (
	"bufio"
	"fmt"
	"os"
)

var (
	dp []int
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	writer := bufio.NewWriter(os.Stdout)
	defer writer.Flush()

	var n int
	fmt.Fscanln(reader, &n)
	dp = make([]int, n+1)
	fmt.Fprintln(writer, joy(n))
}

func joy(n int) int {
	if n == 1 {
		return 0
	}
	if n == 2 {
		return 1
	}
	dp[2] = 1
	dp[n] = (n/2)*(n-n/2) + joy(n/2) + joy(n-n/2)
	return dp[n]
}

🎨 Python3

# https://www.acmicpc.net/problem/14606
import sys

dp = []

def joy(n):
    if n == 1:
        return 0
    if n == 2:
        return 1
    dp[2] = 1
    dp[n] = (n//2)*(n-n//2) + joy(n//2) + joy(n-n//2)
    return dp[n]

if __name__ == "__main__":
    n = int(sys.stdin.readline())
    dp = [0 for _ in range(n+1)]
    print(joy(n))