QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#244749 | #4478. Jo loves counting | ucup-team203 | AC ✓ | 748ms | 37148kb | C++20 | 2.5kb | 2023-11-09 15:19:41 | 2023-11-09 15:19:42 |
Judging History
answer
#include <bits/stdc++.h>
const int N = 2e6 + 10;
using namespace std;
#define ll __int128_t
const ll Mod = (1ll << 57) * 29ll + 1;
ll Pow(ll n, ll k)
{
ll res = 1;
for (ll x = n; k; x = x * x % Mod, k >>= 1)
if (k & 1)
res = res * x % Mod;
return res;
}
ll isp[N], p[N], tot;
ll M;
ll ans = 0, inv2;
ll inv[100];
struct node
{
/* data */
ll now, pos, h;
};
void calc()
{
stack<node> st;
st.push((node){1, 0, 1});
while (!st.empty())
{
node tmp = st.top();
st.pop();
ll now = tmp.now, pos = tmp.pos, h = tmp.h;
// printf("%lld %lld %lld\n",(long long) pos,(long long) M,(long long) p[pos]*p[pos]);
if (p[pos] * p[pos] > M / now)
{
if (now == 1)
{
ans += (M * (M + 1) % Mod * inv2) % Mod;
ans%=Mod;
}
else
{
ans += now * h % Mod * (M / now) % Mod * (M / now + 1) % Mod * inv2 % Mod;
ans = (ans % Mod + Mod) % Mod;
// printf("%lld %lld\n",(long long)now,(long long)(h>0?1:-1)*now*Pow(abs(h),Mod-2)%Mod);
}
// printf("%lld\n",(long long)ans);
continue;
}
// calc(now, pos + 1, h);
if (p[pos] * p[pos] > M / now)
continue;
st.push((node){now, pos + 1, h});
// if(p[pos]==0)
// {
// printf("%lld"p[pos-1]);
// }
for (ll i = 2, pj = p[pos] * p[pos]; now <= M / pj; i++, pj *= p[pos])
{
// calc(now * pj, pos + 1, h * (-i * (i - 1)));
st.push((node){now * pj, pos + 1, h * inv[i] % (ll)Mod});
}
}
}
void solve()
{
long long _;
cin >> _;
M = _;
ans=0;
calc();
ans = ans * Pow(M, Mod - 2) % Mod;
printf("%lld\n", (long long)ans);
}
int main()
{
inv2 = Pow(2, Mod - 2);
for (int i = 1; i < 100; i++)
inv[i] = -Pow(i * (i - 1), Mod - 2);
for (int i = 2; i <= N - 10; i++)
{
if (!isp[i])
p[tot++] = i;
for (int j = 0; i * p[j] < N && j < tot; j++)
{
isp[i * p[j]] = 1;
if (i % p[j])
;
else
break;
}
}
int T;
cin >> T;
while (T--)
{
solve();
}
}
/*
12
4
959139
9859
125987
209411
965585325539
213058376259
451941492387
690824608515
934002691939
1000000000000
1
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 748ms
memory: 37148kb
input:
12 4 959139 9859 125987 209411 965585325539 213058376259 451941492387 690824608515 934002691939 1000000000000 1
output:
2 2544652967005436170 1531132428015608197 4112905740393076193 2210911161352244432 3734327250979959061 3166689602614938339 2205751131915532448 1440445581699720020 350511728590182760 1099683734530790325 1
result:
ok 12 lines