宣传一下算法提高课整理 <—
题目描述
给定一个正整数 k k k,有 k k k 次询问,每次给定三个正整数 n i , e i , d i n_i,e_i,d_i ni,ei,di,求两个正整数 p i , q i p_i,q_i pi,qi,使 n i = p i × q i n_i=p_i×q_i ni=pi×qi, e i × d i = ( p i − 1 ) ( q i − 1 ) + 1 e_i×d_i=(p_i−1)(q_i−1)+1 ei×di=(pi−1)(qi−1)+1。
输入格式
第一行一个正整数 k k k,表示有 k k k 次询问。
接下来 k k k 行,第 i i i 行三个正整数 n i , d i , e i n_i,d_i,e_i ni,di,ei。
输出格式
输出 k k k 行,每行两个正整数 p i , q i p_i,q_i pi,qi 表示答案。
为使输出统一,你应当保证 p i ≤ q i p_i≤q_i pi≤qi。
如果无解,请输出 NO
。
数据范围
以下记 m = n − e × d + 2 m=n−e×d+2 m=n−e×d+2。
保证对于
100
100
100% 的数据,
1
≤
k
≤
1
0
5
1≤k≤10^5
1≤k≤105,对于任意的
1
≤
i
≤
k
1≤i≤k
1≤i≤k,
1
≤
n
i
≤
1
0
18
1≤n_i≤10^{18}
1≤ni≤1018,
1
≤
e
i
×
d
i
≤
1
0
18
1≤e_i×d_i≤10^{18}
1≤ei×di≤1018,
1
≤
m
≤
1
0
9
1≤m≤10^9
1≤m≤109。
样例输入
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
≤
k
≤
1
0
5
1≤k≤10^5
1≤k≤105,
1
≤
n
i
≤
1
0
18
1≤n_i≤10^{18}
1≤ni≤1018,
1
≤
e
i
×
d
i
≤
1
0
18
1≤e_i×d_i≤10^{18}
1≤ei×di≤1018,
1
≤
m
≤
1
0
9
1≤m≤10^9
1≤m≤109
那么1e18的数据一定是要开long long的。
仔细读题可以发现,数据范围处给了一个
令
m
=
n
−
e
×
d
+
2
令m=n-e×d+2
令m=n−e×d+2,我们思考:这是否会是解题的突破口呢?作者在比赛的时候没有看到m,这道题用暴力做的,只拿了20分
对于此等式: e i × d i = ( p i − 1 ) ( q i − 1 ) + 1 e_i×d_i=(p_i−1)(q_i−1)+1 ei×di=(pi−1)(qi−1)+1
化简后: e i × d i = p q − ( p + q ) + 2 e_i×d_i=pq-(p+q)+2 ei×di=pq−(p+q)+2
∵ n = p × q ∵n=p×q ∵n=p×q
∴ e i × d i = n − ( p + q ) + 2 ∴e_i×d_i=n-(p+q)+2 ∴ei×di=n−(p+q)+2
移项,得 p + q = n − e i × d i + 2 p+q=n-e_i×d_i+2 p+q=n−ei×di+2
即 m = p + q m=p+q m=p+q
联立,得
{
n
=
p
×
q
m
=
p
+
q
\left\{\begin{matrix} n=p×q \\ m=p+q \end{matrix}\right.
{n=p×qm=p+q
(图片来自@JcWing)
思路1时间复杂度
共k组数据,每组数据二分范围1~ n / 2 n/2 n/2,因此时间复杂度为 O ( k log m ) O(k\log m) O(klogm)
可以AC
思路1-AC Code
C + + C++ 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:逆造方程
{ n = p × q m = p + q \left\{\begin{matrix} n=p×q \\ m=p+q \end{matrix}\right. {n=p×qm=p+q
运用韦达逆定理逆造方程,则
p 2 + m p + n = 0 p^2+mp+n=0 p2+mp+n=0
运用求根公式 − b ± b 2 − 4 a c 2 a \frac{-b\pm \sqrt{b^2-4ac} }{2a} 2a−b±b2−4ac
得到 △ = m 2 − 4 n △=m^2-4n △=m2−4n
则 p , q p,q p,q分别为 m − △ m-\sqrt{\triangle } m−△与 m + △ m+\sqrt{\triangle } m+△
思路2时间复杂度
k k k次循环,因此是 O ( k ) O(k) O(k)
可真挺ok的
思路2-AC Code
C + + C++ 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;
}
最后,如果觉得对您有帮助的话,点个赞再走吧!