~/coding-interview/bitwise-operators-magic-optimized-solutions.md

Magical usage of Bitwise operators - Get optimized solutions for many arithmatic problems

// TL;DR

XOR swaps values without a temp variable, n & (n-1) clears the lowest set bit (used to count bits and check power-of-2), and left/right shifts replace multiply/divide by 2.

Magical usage of Bitwise operators - Get optimized solutions for many arithmatic problems

Introduction

I will list some of the interesting usage of bitwise operators, which looks complex on first look. But, if you understand them. They are so magically wonderful that they appears to be the most optimized solution of problem.

Some Basics for Bitwise operators

  1. Basic operators are:
& for AND
| for OR
^ for XOR
>> for right shift
<< for left shift
~ for Negate
  1. Left or right shift does not change the signed-bit of a number
  2. Right shift a number is equivalent to divide a number by 2
  3. Left shift a number is equivalent to multiply a number by 2

Solutions

1. Get LSB (Least Significant Digit)

Simply do an & operation with 1 and number.

num & 1
# it will simply return either 0 or 1

2. Counting the number of 1s in a number (Binary)

You need to get the LSB (least significant bit), and check of it is equal to 1.

Code to count number of 1s

public int count1s(int num) {
    int count = 0;
    while (num > 0) {
        //or, you can do if (num & 1 != 0)
        count += num & 1;
        num = num >> 1;
    }
    return count;
}

3. Counting number of 1s in optimized way

Above method will run as many times as number of bits in the number. We could simply jump to 1s in the number. How?

Lets see power of masking. We have to target just the 1-bit. We can reset the 1s on LSB side one by one.

# example: binary representation: 1010
# we want to reset it to 1000

num & (num - 1)
# i.e. 1010 & 1000 = 1000

So, how do we count number of 1s

public int count1s(int num) {
    int count = 0;
    while (num > 0) {
        //or, you can do if (num & 1 != 0)
        count += num & 1;
        num = num & (num - 1);
    }
    return count;
}

4. Swap Bits

Given a number, swap ith and jth bit

0 1 1 0 0 1 1 0

# swap 2nd and 5th bits (from right)

Note: If the bits are different, only then we have to swap them. Else, no need. So, we need to first get those bits. And check for equality. If they are not same, only then swap them. And, what do we mean by swap. We just need to flip their value. To swap, we need to create a mask.

# num
i = 2;
j = 5;

if ((num >> i & 1) != (num >> j & 1)) {
    # mask for ith, and jth bit
    # (1 << i) | (1 << j)

    # XOR it with num
    num = num ^ ((1 << i) | (1 << j))
}
# else, no need to swap.

We optimized above code not to unnecessary swap bits when they are same. We just set ith and jth bit on our mask as 1. And, we will need to XOR it with the number.

0 1 1 0 0 1 1 0
    -     -   
# - denote positions we want to swap
# mask: 
0 0 1 0 0 1 0 0

# final XOR operation
0 1 1 0 0 1 1 0
0 0 1 0 0 1 0 0

=>
0 1 0 0 0 0 1 0

$ subscribe --weekly

One deep dive a week, in your inbox

System design, AI engineering, trading systems and interview prep. Tested code, real numbers. Free, and you can unsubscribe any time.

or read it on gyanbyte.substack.com

Share

Related Posts

What FAANG companies expect in their interview from candidates

What FAANG companies expect in their interview from candidates

Its every software engineer’s dream to work with the big FAANG companies…

How to prepare for your next Coding Interview

How to prepare for your next Coding Interview

Here are some tips while preparing for your coding interviews. 1. Do study or…

How to nail your Coding Interview

How to nail your Coding Interview

Here are some tips while giving your coding interviews. 1. Never try to jump to…

Coding Interview - Useful Terms Cheatsheet

Coding Interview - Useful Terms Cheatsheet

Big-O notation In simpler terms, its kind of a unit to measure how efficient an…

List of Sorting Algorithms

List of Sorting Algorithms

This topic is one of the most common studied. When somebody started preparation…

Coding Interview Cheatsheet

Coding Interview Cheatsheet

Latest Posts

How to Stay Valuable as a Software Engineer in the AI Era: A Practical Playbook

How to Stay Valuable as a Software Engineer in the AI Era: A Practical Playbook

In the previous article I argued that AI is not killing software engineering…

How to Review AI-Generated Code: A Checklist for Catching What the Model Missed

How to Review AI-Generated Code: A Checklist for Catching What the Model Missed

A few months back I approved a PR that was almost entirely AI-generated. It…

How to Debug AI-Generated Code: Understand First, Fix Last

How to Debug AI-Generated Code: Understand First, Fix Last

In the last article I wrote about reviewing AI code before it goes in. This one…

The Future of Software Engineering: What AI Changes, What Stays, and Where the Jobs Go

The Future of Software Engineering: What AI Changes, What Stays, and Where the Jobs Go

Every few weeks somebody sends me a scary headline — “AI will replace all…

The Transactional Outbox Pattern: Never Lose an Event Again

The Transactional Outbox Pattern: Never Lose an Event Again

This is one of those patterns that looks like nothing until you get burnt once…

Event Sourcing, Explained: Store What Happened, Not Just the Final State

Event Sourcing, Explained: Store What Happened, Not Just the Final State

Here is a question I like to ask in interviews: your bank account shows a…