锦绣育才【集训题】阶乘之和

描述

赵老师想知道两个整数 a 和 b 各自的阶乘的和是多少。
注意:
对于 20% 的数据，1<=a,b<=10
对于 40% 的数据，1<=a,b<=20
对于 100% 的数据，1<=a,b<=500

输入描述

两个正整数a, b

输出描述

a, b 两个整数各自的阶乘的和

用例输入 1

2 6
用例输出 1

722

#include <bits/stdc++.h>
using namespace std;
std::vector<int> multiply(const std::vector<int>& num, int x) {
    std::vector<int> result;
    int carry = 0;
    for (int i = 0; i < num.size() || carry; ++i) {
        long long cur = carry;
        if (i < num.size()) cur += (long long)num[i] * x;
        result.push_back(cur % 10);
        carry = cur / 10;
    }
    return result;
}
std::vector<int> factorial(int n) {
    std::vector<int> result = {1};
    for (int i = 2; i <= n; ++i) {
        result = multiply(result, i);
    }
    return result;
}
std::vector<int> add(const std::vector<int>& num1, const std::vector<int>& num2) {
    std::vector<int> result;
    int carry = 0;
    for (int i = 0; i < num1.size() || i < num2.size() || carry; ++i) {
        int cur = carry;
        if (i < num1.size()) cur += num1[i];
        if (i < num2.size()) cur += num2[i];
        result.push_back(cur % 10);
        carry = cur / 10;
    }
    return result;
}
void print(const std::vector<int>& num) {
    for (int i = num.size() - 1; i >= 0; --i) {
        std::cout << num[i];
    }
    std::cout << std::endl;
}
int main() {
    int a, b;
    std::cin >> a >> b;
    std::vector<int> fact_a = factorial(a);
    std::vector<int> fact_b = factorial(b);
    std::vector<int> sum = add(fact_a, fact_b);
    print(sum);

    return 0;
}