题目链接:求解方程
题目描述:
求解一个给定的方程,将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”
分析:技巧,将-号替换成±,再利用+拆分,这样可以直接忽略正负数直接处理,简化处理。还要注意无解和无限解的情况。
代码:
class Solution {
static int[] parse(String str){
String[] element = str.replace("-","+-").split("\\+");
int sumX = 0 ,sumConst = 0;
for (int i = 0; i < element.length; i++) {
if(element[i].equals("x")) sumX++;
else if(element[i].equals("-x")) sumX--;
else if(element[i].contains("x")) sumX += Integer.parseInt(element[i].substring(0,element[i].length()-1));
else if(!element[i].equals("")) sumConst += Integer.parseInt(element[i]);
else continue;
}
return new int[]{sumX,sumConst};
}
public String solveEquation(String equation) {
String[] all = equation.split("=");
int[] left = parse(all[0]);
int[] right = parse(all[1]);
int ansX = left[0]-right[0];
int ansConst = right[1]-left[1];
if(ansX!=0) return "x=" + ansConst/ansX;
else if(ansConst == 0) return "Infinite solutions";
return "No solution";
}
}