A python function to memoize DP problems

memoize.py

def memoize(cache_index=0):
	def decorator(function):
		cache = {}
		def proxy(*args, **kwargs):
			key = args[cache_index]
			if key not in cache:
				cache[key] = function(*args, **kwargs)
			return cache[key]
		return proxy
	return decorator