51Nod1005 大数加法

题目链接: 51Nod1005 大数加法

#include<iostream>
#include<vector>
using namespace std;
vector<int> add(vector<int> &A,vector<int> &B)
{
	vector<int> res;
	int t=0;
	for(int i=0;i<A.size()||i<B.size();++i)
	{
		if(i<A.size())	t+=A[i];
		if(i<B.size())	t+=B[i];
		res.push_back(t%10);
		t/=10;
	}
	while(t)
	{
		res.push_back(t%10);
		t/=10;
	}
	while(res.size()>1&&res.back()==0)	res.pop_back();
	return res;
}
vector<int> sub(vector<int> &A,vector<int> &B)
{
    vector<int> C;
    int t=0;
    for(int i=0;i<A.size();++i)
    {
        t+=A[i];
        if(i<B.size())
            t-=B[i];
        C.push_back((t+10)%10);
        t=(t<0?-1:0);
    }
    while(C.size()>1&&C.back()==0)  C.pop_back();
    return C;
}
bool cmp(vector<int> &A,vector<int> &B)
{
    if(A.size()!=B.size())   return A.size()>B.size();
    int len=A.size();
    for(int i=len-1;i>=0;--i)
    {
        if(A[i]!=B[i])
            return A[i]>B[i];
    }
    return false;
}
int main()
{
	string a,b;
	cin>>a;
	cin>>b;
	vector<int> ans,A,B;
	if(a[0]=='-'&&b[0]!='-')
	{
		for(int i=a.size()-1;i>=1;--i)
			A.push_back(a[i]-'0');
		for(int i=b.size()-1;i>=0;--i)
			B.push_back(b[i]-'0');
		if(cmp(B,A))
			ans=sub(B,A);
		else
		{
			cout<<"-";
			ans=sub(A,B);
		}
	}
	else if(a[0]=='-'&&b[0]=='-')
	{
		for(int i=a.size()-1;i>=1;--i)
			A.push_back(a[i]-'0');
		for(int i=b.size()-1;i>=1;--i)
			B.push_back(b[i]-'0');
		cout<<"-";
		ans=add(A,B);
	}
	else if(a[0]!='-'&&b[0]=='-')
	{
		for(int i=a.size()-1;i>=0;--i)
			A.push_back(a[i]-'0');
		for(int i=b.size()-1;i>=1;--i)
			B.push_back(b[i]-'0');
		if(cmp(A,B))
			ans=sub(A,B);
		else
			ans=sub(B,A);
	}
	else if(a[0]!='-'&&b[0]!='-')
	{
		for(int i=a.size()-1;i>=0;--i)
			A.push_back(a[i]-'0');
		for(int i=b.size()-1;i>=0;--i)
			B.push_back(b[i]-'0');
		ans=add(A,B);
	}
	for(int i=ans.size()-1;i>=0;--i)
		cout<<ans[i];
	return 0;
}