SeaCode

Appearance

Make equal elements Array ​

Problem Description ​

Given an array of all positive integers and an element “x”.

You need to find out whether all array elements can be made equal or not by performing any of the 3 operations: add x to any element in array, subtract x from any element from array, do nothing.

This operation can be performed only once on an element of array.

Problem Constraints

1<=|A|<=1e5
1<=A[i],x<=1e9

Input Format

First argument is array of integers .
Second argument is B which denotes the value of x.

Output Format

Return 1 if we can make all elements equal , otherwise return 0.

Example Input

Input 1:
A=[2,3,1]
X=1

Input 2:
A=[2,3,1]
X=2

Example Output

Ouput 1:
1

Ouput 2:
0

Example Explanation

Explanation 1:
WE can make all elements equal to 2.

Explanation 2:
There is no way to make all elements equal to 0.

Solution ​

swift
import Foundation

class Solution {
	func solve(_ A: inout [Int], _ B: inout Int) -> Int {
        let min = A.min()!
        let max = A.max()!
        if min == max {
            return 1
        }
        
        var target = -1
        
        if (min + B) == max || (min + 2*B) == max{
            target = (min + B)
        } else {
            return 0
        }
        
        for v in A {
            if v < target {
                if (v + B) != target {
                    return 0
                } 
            } else if v > target {
                if (v - B ) != target {
                    return 0
                }
            }
        }
        return 1
	}
}

References ​