求解方程

求解一个给定的方程，将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;

     }

 }

巴特西

Leetcode 640.求解方程

求解方程

最新文章

热门文章