二次方程计算器

题目描述
设计一个二次方程计算器

输入描述:
每个案例是关于x的一个二次方程表达式，为了简单，每个系数都是整数形式。

输出描述:
每个案例输出两个实数（由小到大输出，中间由空格隔开），保留两位小数;如果无解，则输出“No Solution”。

示例1
输入
x^2+x=3x+4
输出
-1.24 3.24

题目解析：查找所有二次项，一次项和常数项，并且将各自的系数组合在一起，不过有的在等号左边，有的在右边，所以移动时需要变符号。每次记住等号的位置，进行查找。

代码：

#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#include<vector>
#include<map>
#include<iomanip>
using namespace std; 

//很恶心，因为要提取的比较多 ,就是分别提出二次项，一次项系数，但是因为可能有多个一次项，二次项
//所以要合并系数，需要根据此时等号的位置，还有可能方程式开头的系数有符号，或者缺少一次项等等，稍微注意 

int computernum(string number,char fuhao){	
	if(number == ""){
		number = "1";
	} 
	int count = 0;
	int sum = 0;
	for(int i = number.size() - 1;i >= 0;i--){
		sum += (number[i] - '0') * pow(10,count);
		count++;
	}
	if(fuhao == '-'){
		sum =  -sum;
	}
	return sum; 
}

int reindexequal(vector<pair<string,char>> vc1){   //确定等号的位置 
	int index = -1;
	for(int i = 0;i< vc1.size(); i++){
		if(vc1[i].first == "="){
			index = i;
			break;
		}
	}
	vc1.clear();
	return index;
}

int main()
{
	string function;
	vector<pair<string,char>> vc;
	while(cin >> function){
		int indexfuhao = 0;
		char fuhao;
		for(int i = 0 ; i < function.size();i++){                                     //拆分数组保存 
			if(function[i] == '+' || function[i] == '-' || function[i] == '='){
				if(i == 0){                 //开始有符号，记住符号 
					fuhao = function[i];
					indexfuhao = i + 1;
					continue;
				}else if(indexfuhao == 0){
					fuhao = '+';
				}else if(fuhao == '='){            // =时不用保存了 
					if(i == indexfuhao){          // 判断等号之后就是符号 
						if(function[i] == '+' || function[i] == '-'){
							fuhao = function[i];
							indexfuhao = i + 1;
							continue;
						}
					}else{                       //等号之后不是符号，修改当前的符号值(当前为等号),如果是别的符号，在if中已经判断修改； 
						fuhao = '+';
					}
				}
				string temp = function.substr(indexfuhao,i-indexfuhao);  //substr，第一个参数开始下标，第二个参数长度
				vc.push_back(make_pair(temp,fuhao));
				fuhao = function[i];
				if(fuhao == '='){         //放入等号 
					vc.push_back(make_pair("=",'='));
				}
				indexfuhao = i + 1;
			}
		}
		if(function[indexfuhao-1] == '='){         //保存最后一个数字 
			fuhao = '+';
			vc.push_back(make_pair(function.substr(indexfuhao,function.size()-indexfuhao),fuhao));
		}else{
			fuhao = function[indexfuhao-1];
			vc.push_back(make_pair(function.substr(indexfuhao,function.size()-indexfuhao),fuhao));
		}
		
		/*查看数组内容 
		for(int i = 0 ; i <vc.size();i++){
			cout << vc[i].first << " " << vc[i].second << endl;
		}*/
		
		int indexequal = reindexequal(vc);  //确定等号位置 ，（方便移动，等号右边移动时取反） 
		int a = 0,b = 0,c = 0;
		for(int i = 0 ; i < vc.size();i++){
			if(vc[i].first.find("x^2") != string::npos){  //找到二次项 ,得到系数a 
				if(i < indexequal){
					a += computernum(vc[i].first.substr(0,vc[i].first.find("x^2")),vc[i].second);
				}else{
					a += -computernum(vc[i].first.substr(0,vc[i].first.find("x^2")),vc[i].second);
				}
				vc.erase(vc.begin() + i);   //删去二次项 （因为二次项值find("x")又会找到二次项） 
			}
		}
		
		indexequal = reindexequal(vc);
		for(int i = 0 ; i < vc.size();i++){
			if(vc[i].first.find("x") != string::npos){  //找到一次项 ,得到系数b 
				if(i < indexequal){
					b += computernum(vc[i].first.substr(0,vc[i].first.find("x")),vc[i].second);
				}else{
					b += -computernum(vc[i].first.substr(0,vc[i].first.find("x")),vc[i].second);
				}
				vc.erase(vc.begin() + i);   //删去一次项 
			}
			
		}
		
		indexequal = reindexequal(vc);
		for(int i = 0 ; i < vc.size();i++){      //计算常数项 
			if(vc[i].first != "="){
				if(i < indexequal){
					c += computernum(vc[i].first,vc[i].second); 
				}else{
					c += -computernum(vc[i].first,vc[i].second); 
				}
			}
		}
		
		//cout << a << " " << b << " " << c;   //查看系数 
		
		double rst1,rst2;                        //计算结果 
		cout << fixed << setprecision(2);
		if(b * b - 4 * a * c >= 0){
			rst1 = (-b + sqrt(b * b - 4 * a * c)) / (2 * a);
			rst2 = (-b - sqrt(b * b - 4 * a * c)) / (2 * a);
			if(rst1 == 0){                 //这是为了通过牛客网 
				cout << rst2 << " " << -rst1 << endl;
			}else{
				cout << rst1 << " " << rst2 << endl;	
			}
			
		}else{
			cout << "No Solution" << endl;
		}
		vc.clear();
	}
    return 0;
}