Equilibrium Index problem in F#

You can become a serverless blackbelt. Enrol to my 4-week online workshop Production-Ready Serverless and gain hands-on experience building something from scratch using serverless technologies. At the end of the workshop, you should have a broader view of the challenges you will face as your serverless architecture matures and expands. You should also have a firm grasp on when serverless is a good fit for your system as well as common pitfalls you need to avoid. Sign up now and get 15% discount with the code yanprs15!

A good friend pointed me to a nice coding challenge she encountered – finding the Equilibrium Index of an array in O(n) time.

After a quick discussion we ended up with 2 solutions, and naturally I had to code them up in F#

 

Solution 1

This is the same approach to the one you saw in the O(n) solution I posted the other day for the Multiply Others problem.

Here, we’ll create two temporary arrays – one that’s the running sum from the front, and the other the running sum from the rear, both are O(n) operations.

Once we have both, we can return all the indices where the front and rear arrays are equal.

(the questions usually only ask for one, but it’s nice to see all of them :-P)

This is a O(n) solution in both time and space.

 

Solution 2

A more space efficient, O(1), solution is to:

  • do one pass to sum the array (as the sum to the right)
  • then starting from the front and iteratively subtract elements from that sum until you find an equilibrium index

for example, given the input array [ -1; 3; -4; 5; 1; -6; 2; 1 ], the sum is 1, so if we start from the front of the array:

  • -1 : sum to the left is 0 (no elements), sum to the right is 1 – -1 = 2, no match
  • 3 : sum to the left is -1, sum to the right is 2 – 3 = -1, match, so index 1 is an equilibrium index
  • -4 : sum to the left is -1 + 3 = 2, sum to the right is -1 – -4 = 3, no match
  • 5 : sum to the left is 2 + -4 = -2, sum to the right is 3 – 5 = -2, match, index 3 is also an equilibrium index
  • 1 : sum to the left is -2 + 5 = 3, sum to the right is -2 – 1 = -3, no match
  • -6 : sum to the left is 3 + 1 = 4, sum to the right is -3 – -6 = 3, no match
  • 2 : sum to the left is 4 + -6 = -2, sum to the right is 3 – 2 = 1, no match
  • 1 : sum to the left is -2 + 2 = 0, sum to the right is 1 – 1 = 0, match, index 7 is also an equilibrium index

So, applying this in F# I ended up with this:

notice that I’m generating the output array via comprehensions, this can look a bit odd to people new to F# so I tend to shy away from it and usually go for seq { … } |> Seq.toArray instead.

 

Try it Yourself

 

Links

Liked this article? Support me on Patreon and get direct help from me via a private Slack channel or 1-2-1 mentoring.
Subscribe to my weekly newsletter


Hi, I’m Yan. I’m an AWS Serverless Hero and I help companies go faster for less by adopting serverless technologies successfully.

Are you struggling with serverless or need guidance on best practices? Do you want someone to review your architecture and help you avoid costly mistakes down the line? Whatever the case, I’m here to help.

Hire me.


Skill up your serverless game with this hands-on workshop.

My 4-week Production-Ready Serverless online workshop is back!

This course takes you through building a production-ready serverless web application from testing, deployment, security, all the way through to observability. The motivation for this course is to give you hands-on experience building something with serverless technologies while giving you a broader view of the challenges you will face as the architecture matures and expands.

We will start at the basics and give you a firm introduction to Lambda and all the relevant concepts and service features (including the latest announcements in 2020). And then gradually ramping up and cover a wide array of topics such as API security, testing strategies, CI/CD, secret management, and operational best practices for monitoring and troubleshooting.

If you enrol now you can also get 15% OFF with the promo code “yanprs15”.

Enrol now and SAVE 15%.


Check out my new podcast Real-World Serverless where I talk with engineers who are building amazing things with serverless technologies and discuss the real-world use cases and challenges they face. If you’re interested in what people are actually doing with serverless and what it’s really like to be working with serverless day-to-day, then this is the podcast for you.


Check out my new course, Learn you some Lambda best practice for great good! In this course, you will learn best practices for working with AWS Lambda in terms of performance, cost, security, scalability, resilience and observability. We will also cover latest features from re:Invent 2019 such as Provisioned Concurrency and Lambda Destinations. Enrol now and start learning!


Check out my video course, Complete Guide to AWS Step Functions. In this course, we’ll cover everything you need to know to use AWS Step Functions service effectively. There is something for everyone from beginners to more advanced users looking for design patterns and best practices. Enrol now and start learning!