Java實現 LeetCode 640 求解方程（計算器的加減法計算）

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);
    }
}