第十四届蓝桥杯C++A组 平方差

第十四届蓝桥杯C++A组 平方差

原题链接

3213 平方差

问题描述

输入两个整数$A$ 和 $B$, 输出$A^2-B^2$ 的值.

输入格式

第一行输入一个整数, 表示$A$.
第二行输入一个整数, 表示$B$.

输出格式

输出仅一行, 包含一个整数, 表示答案

样例输入

20
10

样例输出

300

评测数据规模

对于所有评测数据, $-10^{100}\le A,B\le 10^{100}$.

解题思路

用$string$或者$vector$ 模拟高精度减法和乘法.

参考代码

#include <bits/stdc++.h>
using namespace std;
// 高精度乘法
vector<int> Mul(vector<int> a, vector<int> b) {
    vector<int> res(a.size() + b.size(), 0); // 初始化答案数组, 显然乘积的长度不大于两个因数长度之和
    for (int i = 0; i < a.size(); i++) {
        for (int j = 0; j < b.size(); j++) {
            res[i + j] += a[i] * b[j]; // 先计算每一位的贡献
        }
    }
    for (int i = 0; i < res.size() - 1; i++) { // 从低位到高位, 满10进1
        res[i + 1] += res[i] / 10;
        res[i] %= 10;
    }
    while (res.size() > 1 && res.back() == 0) // 去掉前导0
        res.pop_back();
    return res;
}

vector<int> sub(vector<int> a, vector<int> b) { // 需要保证a, b都非负, 且a大于b
    vector<int> res;
    int t = 0;
    for (int i = 0; i < a.size(); i++) {
        t = a[i] - t;
        if (i < b.size())
            t -= b[i];
        if (t >= 0) {
            res.push_back(t);
            t = 0;
        } else {
            res.push_back(t + 10);
            t = 1;
        }
    }
    while (res.size() > 1 && res.back() == 0)	
        res.pop_back();
    return res;
}

int main() {
    string s1, s2, a, b;
    cin >> s1 >> s2;
	// 输入的数可能带有'-'号,我们只需要提取出绝对值,再根据绝对值大小确定答案的正负
    if (s1[0] != '-' && s2[0] != '-') { // ++	
        a = s1, b = s2;
    } else if (s1[0] != '-' && s2[0] == '-') { // +-
        a = s1;
        b.assign(s2, 1, s2.size() - 1);	// 去掉s2[0]的负号, 即从s2[1]开始, 取长度为s2.size() - 1的string, 并赋值给b
    } else if (s1[0] == '-' && s2[0] != '-') { //-+
        a.assign(s1, 1, s1.size() - 1);
        b = s2;
    } else if (s1[0] == '-' && s2[0] == '-') { //--
        a.assign(s1, 1, s1.size() - 1);
        b.assign(s2, 1, s2.size() - 1);
    }

    if (a.size() < b.size())
        cout << '-';
    else if (a.size() == b.size() && a < b) {
        cout << '-';
    }

    if (a.size() < b.size() || (a < b && a.size() == b.size())) { // 我们保证取了绝对值之后被减数一定大于减数, 如果a的绝对值小于b, 互换ab即可
        string t = a;
        a = b;
        b = t;
    }
    vector<int> A, B;
    for (int i = a.size() - 1; i >= 0; i--) // 将string转换成vector便于计算
        A.push_back(a[i] - '0');
    for (int i = b.size() - 1; i >= 0; i--)
        B.push_back(b[i] - '0');
    auto ans = sub(Mul(A, A), Mul(B, B));

    for (int i = ans.size() - 1; i >= 0; i--) // 从最高位输出答案
        cout << ans[i];

    return 0;
}