Pentagonal numbers are generated by the formula, Pn=n(3n-1)/2. The first ten pentagonal numbers are:
1, 5, 12, 22, 35, 51, 70, 92, 117, 145, …
It can be seen that P4 + P7 = 22 + 70 = 92 = P8. However, their difference, 70 – 22 = 48, is not pentagonal.
Find the pair of pentagonal numbers, Pj and Pk, for which their sum and difference is pentagonal and D = |Pk – Pj| is minimised; what is the value of D?
open System.Collections.Generic // define the Pentagonal function P let P n = n*(3*n-1)/2 // use a dictionary as cache let mutable cache = new Dictionary<int, int>() [1..5000] |> List.iter (fun n -> cache.[n] <- P n) // function to check if a number is pentagonal by looking at the cached values let isPentagonal n = cache.ContainsValue n // predicate function to check if Pk and Pj's sum and diff are both pentagonal let predicate (k, j) = let pk = cache.[k] let pj = cache.[j] isPentagonal (pj + pk) && isPentagonal (pk - pj) // the sequence of k, j pairs to check let kjSeq = [1..5000] |> List.collect (fun k -> [1..k-1] |> List.rev |> List.map (fun j -> (k, j))) // get the first pair of k, j whose sum and difference are both pentagonal let (k, j) = kjSeq |> Seq.filter predicate |> Seq.head let answer = (P k) - (P j)
Having experimented with a few alternative implementations which did not use a mutable dictionary and waited impatiently for the code to never return I settled on this solution which caches the first 5000 values in the pentagonal number sequence for quick lookup later on in the solution.
Going into more detail about the solution itself, to ensure |Pk – Pj| is minimised, the value of k and the difference between k and j must be minimised too as the further apart Pk and Pj are in the pentagonal sequence the bigger their difference will be and the bigger k is the bigger Pk will be too.
The rest of the algorithm here is simple, for each value k (between 1 and 5000, being the size of the cache) check for each j where 1 <= j < k, starting from the biggest j, if the sum and difference of Pk and Pj are both pentagonal. The first k and j which satisfy this predicate will have the smallest difference between Pk and Pj.
I’m an AWS Serverless Hero and the author of Production-Ready Serverless. I have run production workload at scale in AWS for nearly 10 years and I have been an architect or principal engineer with a variety of industries ranging from banking, e-commerce, sports streaming to mobile gaming. I currently work as an independent consultant focused on AWS and serverless.
Here is a complete list of all my posts on serverless and AWS Lambda. In the meantime, here are a few of my most popular blog posts.
- Lambda optimization tip – enable HTTP keep-alive
- You are thinking about serverless costs all wrong
- Many faced threats to Serverless security
- We can do better than percentile latencies
- I’m afraid you’re thinking about AWS Lambda cold starts all wrong
- Yubl’s road to Serverless
- AWS Lambda – should you have few monolithic functions or many single-purposed functions?
- AWS Lambda – compare coldstart time with different languages, memory and code sizes
- Guys, we’re doing pagination wrong