题目内容
求解一个给定的方程,将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 只有一个 ‘=’.
equation 方程由整数组成,其绝对值在 [0, 100] 范围内,不含前导零和变量 ‘x’
来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/solve-the-equation
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
首先,我们知道这个是一个一元一次方程,所以他的求解是很简单的,换句话说,如果我们的输入是告诉你左边x的系数,右边的常数,形如ax=b的形式,我们自然可以轻松给出答案。而我们这个题目的目的,就是把他的输入调整成这个格式。
首先,我们以=为界,将字符串左右分开,左右分开统计,并不影响复杂度,我们依旧是一个字符过一遍,然后,遍历左右的字符串,用string类型变量记录每一个遍历的字符,以±符号作为处理这个string变量的标志,因为可能开头就有-号,所以要提前注明这个string的长度不能为0,然后判定string的最后一位是不是x,如果是,就把掐头去尾,去掉符号和X,中间部分转数字,作为系数统计,如果不是,就只去掉前面的符号,做常数统计,左右两边各有一个系数,一个常数。最后的情况为ax+b=cx+d。
最后的操作虽然简单,但是也是我忽略了小细节的地方,我们这时候把x放左边,常数放右边。然后要先判断有没有解,是否会有无限解,这里的判定条件,我错了两次,主要是没有理清楚,最后才整理出来,先看系数,再看常数的判定逻辑。
包括最后的输出结果部分,我当时是没有用这个to_string的函数,而是直接’0’+anss了,但是这个我们现在来看显然是不对的,一来结果可能为负,二来结果大概率大于10,这样就很麻烦,我就直接改用to_string了。
代码
class Solution {
public:
string solveEquation(string equation) {
int biao=equation.find('=');
string zuo=equation.substr(0,biao);
string you=equation.substr(biao+1,equation.length());
int zuoxi=0;int zuoshu=0;
int youxi=0;int youshu=0;
vector<string> os;vector<string> os2;
string duan="";
for(int i=0;i<zuo.length();i++){
if(zuo[i]=='+'||zuo[i]=='-'){
if(duan.length()==0){
duan+=zuo[i];
continue;
}
else{
os.push_back(duan);
duan="";
}
}
duan+=zuo[i];
}
os.push_back(duan);
duan="";
for(int i=0;i<you.length();i++){
if(you[i]=='+'||you[i]=='-'){
if(duan.length()==0){
duan+=you[i];
continue;
}
else{
os2.push_back(duan);
duan="";
}
}
duan+=you[i];
}
os2.push_back(duan);
duan="";
for(int i=0;i<os.size();i++){
if(os[i][os[i].length()-1]=='x'){
if(os[i][0]=='+'||os[i][0]=='-'){
if(os[i].length()==2){
int num=os[i][0]=='+'?1:-1;
zuoxi+=num;
continue;
}
string xishu=os[i].substr(1,os[i].length()-2);
int num=os[i][0]=='+'?stoi(xishu):-stoi(xishu);
zuoxi+=num;
}
else{
if(os[i].length()==1){
zuoxi+=1;
continue;
}
string xishu=os[i].substr(0,os[i].length()-1);
int num=stoi(xishu);
zuoxi+=num;
}
}
else{
if(os[i][0]=='+'||os[i][0]=='-'){
string xishu=os[i].substr(1,os[i].length()-1);
int num=os[i][0]=='+'?stoi(xishu):-stoi(xishu);
zuoshu+=num;
}
else{
string xishu=os[i];
int num=stoi(xishu);
zuoshu+=num;
}
}
}
for(int i=0;i<os2.size();i++){
if(os2[i][os2[i].length()-1]=='x'){
if(os2[i][0]=='+'||os2[i][0]=='-'){
if(os2[i].length()==2){
int num=os2[i][0]=='+'?1:-1;
youxi+=num;
continue;
}
string xishu=os2[i].substr(1,os2[i].length()-2);
int num=os2[i][0]=='+'?stoi(xishu):-stoi(xishu);
youxi+=num;
}
else{
if(os2[i].length()==1){
youxi+=1;
continue;
}
string xishu=os2[i].substr(0,os2[i].length()-1);
int num=stoi(xishu);
youxi+=num;
}
}
else{
if(os2[i][0]=='+'||os2[i][0]=='-'){
string xishu=os2[i].substr(1,os2[i].length()-1);
int num=os2[i][0]=='+'?stoi(xishu):-stoi(xishu);
youshu+=num;
}
else{
string xishu=os2[i];
int num=stoi(xishu);
youshu+=num;
}
}
}
int x_xi=zuoxi-youxi;
int shu=youshu-zuoshu;
string ans="x=";
if(x_xi==0){
if(shu!=0){
return "No solution";
}
return "Infinite solutions";
}
else{
int anss=shu/x_xi;
// if(anss<0){
// ans+='-';
// anss=-anss;
// }
string insert=to_string(anss);
ans+=insert;
}
return ans;
}
};