BOJ / 9461번 / 파도반 수열 [Go] [Python3]

👩🏻‍💻 문제

9461번: 파도반 수열

---

✍🏻 풀이

🎨 Go

package main

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

func main() {
	reader := bufio.NewReader(os.Stdin)
	var t int
	fmt.Fscanln(reader, &t)
	writer := bufio.NewWriter(os.Stdout)
	defer writer.Flush()

	for i :=0; i<t; i++ {
		var n int
		fmt.Fscanln(reader, &n)
		lengths := calculate(n)
		fmt.Fprintf(writer, "%d\n", lengths[n-1])
	}
}

func calculate(n int) (lengths []int) {
	lengths = append(lengths, 1, 1, 1)
	if n <= 3 {
		return
	}
	for i := 3; i < n+1 ; i++ {
		lengths = append(lengths, lengths[i-2]+ lengths[i-3])
	}
	return
}

🎨 Python3

import sys

def calculate(n):
    lengths = [1, 1, 1]
    if n <= 3:
        return lengths
    for i in range(3, n+1):
        lengths.append(lengths[i-2]+lengths[i-3])
    return lengths
        
if __name__ == "__main__":
    t = int(sys.stdin.readline())
    for i in range(t):
        n = int(sys.stdin.readline())
        lengths = calculate(n)
        print(lengths[n-1])