Leetcode 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"

 public class Solution {
     public String coeff(String x) {
         if (x.length() > 1 && x.charAt(x.length() - 2) >= '0' && x.charAt(x.length() - 2) <= '9')
             return x.replace("x", "");
         return x.replace("x", "1");
     }
     public String solveEquation(String equation) {
         String[] lr = equation.split("=");
         int lhs = 0, rhs = 0;
         for (String x: breakIt(lr[0])) {
             if (x.indexOf("x") >= 0) {
                 lhs += Integer.parseInt(coeff(x));
             } else
                 rhs -= Integer.parseInt(x);
         }
         for (String x: breakIt(lr[1])) {
             if (x.indexOf("x") >= 0)
                 lhs -= Integer.parseInt(coeff(x));
             else
                 rhs += Integer.parseInt(x);
         }
         if (lhs == 0) {
             if (rhs == 0)
                 return "Infinite solutions";
             else
                 return "No solution";
         }
         return "x=" + rhs / lhs;
     }
     public List < String > breakIt(String s) {
         List < String > res = new ArrayList < > ();
         String r = "";
         for (int i = 0; i < s.length(); i++) {
             if (s.charAt(i) == '+' || s.charAt(i) == '-') {
                 if (r.length() > 0)
                     res.add(r);
                 r = "" + s.charAt(i);
             } else
                 r += s.charAt(i);
         }
         res.add(r);
         return res;
     }
 }