QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#770246#8336. Indeterminate Equationxianjing#AC ✓1ms4088kbC++233.6kb2024-11-21 21:14:462024-11-21 21:14:48

Judging History

This is the latest submission verdict.

  • [2024-11-21 21:14:48]
  • Judged
  • Verdict: AC
  • Time: 1ms
  • Memory: 4088kb
  • [2024-11-21 21:14:46]
  • Submitted

answer

#pragma GCC optimize(1, 2, 3, "Ofast", "inline")
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
using i64 = long long;
using i128 = __int128;
i64 mul(i64 a, i64 b, i64 m) {
	return static_cast<__int128> (a) * b % m;
}
i64 power(i64 a, i64 b, i64 m) {
	i64 res = 1 % m;
	for (; b; b >>= 1, a = mul(a, a, m))
		if (b & 1) res = mul(res, a, m);
	return res;
}
bool isprime(i64 n) {
	if (n < 2) 
		return false;
	static constexpr int A[] = {2, 3, 5, 7, 11, 13, 17, 19, 23};
	int s = __builtin_ctzll(n - 1);
	i64 d = (n - 1) >> s;
	for (auto a : A) {
		if (a == n)
			return true;
		i64 x = power(a, d, n);
		if (x == 1 || x == n - 1)
			continue;
		bool ok = false;
		for (int i = 0; i < s - 1; ++i) {
			x = mul(x, x, n);
			if (x == n - 1) {
				ok = true;
				break;
			}
		}
		if (!ok)
			return false;
	}
	return true;
}

vector<i64> factorize(i64 n) {
	vector<i64> p;
	function<void(i64)> f = [&] (i64 n) {
		if (n <= 10000) {
			for (int i = 2; i * i <= n; ++i)
				for (; n % i == 0; n /= i)
					p.push_back(i);
			if (n > 1)
				p.push_back(n);
			return ;
		}
		if (isprime(n)) {
			p.push_back(n);
			return;
		}
		auto g = [&] (i64 x) {
			return (mul(x, x, n) + 1) % n;
		};
		i64 x0 = 2;
		while (true) {
			i64 x = x0;
			i64 y = x0;
			i64 d = 1;
			i64 power = 1, lam = 0;
			i64 v = 1;
			while (d == 1) {
				y = g(y);
				++lam;
				v = mul(v, std::abs(x - y), n);
				if (lam % 127 == 0) {
					d = std::__gcd(v, n);
					v = 1;
				}
				if (power == lam) {
					x = y;
					power *= 2;
					lam = 0;
					d = std::__gcd(v, n);
					v = 1;
				}
			}
			if (d != n) {
				f(d);
				f(n / d);
				return ;
			}
			++x0;
		}
	};
	f(n);
	sort(p.begin(), p.end());
	return p;
}
void solve(){
	i64 n, k;
	cin >> n >> k;
	
	auto qpow = [&] (i128 a, i128 b) -> i128 {
		i128 res = 1;
		while (b) {
			if (b & 1) res = res * a;
			a = a * a;
			b >>= 1;
		}
		return res;
	};
	
	if (k >= 6) {
		int ans = 0;
		int MAXN = max(pow(n / k + 1, 1.0 / (k - 1)) + 1, 2.1);
		for (i64 b = 1; b <= MAXN; b++) {
			i128 a = pow(qpow(b, k) + n, 1.0 / k);
			if (qpow(a, k) - qpow(b, k) == n) {
				ans++;
			}
		};
		cout << ans << '\n';
		
		return ;
	}
	
	vector<i64> pf = factorize(n), fact;
	
	auto dfs = [&] (auto self, i64 d, int i) -> void {
		if (i == pf.size()) {
			fact.push_back(d);
			return ;
		}
		int j = i;
		while (j < pf.size() && pf[j] == pf[i]) j++;
		for (int cnt = 0; cnt <= j - i; cnt++) {
			self(self, d, j);
			d *= pf[i];
		}
	};
	
	dfs(dfs, 1, 0);
	
//	for (auto x : fact) {
//		cout << x << ' ' ;
//	}
//	cout << '\n' << fact.size() << '\n';
	
	// (b + d) ^ k - b ^ k = n
	// k * b ^ (k - 1) * d > n
	// b = pow(n / (k * d), 1 / (k - 1)))
	
	int ans = 0;
	
	for (i64 d : fact) {
		if (k == 3) {
			i128 delta = (i128)12 * n / d - 3 * (i128)d * d;
			i64 b = (-3 * d + sqrtl(delta)) / 6, a = b + d;
			if (b > 0 && a * a + a * b + b * b == n / d) ans++;
			continue;
		}
		
		i64 l = 2, r = max(pow(n / (i128)(k * d) + 1, 1.0 / (k - 1)) + 1, 2.1);
		
		auto check = [&] (i64 mid) -> int {
			i128 a = 1, b = 1;
			int cnt = 0;
			while (a - b <= n && cnt <= k + 2) {
				a = a * (mid + d);
				b = b * mid;
				cnt++;
			}
			return cnt - 1;
		};
		
		while (l != r) {
			i64 mid = l + r >> 1;
			if (k <= check(mid)) l = mid + 1;
			else r = mid;
		}
		l--;
		if (qpow(l + d, k) - qpow(l, k) == n) {
			ans++;
		}
	}
	cout << ans << '\n';
}

signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	int t = 1;
	cin >> t;
	while(t --){
		solve();
	}
	return 0;
}


这程序好像有点Bug,我给组数据试试?

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 4032kb

input:

3
7 3
15 4
31 5

output:

1
1
1

result:

ok 3 number(s): "1 1 1"

Test #2:

score: 0
Accepted
time: 1ms
memory: 3880kb

input:

20
409013310800583799 32
70368744177663 46
592570256192463681 4
360020145419649 5
357385021818058297 3
950227088646484702 56
127 7
718303642731669822 3
651621023623339377 45
405657994164855469 3
4095 12
288230376151711743 58
224251587219143167 5
2221626677255791 3
2953488086475199 4
6672460861685157...

output:

0
1
1
1
1
0
1
0
0
1
1
1
1
1
1
1
0
1
0
0

result:

ok 20 numbers

Test #3:

score: 0
Accepted
time: 1ms
memory: 4036kb

input:

20
299663213308337404 3
789663530028185253 15
899102550518872677 5
908651241968426417 38
10161731698203367 3
172563904510845597 3
24904219972305241 3
72057594037927935 56
900135928346130972 6
63 6
32370674179164127 3
638119226609365944 62
67108863 26
257821881508336320 3
815804918211865461 12
910411...

output:

1
0
1
0
1
1
1
1
0
1
1
0
1
1
0
0
1
1
1
1

result:

ok 20 numbers

Test #4:

score: 0
Accepted
time: 1ms
memory: 3792kb

input:

20
987212517904356866 10
625343233082041 3
281474976710655 48
35184372088831 45
8191 13
33554431 25
387412204026720769 3
109355953976006437 3
2198205420145 4
47103092714538256 21
448246606222334791 7
99826963551194419 3
417720736167446718 26
9007199254740991 53
130481423309676535 8
2251799813685247 ...

output:

0
1
1
1
1
1
1
1
1
0
0
1
0
1
0
1
1
1
1
1

result:

ok 20 numbers

Test #5:

score: 0
Accepted
time: 1ms
memory: 3900kb

input:

20
369621295427910453 63
829540864429248940 50
328188506235593452 38
801208558894131272 13
872695838727528838 3
1073741823 30
252424883748225843 19
276360191653207583 13
23680201591411597 3
605336327501498607 29
3368576601423011 5
541099690642302715 3
110139179543268157 3
1073741823 30
5091219638745...

output:

0
0
0
0
1
1
0
0
1
0
1
1
1
1
0
1
1
1
1
1

result:

ok 20 numbers

Test #6:

score: 0
Accepted
time: 1ms
memory: 3784kb

input:

20
8191 13
58185299844488917 3
946219666221344402 19
199313544345127719 42
305224472569938018 6
274877906943 38
8191 13
4398046511103 42
181275224823960757 3
26525260926988201 3
562949953421311 49
4464847706965471 3
457186927518156860 5
158117286447030969 46
549755813887 39
555307613261368 5
4401772...

output:

1
1
0
0
0
1
1
1
1
1
1
1
0
0
1
1
1
0
0
0

result:

ok 20 numbers

Test #7:

