QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#346194 | #983. The Hash Table | ucup-team1209 | AC ✓ | 15ms | 3716kb | C++20 | 2.0kb | 2024-03-07 22:57:53 | 2024-03-07 22:57:54 |
Judging History
answer
#include<bits/stdc++.h>
using std::cin, std::cout;
using ll = long long;
using u64 = unsigned long long;
using db = double;
const int N = 200;
int t, n, m;
int p[N], c[N], cc;
ll ans = 0;
ll ipow(ll a, ll b) {
ll ans = 1;
for(;b;--b) ans *= a;
return ans;
}
u64 floor_sum(u64 n, u64 m, u64 a, u64 b) {
u64 ans = 0;
for(;;) {
if(a >= m) ans += n * (n - 1) / 2 * (a / m), a %= m;
if(b >= m) ans += n * (b / m), b %= m;
u64 ymax = a * n + b; // use u128 if it's big
if(ymax < m) break;
n = ymax / m;
b = ymax % m;
std::swap(m, a);
}
return ans;
}
ll solve(int index, ll a, ll b) {
if(index == cc) {
ll sum = 0;
if(b % 2 == 0 || a % 2 == 0) {
if(a % 2) a += a;
if(b % 2) b += b;
ll l = 0, mid = (n - 1) / a, r = (2 * n - 1) / a;
sum += floor_sum(mid + 1, b, a, 0);
sum += floor_sum(r - mid, b, a, 2 * n - 1 - r * a);
} else {
ll l = 0, mid = (n - 1) / a, r = (2 * n - 1) / a;
sum += floor_sum((mid + 1 + 1) / 2, b * 2, a * 2, 0);
sum += floor_sum((mid + 1 + 0) / 2, b * 2, a * 2, b + a);
int sgn = r & 1;
sum += floor_sum((r - mid + !sgn) / 2, b * 2, a * 2, 2 * n - 1 - r * a + sgn * a);
sum += floor_sum((r - mid + sgn) / 2, b * 2, a * 2, 2 * n - 1 - r * a + b + !sgn * a);
}
return sum;
}
ll ans = 0;
for(int i = 0;i <= c[index];++i) {
ans += solve(index + 1, a * ipow(p[index], i), b * ipow(p[index], c[index] - i));
}
for(int i = 1;i <= c[index];++i) {
ans -= solve(index + 1, a * ipow(p[index], i), b * ipow(p[index], c[index] - i + 1));
}
return ans;
}
void solve() {
cin >> n >> m, cc = 0;
memset(c, 0, sizeof(c));
int x = m;
for(int i = 2;i * i <= x;++i) {
if(x % i == 0) {
p[cc] = i;
while(x % i == 0) {
++ c[cc];
x /= i;
}
++ cc;
}
}
if(x > 1) p[cc] = x, c[cc] = 1, ++ cc;
ans = solve(0, 1, 1);
cout << ans << '\n';
}
int main() {
std::ios::sync_with_stdio(false), cin.tie(0);
cin >> t;
for(int i = 0;i < t;++i) {
solve();
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3712kb
input:
3 5 4 1234 5678 5 4
output:
4 229 4
result:
ok 3 tokens
Test #2:
score: 0
Accepted
time: 1ms
memory: 3672kb
input:
5 919191919 998244353 919191919 308308924 124312512 700980343 199712020 199712020 1000000000 1000000000
output:
420069742 18975162173 34523625 619107226 36400000000
result:
ok 5 tokens
Test #3:
score: 0
Accepted
time: 0ms
memory: 3716kb
input:
5 351177178 2 236814804 3 406487669 4 107706571 5 206252441 6
output:
30831352411422332 15578125268692158 41308056059019556 2088126907497711 5908342860201211
result:
ok 5 tokens
Test #4:
score: 0
Accepted
time: 12ms
memory: 3628kb
input:
5 795965436 698377680 775372176 735134400 891189723 555660000 255414898 555660000 649803503 967458816
output:
252131828509 252898591532 196092107465 16019483439 23110007395
result:
ok 5 tokens
Test #5:
score: 0
Accepted
time: 13ms
memory: 3576kb
input:
5 980056074 669278610 786804145 536870912 94743614 669278610 925596206 551350800 277753133 735134400
output:
151494104505 16332684795 1386109471 391665329539 32367779875
result:
ok 5 tokens
Test #6:
score: 0
Accepted
time: 15ms
memory: 3708kb
input:
5 407928831 901800900 959460685 551350800 929921323 901800900 880362378 551350800 169887581 551350800
output:
24176516170 420863755861 126194540248 354298688005 13128893811
result:
ok 5 tokens
Test #7:
score: 0
Accepted
time: 1ms
memory: 3620kb
input:
5 755923335 903960666 248398318 89883150 32020160 729647777 48000958 745177909 261977197 120541067
output:
5122927135 10499768724 1250410 5736978 980686184
result:
ok 5 tokens
Test #8:
score: 0
Accepted
time: 1ms
memory: 3644kb
input:
5 120618885 904615503 860299168 970357266 2922308 99586084 116144876 802899228 621873179 98196855
output:
72362680 1890408813 77476 180492837 23279216759
result:
ok 5 tokens
Test #9:
score: 0
Accepted
time: 0ms
memory: 3668kb
input:
5 337929091 905270340 472298323 850733078 973824456 50926103 184354331 860587779 981801929 75885412
output:
2695205304 791722210 18137207967 13641011 50173808550
result:
ok 5 tokens
Test #10:
score: 0
Accepted
time: 1ms
memory: 3632kb
input:
5 725900160 906132534 140660629 407214962 935092856 84357639 706890499 221743500 461717852 516104007
output:
3448528660 288899219 61675795925 96294825272 475220310
result:
ok 5 tokens
Test #11:
score: 0
Accepted
time: 1ms
memory: 3648kb
input:
5 943210366 906787371 752627016 287590774 905995005 183214075 775132721 279464819 821646602 493792564
output:
11587484479 1606204266 21788486368 3802176853 10347608659
result:
ok 5 tokens
Test #12:
score: 0
Accepted
time: 1ms
memory: 3580kb
input:
5 943210366 900720143 752627016 900720143 905995005 942060233 775132721 942060233 821646602 962178361
output:
1522977645 931104902 1306217667 941841841 691366758
result:
ok 5 tokens