[LeetCode] Solve the Equation

Posted immjc

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[LeetCode] Solve the Equation相关的知识,希望对你有一定的参考价值。

Solve a given equation and return the value of x in the form of string "x=#value". The equation contains only ‘+‘, ‘-‘ operation, the variable x and its coefficient.

If there is no solution for the equation, return "No solution".

If there are infinite solutions for the equation, return "Infinite solutions".

If there is exactly one solution for the equation, we ensure that the value of x is an integer.

Example 1:

Input: "x+5-3+x=6+x-2"
Output: "x=2"

 

Example 2:

Input: "x=x"
Output: "Infinite solutions"

 

Example 3:

Input: "2x=x"
Output: "x=0"

 

Example 4:

Input: "2x+3x-6x=x+2"
Output: "x=-1"

 

Example 5:

Input: "x=x+2"
Output: "No solution"

 

模拟方程式的求解。需要对符号,字母,数字进行判断后统计。引入两个变量,一个记录x的系数,一个记录数字的总和。维护一个符号变量用于计算数字的总和,在等号左边符号变量为正,在等号右边符号变量为负。

class Solution {
public:
    string solveEquation(string equation) {
        int n = equation.size(), sign = 1, coeff = 0, sum = 0, i = 0;
        for (int j = 0; j < n; j++) {
            // cal the sum of equation.
            if (equation[j] == + || equation[j] == -) {
                if (j > i) 
                    sum += sign * stoi(equation.substr(i, j-i));
                i = j;
            }
            // cal the coeff of equation.
            else if (equation[j] == x) {
                if (i == j || equation[j - 1] == +)
                    coeff += sign;
                else if (equation[j - 1] == -)
                    coeff -= sign;
                else 
                    coeff += sign * stoi(equation.substr(i, j - i));
                i = j + 1;
            }
            // flip the sign after meeting ‘=‘.
            else if (equation[j] == =) {
                // calc the sum if the num near the ‘=‘
                if (j > i) 
                    sum += sign * stoi(equation.substr(i, j - i));
                sign = -1;
                i = j + 1;
            }
        }
        // cal the last num of equation.
        if (i < n) 
            sum += sign * stoi(equation.substr(i));
        // judge the coeff and sum.
        if (coeff == 0 && sum == 0) 
            return "Infinite solutions";
        if (coeff == 0 && sum != 0) 
            return "No solution";
        int res = -sum/coeff;
        return "x=" + to_string(res);
    }
};
// 3 ms

 

 

以上是关于[LeetCode] Solve the Equation的主要内容,如果未能解决你的问题,请参考以下文章

[LeetCode] Solve the Equation 解方程

leetcode640. Solve the Equation

Reverse Words in a String leetcode

Codeforces Round #648 (Div. 2).D Solve The Maze

注册gitLab时报错There was an error with the reCAPTCHA. Please solve the reCAPTCHA again.

Codeforces Round #648 (Div. 2) D - Solve The Maze bfs