1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
| #include<iostream> #include<vector>
using namespace std;
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;
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 = 1; else t = 0; }
while(C.size() > 1 && C.back() == 0) C.pop_back(); return C; }
vector<int> mul(vector<int> &A, vector<int> &B) { vector<int> C(A.size() + B.size());
for(int i = 0; i < A.size(); i++) for(int j = 0; j < B.size(); j++) C[i + j] += A[i] * B[j];
for(int i = 0, t = 0; i < C.size(); i++) { t += C[i]; C[i] = t % 10; t /= 10; }
while(C.size() > 1 && C.back() == 0) C.pop_back();
return C; }
int main() { vector<int> A, B, C; string aa, bb, a, b;
cin>>aa>>bb;
if(aa[0] == '-') for(int i = 1; i < aa.size(); i++) a += aa[i]; else a = aa; if(bb[0] == '-') for(int i = 1; i < bb.size(); i++) b += bb[i]; else b = bb;
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');
A = mul(A, A); B = mul(B, B);
if(cmp(A, B)) C = sub(A, B); else { cout<<'-'; C = sub(B, A); }
for(int i = C.size() - 1; i >= 0; i--) { cout<<C[i]; } cout<<endl; return 0; }
|