score: 0
Accepted
time: 0ms
memory: 3928kb

input:

20
4398046511103 42
833253453410579766 62
409534938012033754 63
72057594037927935 56
536870911 29
34951725759140191 3
90835460846790978 13
16777215 24
2251799813685247 51
699719950959624759 54
248653563296585041 3
421715905914046949 3
605636896545987345 4
524287 19
241759398856733802 6
4935421722667...

output:

1
0
0
1
1
1
0
1
1
0
1
1
1
1
0
0
1
1
1
0

result:

ok 20 numbers

Test #8:

score: 0
Accepted
time: 0ms
memory: 4080kb

input:

20
106552237922246881 3
412052866942149907 3
2147483647 31
796082436742414414 24
408359486510009476 12
63 6
17592186044415 44
118509271529004788 9
3979116231690817 3
262143 18
70368744177663 46
42768140446560097 3
8589934591 33
386734487134757674 12
4503599627370495 52
536870911 29
28823037615171174...

output:

1
1
1
0
0
1
1
0
1
1
1
1
1
0
1
1
1
1
0
1

result:

ok 20 numbers

Test #9:

score: 0
Accepted
time: 1ms
memory: 3840kb

input:

20
134217727 27
83420139097331842 62
274135990917964800 5
434500257687581785 35
501744216766405008 15
451335986349193 5
868853355445748320 13
301001412129614 5
67108863 26
953222376692677434 25
1886006831138137 3
15857519990553792 5
69764165614656105 63
81798613090978231 3
989051271211163304 41
1867...

output:

1
0
1
0
0
1
0
1
1
0
1
1
0
1
0
0
1
1
1
1

result:

ok 20 numbers

Test #10:

score: 0
Accepted
time: 1ms
memory: 3936kb

input:

20
290234946576194630 24
324991611316950487 3
377254568067887797 3
199176477432292800 5
481817525933591580 44
306179318310704225 5
184460035325889819 5
96166909590250592 3
69908027519627596 3
7371051187969 3
562949953421311 49
513570208984730028 6
180646903364861467 3
622083445561687933 56
145588025...

output:

0
1
1
1
0
1
1
1
1
1
1
0
1
0
1
1
1
1
0
1

result:

ok 20 numbers

Test #11:

score: 0
Accepted
time: 1ms
memory: 3936kb

input:

20
70368744177663 46
137438953471 37
72057594037927935 56
1898262015327215 4
593117722299956521 62
81438002329886371 3
399674414175214669 3
215867547011594871 32
584844110936591318 53
159843794446621357 3
54115722785729521 3
278593973273873791 3
741200008753001294 5
62103974292968851 3
7521806114753...

output:

1
1
1
1
0
1
1
0
0
1
1
1
1
1
0
1
1
1
1
1

result:

ok 20 numbers

Test #12:

score: 0
Accepted
time: 1ms
memory: 3876kb

input:

20
15 4
639325486456928631 24
367160239938275019 3
35135369151859768 5
134217727 27
162403073577381280 4
4398046511103 42
849266064473374657 22
467964515935413061 3
8388607 23
137438953471 37
215086822275193505 3
305420275067478247 3
103789388695665446 18
215568298601022841 3
4503599627370495 52
469...

output:

1
0
1
1
1
1
1
0
1
1
1
1
1
0
1
1
1
1
0
1

result:

ok 20 numbers

Test #13:

score: 0
Accepted
time: 0ms
memory: 3904kb

input:

20
137438953471 37
4503599627370495 52
360622068085547772 10
210287245019037702 35
74711655323394929 56
98155419968374287 21
102533060809494095 4
140737488355327 47
933243045575014403 11
72057594037927935 56
149720681727843610 57
1073741823 30
591686942125381888 31
119086672276608015 4
3477271960851...

output:

1
1
0
0
0
0
1
1
0
1
0
1
0
1
1
1
1
0
1
1

result:

ok 20 numbers

Test #14:

score: 0
Accepted
time: 1ms
memory: 3880kb

input:

20
819477969526523940 9
997198472969039861 35
233055230837054597 53
746904402177708777 27
67108863 26
4194303 22
390095551418097427 3
927752109611997442 48
32767 15
390318061363802287 3
686611024540234609 12
20402267600069161 3
698884614026798317 52
562949953421311 49
1550940349430625 4
511 9
759139...

output:

0
0
0
0
1
1
1
0
1
1
0
1
0
1
1
1
0
1
1
0

result:

ok 20 numbers

Test #15:

score: 0
Accepted
time: 1ms
memory: 3900kb

input:

20
347092101157932621 55
523357334205590686 31
686906733435643515 60
120007640497588865 3
347422310624045419 3
301665674749200847 3
866910816105601352 30
5127183360 4
68719476735 36
4194303 22
483964008794649279 18
234063663878120671 3
123845819701080601 3
285423692080282687 3
3912628851220100 5
310...

output:

0
0
0
1
1
1
0
1
1
1
0
1
1
1
1
1
1
0
0
0

result:

ok 20 numbers

Test #16:

score: 0
Accepted
time: 1ms
memory: 4088kb

input:

20
33554431 25
528392007319786400 31
16482551564781760 4
838990357775672509 32
942785412798517385 38
439170874733624419 3
29596780167767827 3
83285573011773301 3
950527866918480 4
2251799813685247 51
4095 12
273991306079656475 5
616699112264653214 52
51492842248899972 25
18785737008450461 15
9959774...

output:

1
0
1
0
0
1
1
1
1
1
1
1
0
0
0
0
0
1
1
1

result:

ok 20 numbers

Test #17:

score: 0
Accepted
time: 1ms
memory: 3808kb

input:

20
2097151 21
3400566422153238 3
631760588083252674 28
167808767206605768 3
323394699435360469 3
335695376290788387 17
35345259238590038 23
39661511301666151 3
102003212137599169 3
131071 17
6017133684009169 3
810416549631456806 61
78371917503410782 3
255 8
292354900238725376 47
394542618557932158 1...

output:

1
1
0
1
1
0
0
1
1
1
1
0
0
1
0
0
1
1
0
1

result:

ok 20 numbers

Test #18:

score: 0
Accepted
time: 1ms
memory: 3936kb

input:

20
8191 13
34450976592326175 39
115992634653776988 30
30841490245990321 3
730558013031329906 56
69661494843438160 4
193199298983587 3
745954000719453413 3
14062568319648604 3
310197905349117863 26
323849701457243377 12
127287856421571104 3
2857372929472969 3
348006386420030269 3
16383 14
524287 19
1...

output:

1
0
0
1
0
1
1
1
1
0
0
1
1
1
1
1
1
1
1
1

result:

ok 20 numbers

Test #19:

score: 0
Accepted
time: 1ms
memory: 3872kb

input:

20
328331949339594705 4
440018735280441037 3
651726196698130641 25
35184372088831 45
560815166864882104 3
1665749740436467 3
725187027841738697 10
97742833161599557 3
628708937360761 3
144115188075855871 57
45246298445492401 3
897305684378354416 4
435615967025653155 62
51390152812182661 3
5497558138...

output:

1
1
0
1
1
1
0
1
1
1
1
1
0
1
1
1
1
1
1
1

result:

ok 20 numbers

Test #20:

score: 0
Accepted
time: 1ms
memory: 3816kb

input:

20
10287757755011200 5
75165632304477787 3
208610838312868531 3
17592186044415 44
262143 18
472628059236086465 33
274441644154738406 14
63 6
70368744177663 46
9120350170569787 3
68931120362029927 3
244281394805637637 3
616071809801248136 44
36284940549244431 3
518603065311042205 31
4294967295 32
650...

output:

1
1
1
1
1
0
0
1
1
1
1
1
0
1
0
1
0
0
0
0

result:

ok 20 numbers

Test #21:

score: 0
Accepted
time: 1ms
memory: 3804kb

input:

20
951043950767860780 64
72057594037927935 56
4398046511103 42
268435455 28
984744214330259561 54
536393574249559831 3
134217727 27
779206543455963934 53
441797261073316237 3
10900408805331694 5
16383 14
755235903449005803 3
235576685499319964 3
4398046511103 42
16992310045669219 3
536870911 29
3180...

output:

0
1
1
1
0
1
1
0
1
1
1
1
1
1
1
1
1
0
0
1

result:

ok 20 numbers

Extra Test:

score: 0
Extra Test Passed