codevs 计算器的改良

#include<iostream>
#include<cctype>
#include<vector>
#include<cstdio>
using namespace std;

//
int get_num(string s)
{
    int sum=;
    for(int i=;i<s.length();i++)
    {
        int k=s[i]-'';
        sum=sum*+k;
    }
    //cout<<sum<<endl;
    return sum;
}

int main()
{
    string s="6a-5+1=2-2a";
    cin>>s;
    vector<string> s0;
    //第一步，实现分离
    char ch;
    int loc=;
    for(int i=;i<s.length();i++)
    {
        if(s[i]=='=')
        {
            s0.push_back(s.substr(loc,i-loc));
            s0.push_back("=");
            loc=i+;
        }
        else if(s[i]=='+'||s[i]=='-')
        {
            string tmp=s.substr(loc,i-loc);
            s0.push_back(tmp);
            loc=i;
        }
    }
    s0.push_back(s.substr(loc,s.length()-loc));
    //for(int i=0;i<s0.size();i++)
    //    cout<<s0[i]<<endl;
    //分离后的计算
    int x=,c=;
    int f=;
    for(int i=;i<s0.size();i++){
        if(s0[i]=="=")
        {
            f=-;
            continue;
        }
        int kt=s0[i].length();
        if(isalpha(s0[i][kt-])){
            ch=s0[i][kt-];
            if(s0[i][]=='-')
                x-=f*get_num(s0[i].substr(,kt-));
            else if(s0[i][]=='+')
                x+=f*get_num(s0[i].substr(,kt-));
            else x+=f*get_num(s0[i].substr(,kt-));
        }
        else
        {
            if(s0[i][]=='-')
                c+=f*get_num(s0[i].substr(,kt));
            else if(s0[i][]=='+')
                c-=f*get_num(s0[i].substr(,kt));
            else c-=f*get_num(s0[i].substr(,kt));
        }
    }
    //cout<<x<<endl;
    //cout<<c<<endl;
    printf("%c=%.3f",ch,c*1.0/x);
    return ;
}

一道很简单的题目，但是花了超级多的时间。

http://codevs.cn/problem/1015/

情况很多要依次处理，我是先把他们拆开打散，然后做相应的处理。

主要用到了C++ STL 中的string数据结构类型