【洛谷】一道类似基于大整数分解的公钥密码的算法题

P8814 [CSP-J 2022] 解密

题源：P8814 [CSP-J 2022] 解密

发现一个类RSA数学原理的因子分解问题变体，以此留作记录。

题目描述

给定一个正整数 $k$，有 $k$ 次询问，每次给定三个正整数 $n_i,e_i,d_i$，求两个正整数 $p_i,q_i$，使 $n_i=p_i\times q_i$、$e_i\times d_i=(p_i-1)(q_i-1)+1$。

输入格式

第一行一个正整数 $k$，表示有 $k$ 次询问。
接下来 $k$ 行，第 $i$ 行三个正整数 $n_i,d_i,e_i$。

输出格式

输出 $k$ 行，每行两个正整数 $p_i,q_i$ 表示答案。
为使输出统一，你应当保证 $p_i\le q_i$。
如果无解，请输出 NO。

样例 #1

样例输入 #1

10
770 77 5
633 1 211
545 1 499
683 3 227
858 3 257
723 37 13
572 26 11
867 17 17
829 3 263
528 4 109

样例输出 #1

2 385
NO
NO
NO
11 78
3 241
2 286
NO
NO
6 88

提示

【样例 #2】

见附件中的 decode/decode2.in 与 decode/decode2.ans。

【样例 #3】

见附件中的 decode/decode3.in 与 decode/decode3.ans。

【样例 #4】

见附件中的 decode/decode4.in 与 decode/decode4.ans。

【数据范围】

以下记 $m=n-e\times d+2$。

保证对于 $100\%$ 的数据，$1\le k\le {10}^5$，对于任意的 $1\le i\le k$，$1\le n_i\le {10}^{18}$，$1\le e_i\times d_i\le {10}^{18}$
，$1\le m\le {10}^9$。

测试点编号	$k\le$	$n\le$	$m\le$	特殊性质
$1$	$10^3$	$10^3$	$10^3$	保证有解
$2$	$10^3$	$10^3$	$10^3$	无
$3$	$10^3$	$10^9$	$6\times 10^4$	保证有解
$4$	$10^3$	$10^9$	$6\times 10^4$	无
$5$	$10^3$	$10^9$	$10^9$	保证有解
$6$	$10^3$	$10^9$	$10^9$	无
$7$	$10^5$	$10^{18}$	$10^9$	保证若有解则 $p=q$
$8$	$10^5$	$10^{18}$	$10^9$	保证有解
$9$	$10^5$	$10^{18}$	$10^9$	无
$10$	$10^5$	$10^{18}$	$10^9$	无

思路

最开始想到的依旧是最为朴素的一种 —— 半区间扫描出较小因子后做除法得出较大因子，最后再检查是否符合 $n_i=p_i\times q_i$、$e_i\times d_i=(p_i-1)(q_i-1)+1$ 这个条件。然而受制于数据本身的庞大规模，始终采取这样的扫描式手段对于计算的效率必是大打折扣，于是乎就会出现…👇

甚至是…👇

因此再看题面，两个条件式可以采取进一步的联立求解，此处也需要配合完全平方式的复合运算，由此可解得：
$$\{\begin{array}{l}p=\frac{n-ed+2-\sqrt{(n-ed-2{)}^2-4n}}{2}\\ q=\frac{n-ed+2+\sqrt{(n-ed-2{)}^2-4n}}{2}\end{array}$$

代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#define OK 1
#define NO 0
typedef long long LL;

// const int INF = 2e9;
// const double pi = 3.14;


int main() {
    int k; scanf("%d", &k);
    while (k--) {
        LL n, e, d; scanf("%lld %lld %lld", &n, &e, &d);
        // p - q 表达式
        LL psubq = sqrt((n - e * d + 2) * (n - e * d + 2) - 4 * n);
        // p + q 表达式
        LL paddq = n - e * d + 2;
        // 配凑
        LL p = (psubq + paddq) / 2;
        // 互补做差
        LL q = paddq - p;
        if (p * q == n && e * d == (p - 1) * (q - 1) + 1 && p && q) {
            printf("%lld %lld\n", (LL)fmin(p, q), (LL)fmax(p, q));
        } else {
            printf("NO\n");
        }
    }
    return 0;
}

小结

这一种问题为公钥密码RSA的简化版本，并没有严格要求因子 $p,q$ 为素数，即便如此，其中蕴含的求根公式构造的方法同样是可以运用的，因此个人感到值得记录下来。
至此已已，再会。

每一个不曾起舞的日子，都是对生命的辜负。