高精度减法

题目描述

邪狼掌握了简单高精度加法程序的编写后，修罗王要求邪狼写一个简单高精度减法的程序。即求出A-B的值，其中A和B均为非负整数，其位数不超过5000。

输入

共两行数据，第一行为一个非负整数A，第二行为一个非负整数B，A、B的位数均在5000以内。

输出

输出一个整数，即A-B的值。

样例输入 Copy

2
1

样例输出 Copy

1

代码

#include <bits/stdc++.h>
using namespace std;
bool cmp(vector<int> A,vector<int> B)           //判断A,B大小
{
    if(A.size()!=B.size()) return A.size()>B.size();
    for(int i=A.size()-1;i>=0;i--)
        if(A[i]!=B[i]) return A[i]>B[i];
    return true;
}
vector<int> sub(vector<int> A,vector<int> B)     //大减小
{
    vector<int> C;
    for(int i=0,t=0;i<A.size();i++)
    {
        t=A[i]-t;
        if(i<B.size()) t-=B[i];
        C.push_back((t+10)%10);
        if(t>=0) t=0;
            else t=1;
    }
    while(C.size()>1&&!C.back()) C.pop_back();     //去除前导0
    return C;
}
int main()
{
    string a,b;
    cin>>a>>b;
    vector<int> A,B;
    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');
    if(cmp(A,B)){
            auto C=sub(A,B);
            for(int i=C.size()-1;i>=0;i--) cout<<C[i];
    }
    else {
        auto C=sub(B,A);
        cout<<"-";
        for(int i=C.size()-1;i>=0;i--) cout<<C[i];
    }
    return 0;
}