C++实现Diffie-Hellman密钥交换算法(不需要大数支持)输入 第一行输入素数p 第二行输入本原根g 第三行输入A的私钥a 第四行输入B的私钥b 输出 第一行输出A的计算数 生成密钥 第二行输出B的计算数 生成密钥 代码详细点

下面是C++实现Diffie-Hellman密钥交换算法的代码：

#include <iostream>
#include <cmath>

using namespace std;

int mod_pow(int a, int b, int mod) {
    int res = 1 % mod;
    while (b > 0) {
        if (b & 1) {
            res = (long long)res * a % mod;
        }
        a = (long long)a * a % mod;
        b >>= 1;
    }
    return res;
}

int main() {
    int p, g, a, b;
    cout << "输入素数p：";
    cin >> p;
    cout << "输入本原根g：";
    cin >> g;
    cout << "输入A的私钥a：";
    cin >> a;
    cout << "输入B的私钥b：";
    cin >> b;

    // 计算 A 和 B 的公钥
    int A = mod_pow(g, a, p);
    int B = mod_pow(g, b, p);

    cout << "A的计算数：" << A << endl;
    cout << "B的计算数：" << B << endl;

    // 计算共享密钥
    int K1 = mod_pow(B, a, p);
    int K2 = mod_pow(A, b, p);

    if (K1 == K2) {
        cout << "共享密钥为：" << K1 << endl;
    } else {
        cout << "密钥计算错误！" << endl;
    }

    return 0;
}

代码实现中使用了一个快速幂算法`mod_pow`，用来计算指数幂模的结果。在主函数中，先输入素数p、本原根g、A和B的私钥，然后计算出A和B的公钥，并输出计算数。最后，计算出共享密钥，如果两个计算结果相等，则输出共享密钥，否则输出错误信息。