640. 求解方程

640. 求解方程https://leetcode.cn/problems/solve-the-equation/

难度中等190

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

如果方程没有解或存在的解不为整数，请返回 "No solution" 。如果方程有无限解，则返回 “Infinite solutions” 。

题目保证，如果方程中只有一个解，则 'x' 的值是一个整数。

 

示例 1：

输入: equation = "x+5-3+x=6+x-2"
输出: "x=2"

示例 2:

输入: equation = "x=x"
输出: "Infinite solutions"

示例 3:

输入: equation = "2x=x"
输出: "x=0"

 

提示:

• 3 <= equation.length <= 1000
• equation 只有一个 '='. 
• 方程由绝对值在 [0, 100]  范围内且无任何前导零的整数和变量 'x' 组成。

class Solution {
    public String solveEquation(String equation) {
        int n=0,a=0;
        int Symbol = 1; // 记录符号
        int flag = 0;
        int tagN = 0; 

        for(int i=0;i<equation.length();i++)
        {
            if(equation.charAt(i)>='0' && equation.charAt(i)<='9') //数字
            {
                tagN = tagN*10 + (equation.charAt(i)-'0');
            }
            else if(equation.charAt(i)=='x')
            {
                if(i==0 || equation.charAt(i-1)=='-'||equation.charAt(i-1)=='+'||equation.charAt(i-1)=='=')  n += Symbol;
                else n += Symbol*tagN;
                tagN =0;
            } 
            else
            {
                if(equation.charAt(i)=='=') flag = 1;
                a = a+(Symbol*tagN);
                Symbol = 1;
                if(equation.charAt(i)=='-')
                {
                    if(flag==0) Symbol = -1;
                    if(flag==1) Symbol = 1;
                }
                else if(flag==1 &&( Symbol==1 || equation.charAt(i)=='+'))  Symbol = -1;
                tagN =0;
            }
        }
        if(equation.charAt(equation.length()-1)!='x') 
        {
            a = a+(Symbol*tagN*flag);
        }
        if(a==0 && n==0) return "Infinite solutions";
        if(a!=0 && n==0 || a%n!=0) return "No solution";
        if(a==0) return "x=0";
        if(a%n==0) a = a/n; 
        return "x="+(-a);
    }
}