zoukankan      html  css  js  c++  java
  • [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
  • 相关阅读:
    linux最常用命令记录(一)
    2020centos解决“nginx 403 Forbidden"错误的故事
    nginx显示静态html爆502 bad gateway的错误提示
    codeigniter框架的使用感受和注意事项
    网闸
    抗DDOS防火墙
    负载均衡
    上网行为管理
    漏洞扫描系统
    网络分析系统
  • 原文地址:https://www.cnblogs.com/immjc/p/7158013.html
Copyright © 2011-2022 走看看