高精度減法模板

題目描述  （原題鏈接）
給定兩個正整數，計算它們的差。

輸入格式
共兩行，每行包含一個整數。

輸出格式
共一行，包含所求的差。

數據範圍
1≤整數長度≤

樣例

輸入樣例：
32
11
輸出樣例：
21

(大整數減法) O(n)

#include<bits/stdc++.h>
using namespace std;

//判斷A>=B
bool cmp(vector<int> a,vector<int> 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){
    int t=0;           //t是借位
    vector<int> res;
    for(int i=0;i<a.size();i++){
        t=a[i]-t;
        if(i<b.size()) t-=b[i];
        res.push_back((t+10)%10);
        if(t<0) t=1;
        else t=0;
    }
    while(res.size()>1&&res.back()==0) res.pop_back();
    return res;
}

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];
    }
}