[CSP-J 2022] 解密 题解

题目描述

给定一个正整数 $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$。

思路

这道题可以看成一道数学题（还有其他解法，不过我不会 ）。
看到题目先化简 ed：
$$ed=(p-1)(q-1)+1$$
化简得：
$$ed=pq-p-q+2$$
将 n = pq 带入原式得：
$$ed=n-p-q+2$$
整理原式得
$$p+q=n-ed+2$$

现在已知
$$\{\begin{array}{l}p+q=n-ed+2\\ pq=n\end{array}$$
​
要求 p、q。

其中 n,e,d 为常数。

如果我们能推出 p - q，那么问题就迎刃而解了。
我们运用一下完全平方公式（初二会学，没学的，建议学一下再看下面的内容）

$$\begin{array}{rl}(p+q)²-(p-q)²& =(p²+2pq+q²)-(p²-2pq+q²)\\ & =p²+2pq+q²-p²+2pq-q²\\ & =4pq\end{array}$$
$$\begin{array}{rl}(p+q)²-(p-q)²& =4pq\end{array}$$
$$\begin{array}{rl}(p+q)²-4pq& =(p-q)²\end{array}$$
交换左右两项可得：
$$\begin{array}{rl}(p-q)& =(p+q)²-4pq\end{array}$$
左边脱掉平方得：
$$p-q=\pm \sqrt{(p+q)²-4pq}$$
​将 n = pq，n - e *d + 2 = p + q=p+q 带入得：
$$\{\begin{array}{l}p+q=n-ed+2\\ p-q=\sqrt{(n-ed+2)²-4n}\end{array}$$
然后我们将上下两个式子加起来，就得到了
$$\begin{array}{r}2p=n-ed+2+\sqrt{(n-ed+2)²-4n}\end{array}$$
再同时除以2：
$$\begin{array}{r}p=\frac{n-ed+2+\sqrt{(n-ed+2)²-4n}}{2}\end{array}$$
这时，我们得出了p，也知道n−e×d+2 为常数，所以可以解出 q 为：
$$\begin{array}{r}q=\frac{n-ed+2-\sqrt{(n-ed+2)²-4n}}{2}\end{array}$$
就这样，我们可以得到：
$$\{\begin{array}{l}p=\frac{n-ed+2+\sqrt{(n-ed+2)²-4n}}{2}\\ q=\frac{n-ed+2-\sqrt{(n-ed+2)²-4n}}{2}\end{array}$$
再看到题目要求，p、q 要求是正整数。因为我们知道，如果平方根不是正数，只可能是 0、正有理数、无理数（平方开不尽）、不在实数范围内的虚数（被开方数小于 0）。

0就在在算出p和q之后判断一下就行。对于小数和无理数，我们可以先用 long long 强制转换，舍弃小数点后带入题目给出的式子，检验答案是否正确。对于虚数，函数 sqrt 的被开方数为负数的时候会返回 NaN，用 long long 储存 NaN 时值为 0，和我们处理 0 时思路一样。如果答案符合上述条件，那么一定是整数，也就是符合题目要求的答案。

那时间复杂度如何保证呢？加减乘除都是 O(1) 的，sqrt 也是 O(1) 级别的，所以单次询问时间复杂度为 O(1)，总询问复杂度为 O(k)，最大运算量为 10^5，所以不会超时。

思路就到这里了，下面就是代码啦！

代码

#include<bits/stdc++.h>
using namespace std;
long long n,d,e,t;
int main(){
	//freopen("ans.in","r",stdin);
	//freopen("ans.out","w",stdout);
	cin>>t;
	while(t--){
		cin>>n>>d>>e;
		long long delta=(n-e*d+2)*(n-e*d+2)-4*n;
		if(delta<0){//特判 
			cout<<"NO"<<endl;
			continue;
		}
		long long p=1.0*((n-e*d+2)-sqrt(delta))/2;//p要小于q 
		long long q=1.0*((n-e*d+2)+sqrt(delta))/2;
		if(p>0 && q>0 && p*q==n && e*d==(p-1)*(q-1)+1){//判断p和q合不合理 
			cout<<p<<" "<<q<<endl;
		}
		else {
			cout<<"NO"<<endl;
		}
	}
	return 0;
}

完结撒花✿✿ヽ(°▽°)ノ✿