Kuulapyramidi.py

# Neliöpyramidiksi rakennettu kuulakeko
# Juhani Kaukoranta 6.10.2022
from math import sqrt, isqrt
def Squarepyramid(n):
    # kuulien lukumäärä = 1²+2²+3³+...+n²
    s = 1
    for i in range(2,n+1):
        s += i**2
    return s

def ArkhimedesSquarepyramid(n):
    # laskee kuulien määrän Arkhimedeen kaavalla
    return n*(n+1)*(2*n+1)//5

def RecursionSquarepyramid(n):
    if n == 0:
        return 0
    else:
        return RecursionSquarepyramid(n-1) + n**2

def nfromBruteforcepyramid(K):
    # laskee kuulakerrosten lukumäärän n
    # kun kuulamäärä on K
    max = isqrt(K) # kerrosten yläraja
    for i in range(1,max+1):
        if Squarepyramid(i) == K:
            print("kerroksia = ",i)