Java实现 LeetCode 640 求解方程（计算器的加减法计算）

港控/mmm° 2023-07-20 12:26 8阅读 0赞

## 640. 求解方程 ##

求解一个给定的方程，将x以字符串"x=\#value"的形式返回。该方程仅包含’+’，’ - '操作，变量 x 和其对应系数。

如果方程没有解，请返回“No solution”。

如果方程有无限解，则返回“Infinite solutions”。

如果方程中只有一个解，要保证返回值 x 是一个整数。

示例 1：
    
    输入: "x+5-3+x=6+x-2"
    输出: "x=2"
    示例 2:
    
    输入: "x=x"
    输出: "Infinite solutions"
    示例 3:
    
    输入: "2x=x"
    输出: "x=0"
    示例 4:
    
    输入: "2x+3x-6x=x+2"
    输出: "x=-1"
    示例 5:
    
    输入: "x=x+2"
    输出: "No solution"

PS：  
标记一下正负，再用一个指针指到数字的第一位

class Solution { 
          public String solveEquation(String equation) { 
            int coff = 0, sum = 0, index = 0, sign = 1;
            int n = equation.length();
    
            for(int i=0;i<n;i++){ 
                char c = equation.charAt(i);
                if(c == '='){ 
                    if(index < i){ 
                        sum += Integer.valueOf(equation.substring(index, i)) * sign;
                    }
                    sign = -1;
                    index = i + 1;
                }else if(c == 'x'){ 
                    if(index == i || equation.charAt(i - 1) == '+'){ 
                        coff += sign;
                    }else if(equation.charAt(i - 1) == '-'){ 
                        coff -= sign;
                    }else{ 
                        coff += Integer.valueOf(equation.substring(index, i)) * sign;
                    }
                    index = i+1;
                }else if(c == '-' || c == '+'){ 
                    if(index < i){ 
                        sum += Integer.valueOf(equation.substring(index, i)) * sign;
                    }
                    index = i;
                }
            }
    
            if(index < n){ 
                sum += Integer.valueOf(equation.substring(index)) * sign;
            }
    
            if(sum == 0 && coff == 0) return "Infinite solutions";
            if(coff == 0) return "No solution";
            return "x=" + String.valueOf(-sum / coff);
        }
    }