QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#880832#7735. Primitive RootmvecWA 26ms3840kbC++231.3kb2025-02-03 21:19:562025-02-03 21:19:57

Judging History

This is the latest submission verdict.

  • [2025-02-03 21:19:57]
  • Judged
  • Verdict: WA
  • Time: 26ms
  • Memory: 3840kb
  • [2025-02-03 21:19:56]
  • Submitted

answer

#include <iostream>
#include <cmath>

using namespace std;

typedef unsigned long long ull;

int GetDigit(ull x)
{
    return log2(x) + 1;
}

ull Solve1(ull p, ull m)
{
    int d_up = GetDigit(m);
    ull k = 0, res = 0;
    while (GetDigit(k * p + 1) <= d_up)
    {
        if (((k * p + 1) ^ (p - 1)) <= m)
            res++;
        k++;
    }

    return res;
}

ull Solve2(ull p, ull m)
{
    int d_ps1 = GetDigit(p - 1);
    ull full = m & (-1ULL - (ull)(pow(2, d_ps1 + 1) - 1));

    ull res = full / p;
    if (full % p)
        res++;
    if (!full)
        res = 0;

    ull start = full / p * p + 1;
    while (start <= full)
        start += p;
    for (int i = 0; i <= 4; i++)
        if (((start + i * p) ^ (p - 1)) <= m)
            res++;

    return res;
}

int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        ull p, m;
        cin >> p >> m;

        if (!m)
        {
            int ans = 0;
            if (p == 2)
                ans++;
            cout << ans << '\n';
        }
        else
        {
            if (GetDigit(p - 1) >= GetDigit(m))
                cout << Solve1(p, m) << '\n';
            else
                cout << Solve2(p, m) << '\n';
        }
    }

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

3
2 0
7 11
1145141 998244353

output:

1
2
872

result:

ok 3 lines

Test #2:

score: -100
Wrong Answer
time: 26ms
memory: 3712kb

input:

47595
2 0
2 1
2 2
2 3
2 4
2 5
2 6
2 7
2 8
2 9
2 10
2 11
2 12
2 13
2 14
2 15
2 16
2 17
2 18
2 19
2 20
2 21
2 22
2 23
2 24
2 25
2 26
2 27
2 28
2 29
2 30
2 31
2 32
2 33
2 34
2 35
2 36
2 37
2 38
2 39
2 40
2 41
2 42
2 43
2 44
2 45
2 46
2 47
2 48
2 49
2 50
2 51
2 52
2 53
2 54
2 55
2 56
2 57
2 58
2 59
2 60...

output:

1
1
2
2
3
3
4
4
5
5
6
6
7
7
8
8
9
9
10
10
11
11
12
12
13
13
14
14
15
15
16
16
17
17
18
18
19
19
20
20
21
21
22
22
23
23
24
24
25
25
26
26
27
27
28
28
29
29
30
30
31
31
32
32
33
33
34
34
35
35
36
36
37
37
38
38
39
39
40
40
41
41
42
42
43
43
44
44
45
45
46
46
47
47
48
48
49
49
50
50
51
51
52
52
53
53
...

result:

wrong answer 518th lines differ - expected: '5', found: '6'