TDT4145 Extendible hashing

exthash.py

# Simple script used in extendible hashing tasks for course TDT4145 at NTNU
import math
inputs = [int(n) for n in input("Paste list of keys, sep with ',': ").split(',')]

n = input("Enter hash function, k mod ")
if n == '':
    # Fallback if no hash function is given
    m = max(inputs)
    n = math.floor(math.sqrt(m))**2
else: n = int(n)
binlen = int(math.log(n, 2))

def hsh(k):
    return k % n

bin_vals = []
print("k", "h(k)", "bin", sep='\t')
for inp in inputs:
    bin_vals.append(bin(hsh(inp))[2:].zfill(binlen))
    print(inp, hsh(inp), bin_vals[-1], sep='\t')