AcWing寒假每日一题2023-1.20解密

宣传一下算法提高课整理 <—

题目描述

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

数据范围

以下记 $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$。

样例输入

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

样例输出

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

---

思路1：二分

先看数据范围：$1\le k\le 10^5$，$1\le n_i\le 10^{18}$，$1\le e_i\times d_i\le 10^{18}$，$1\le m\le 10^9$
那么1e18的数据一定是要开long long的。
仔细读题可以发现，数据范围处给了一个$令m=n-e\times d+2$，我们思考：这是否会是解题的突破口呢？作者在比赛的时候没有看到m，这道题用暴力做的，只拿了20分

对于此等式： $e_i\times d_i=(p_i-1)(q_i-1)+1$

化简后： $e_i\times d_i=pq-(p+q)+2$

$\because n=p\times q$

$\therefore e_i\times d_i=n-(p+q)+2$

移项，得$p+q=n-e_i\times d_i+2$

即$m=p+q$

联立，得
$\{\begin{array}{c}n=p\times q\\ m=p+q\end{array}$


（图片来自@JcWing）

思路1时间复杂度

共k组数据，每组数据二分范围1~$n/2$,因此时间复杂度为$O(k\log m)$

可以AC

思路1-AC Code

$C++$

#include <iostream>
#include <cstring>

using namespace std;

typedef long long LL;

LL k, n, m, e, d;
LL p, q;

int main()
{
	scanf("%lld", &k);
	
	while (k -- )
	{
		scanf("%lld%lld%lld", &n, &d, &e);
		m = n - e * d + 2;
		
		LL l = 1, r = m >> 1;
		while (l < r)
		{
			LL mid = l + r >> 1;
			if (mid * (m - mid) >= n) r = mid;
			else l = mid + 1;
		}
		
		p = l, q = m - l;
		if (p * q == n) printf("%lld %lld\n", p, q);
		else puts("NO");
	}
	
	return 0;
}

思路2：逆造方程

$\{\begin{array}{c}n=p\times q\\ m=p+q\end{array}$

运用韦达逆定理逆造方程，则

$p^2+mp+n=0$

运用求根公式$\frac{-b\pm \sqrt{b^2-4ac}}{2a}$

得到$△=m^2-4n$

则$p,q$分别为$m-\sqrt{△}$与$m+\sqrt{△}$

思路2时间复杂度

$k$次循环，因此是$O(k)$

可真挺ok的

思路2-AC Code

$C++$

#include <iostream>
#include <cmath>

using namespace std;

typedef long long LL;

int main()
{
    int k;
    scanf("%d", &k);

    while (k -- )
    {
        LL n, d, e;
        scanf("%lld%lld%lld", &n, &d, &e);
        LL m = n - e * d + 2;
        LL dt = m * m - 4 * n;
        LL r = sqrt(dt);

        if (dt < 0 || r * r != dt) puts("NO");
        else printf("%lld %lld\n", (m - r) / 2, (m + r) / 2);
    }

    return 0;
}

最后，如果觉得对您有帮助的话，点个赞再走吧！