Same Stripes ServiceNow Python Interview Question

You are given an matrix. Your task is to determine if the matrix has diagonal stripes where all elements in each diagonal from top-left to bottom-right are of the same stripe—that is, they are identical.

In this context, each diagonal stripe runs from the top-left corner to the bottom-right corner of the matrix. Check if every diagonal stripe consists entirely of the same number.

Return if all diagonal stripes are of the same stripe, otherwise return .

Example #1

Input: matrix =

Output:

Explanation:
In this grid, the diagonals are:

All elements in each diagonal ar identical. Thus, the answer is .

Example #2

Input: matrix =

Output:

![Same Stripe DataLemur Example 2]

Explanation:
The diagonal does not consist of elements of the same stripe.

Tracking diagonal elements

What if we could figure out which elements in a matrix are on the same diagonal stripe? This would make the problem easier. If we can group the elements on the same diagonal together, we could just check if all elements in each diagonal are equal.

But how do we determine if elements are on the same diagonal stripe? Let's use this example:

• The elements are on the same diagonal. Their positions are . The difference between their row and column () is the same for all of them: .
• Similarly, the elements are also on the same diagonal. Their positions are , and their difference is also the same: .

Implementation:

We can use a dictionary to track diagonal elements efficiently. The key of the dictionary will be the difference , and the value will be the first element we encounter for that diagonal. Then, we'll compare subsequent elements on that diagonal to the stored value. If any mismatch is found, we return immediately. If we finish traversing the matrix without finding a mismatch, we return .

Here's how the approach works:

1. If is not in the dictionary, we store the current element in the dictionary as the first element of this diagonal.
2. If is already in the dictionary, we compare the stored element with the current one. If they are different, we return immediately.
3. After the entire matrix is traversed, return if no mismatches were found.

Here is the full code:

Space-Optimized Approach:

We can optimize the space by not storing the elements at all. Instead, we directly compare the current element with the previous element in the same diagonal. The previous element for any position would be (i.e., the element "northwest" of the current one). If the current element doesn't match the previous one, we return .

• The elements are in the same diagonal. When we’re at (position ), we check its previous element in the diagonal, which is (position ).
• If any mismatch occurs in this check, we return .

Here is the full code: