QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#732648#8336. Indeterminate Equationwarner1129#AC ✓2ms3828kbC++203.0kb2024-11-10 15:25:372024-11-10 15:25:38

Judging History

This is the latest submission verdict.

  • [2024-11-10 15:25:38]
  • Judged
  • Verdict: AC
  • Time: 2ms
  • Memory: 3828kb
  • [2024-11-10 15:25:37]
  • Submitted

answer

#include <bits/stdc++.h>

using namespace std;

#define ff first
#define ss second
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()

using i64 = long long;
using u64 = unsigned long long;
using i128 = __int128;

u64 mul(u64 a, u64 b, u64 M) {
    i64 r = a * b - M * u64(1.L / M * a * b);
    return r + M * ((r < 0) - (r >= (i64)M));
}
u64 power(u64 a, u64 b, u64 M) {
    u64 r = 1;
    for (; b; b /= 2, a = mul(a, a, M))
        if (b & 1) r = mul(r, a, M);
    return r;
}
bool isPrime(u64 n) {
    if (n < 2 or n % 6 % 4 != 1) return (n | 1) == 3;
    auto magic = {2, 325, 9375, 28178, 450775, 9780504, 1795265022};
    u64 s = __builtin_ctzll(n - 1), d = n >> s;
    for (u64 x : magic) {
        u64 p = power(x % n, d, n), i = s;
        while (p != 1 and p != n - 1 and x % n and i--) {
            p = mul(p, p, n);
        }
        if (p != n - 1 and i != s) return 0;
    }
    return 1;
}
u64 pollard(u64 n) {
    u64 c = 1;
    auto f = [&](u64 x) { return mul(x, x, n) + c; };
    u64 x = 0, y = 0, p = 2, q, t = 0;
    while (t++ % 128 or gcd(p, n) == 1) {
        if (x == y) c++, y = f(x = 2);
        if (q = mul(p, x > y ? x - y : y - x, n)) p = q;
        x = f(x), y = f(f(y));
    }
    return gcd(p, n);
}
u64 primeFactor(u64 n) {
    return isPrime(n) ? n : primeFactor(pollard(n));
}

void solve() {
    i64 n, k;
    cin >> n >> k;

    vector<pair<i64, i64>> divs;
    {
        i64 x = n;
        while (x > 1) {
            i64 p = primeFactor(x);
            int c = 0;
            while (x % p == 0) {
                x /= p;
                c++;
            }
            divs.push_back({p, c});
        }
    }

    auto check = [&](i64 d) -> bool {
        const i64 t = n / d;
        auto cal = [&](i64 x) -> i64 {
            i64 s = 1;
            for (int i = 1; i <= k - 1; i++) {
                if (s > t / (x + d)) {
                    return t + 1;
                }
                s *= (x + d);
            }
            i64 cur = 0;
            for (int i = k - 1; i >= 0; i--) {
                cur += s;
                if (cur > t) {
                    return t + 1;
                }
                if (s != 1) {
                    s /= (x + d);
                    s *= x;
                }
            }
            return cur;
        };

        const i64 L = 1E9 + 10;
        i64 x = *ranges::partition_point(views::iota(1LL, L),
                [&](i64 x) {
                    return cal(x) < t;
                });
        // if (cal(x) == t) cerr << "a = " << d + x << " b = " << x << '\n';
        return cal(x) == t;
    };

    i64 ans = 0;
    auto dfs = [&](auto &&self, i64 x, int i) -> void {
        if (i == divs.size()) {
            ans += check(x);
            return;
        }
        auto [p, c] = divs[i];
        for (int j = 0; j <= c; j++) {
            self(self, x, i + 1);
            x *= p;
        }
    };
    dfs(dfs, 1, 0);

    cout << ans << '\n';
}

int main() {
    cin.tie(0)->sync_with_stdio(false);

    int t = 1;
    cin >> t;
    while (t--) {
        solve();
    }

    return 0;
}

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

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

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: 2ms
memory: 3612kb

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: 3548kb

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: 2ms
memory: 3596kb

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: 3592kb

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: 3828kb

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: 3536kb

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: 1ms
memory: 3784kb

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: 3596kb

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: 2ms
memory: 3620kb

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: 3464kb

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: 3528kb

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: 3516kb

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: 3588kb

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: 3764kb

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: 3532kb

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: 3752kb

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: 3464kb

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: 2ms
memory: 3784kb

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: 3784kb

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: 2ms
memory: 3752kb

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