高精度减法+高精度乘法计算平方差

 

高精度减法+高精度乘法

#include<iostream>
#include<vector>
using namespace std;
vector<int> multiply(vector<int> &A,vector<int> &B)
{
    int len=A.size()+B.size();
    
    vector<int> result(len,0);
    
    for(int i=A.size()-1;i>=0;i--)
    {
        for(int j=B.size()-1;j>=0;j--)
        {
            int t=A[i]*B[j];
            int p1=i+j,p2=i+j+1;
            int sum=t+result[p2];
            result[p1]+=sum/10;
            result[p2]=sum%10;
            //因为是倒序遍历，所以p1是高位
        }
    }
    
    //去除前导0
    int i=0;
    while(i<result.size()&&result[i]==0) i++;
    
    if(i==result.size()) return{0}; 
    
    return vector<int>(result.begin()+i,result.end());
}
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)
{
    vector<int> C;
    int t=0;//进位
    for(int i=0;i<A.size();i++)//cmp后，第一个参数一定是数字大的那一个
    {
        t=A[i]-t;
        if(i<B.size()) t-=B[i];
        C.push_back((t+10)%10);//这里加10是因为t可能为负数需要进位
        if(t<0) t=1;
        else t=0;
    }
    
    while(C.size()>1&&C.back()==0) C.pop_back();    
    return C;
    
}
int main()
{
    string a,b;
    cin>>a>>b;
    vector<int> A,B;
    for(char c:a) if(c!='-') A.push_back(c-'0');
    for(char c:b) if(c!='-') B.push_back(c-'0');
    
    vector<int> A1=multiply(A,A);
    vector<int> B1=multiply(B,B);
    vector<int> A2;
    vector<int> B2;
    
    for(int i=A1.size()-1;i>=0;i--) A2.push_back(A1[i]);
    for(int i=B1.size()-1;i>=0;i--) B2.push_back(B1[i]);
    
    vector<int> C;
    
    if(cmp(A2,B2)){
        C  =sub(A2,B2);
    }else{
        cout<<"-";
        C = sub(B2,A2);
    }
    // puts("");
    for(int i=C.size()-1;i>=0;i--) cout<<C[i];
    return 0;
}