Задача: 640. Solve the Equation
Сложность: medium

Решите заданное уравнение и верните значение 'x' в виде строки "x=#value". Уравнение содержит только операции '+', '-', переменную 'x' и ее коэффициент. Вы должны вернуть "No solution", если для уравнения нет решения, или "Infinite solutions", если для уравнения существует бесконечное количество решений. Если для уравнения существует ровно одно решение, мы убеждаемся, что значение 'x' является целым числом.

Пример:

Input: s = "*"
Output: 9

👨‍💻 Алгоритм:

1⃣Разделение уравнения: Разделите уравнение на левую и правую части относительно знака равенства '='.

2⃣Парсинг и упрощение: Пройдитесь по каждой части уравнения, упрощая ее до суммы коэффициентов 'x' и числовых значений.

3⃣Решение уравнения: Используйте уравнение вида ax + b = cx + d, чтобы решить для 'x'. Если коэффициенты 'x' равны и числовые значения равны, уравнение имеет бесконечное количество решений. Если коэффициенты 'x' равны, но числовые значения различны, решения нет. В противном случае вычислите значение 'x'.

😎 Решение:

class Solution {
    func solveEquation(_ equation: String) -> String {
        func parse(_ s: String) -> (Int, Int) {
            var coeff = 0
            var constPart = 0
            var sign = 1
            var num = 0
            var i = s.startIndex
            
            while i < s.endIndex {
                if s[i] == "+" {
                    sign = 1
                    i = s.index(after: i)
                } else if s[i] == "-" {
                    sign = -1
                    i = s.index(after: i)
                } else if s[i].isNumber {
                    num = 0
                    while i < s.endIndex && s[i].isNumber {
                        num = num * 10 + s[i].wholeNumberValue!
                        i = s.index(after: i)
                    }
                    if i < s.endIndex && s[i] == "x" {
                        coeff += sign * num
                        i = s.index(after: i)
                    } else {
                        constPart += sign * num
                    }
                } else if s[i] == "x" {
                    coeff += sign
                    i = s.index(after: i)
                }
            }
            return (coeff, constPart)
        }
        
        let parts = equation.split(separator: "=")
        let (leftCoeff, leftConst) = parse(String(parts[0]))
        let (rightCoeff, rightConst) = parse(String(parts[1]))
        
        let coeff = leftCoeff - rightCoeff
        let constPart = rightConst - leftConst
        
        if coeff == 0 {
            return constPart == 0 ? "Infinite solutions" : "No solution"
        }
        return "x=\(constPart / coeff)"
    }
}

Ставь 👍 и забирай 📚 Базу знаний