CSES - Putka Open 2020 – 4/5 - Results
Submission details
Task:Tekijät
Sender:Yytsi
Submission time:2020-11-08 15:09:50 +0200
Language:C++ (C++17)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED12
#2ACCEPTED35
#3ACCEPTED53
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1, 2, 3details
#2ACCEPTED0.02 s1, 2, 3details
#3ACCEPTED0.04 s2, 3details
#4ACCEPTED0.05 s3details
#5ACCEPTED0.08 s3details
#6ACCEPTED0.22 s3details
#7ACCEPTED0.04 s3details

Compiler report

input/code.cpp:1:0: warning: ignoring #pragma optimize  [-Wunknown-pragmas]
 #pragma optimize ("-O3")

Code

#pragma optimize ("-O3")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int n;
int smallestFactor[1010101];
vector<int> primeFac(int x) {
if (x == 1) return {1};
vector<int> facs;
while (x > 1) {
int small = smallestFactor[x];
if (facs.empty() || facs.back() != small) facs.push_back(small);
x /= small;
}
return facs;
}
ll cnt[1010101]; // cnt of divisors seen so far
int main() {
ios_base::sync_with_stdio(0); cin.tie(0);
cin>>n;
for (int d = 2; d <= 1000000; d++) {
if (smallestFactor[d] > 0) continue;
for (int x = d; x <= 1000000; x += d) {
if (smallestFactor[x] == 0) smallestFactor[x] = d;
else smallestFactor[x] = min(smallestFactor[x], d);
}
}
ll ans = 0;
for (int i = 0; i < n; i++) {
int x; cin>>x;
if (x == 1) { ans += (ll)i; continue; }
vector<int> facs = primeFac(x);
int m = facs.size();
int res = 0;
for (int S = (1 << m) - 1; S >= 1; S--) {
int div = 1;
for (int j = 0; j < m; j++) {
if (S & (1 << j)) {
div *= facs[j];
}
}
res += __builtin_popcount(S) % 2 ? +cnt[div] : -cnt[div];
cnt[div]++;
}
ans += ll(i - res);
}
cout<<ans<<"\n";
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3043

user output
3043

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
71 19 26 18 57 78 80 89 31 26 ...

correct output
3086

user output
3086

Test 3

Group: 2, 3

Verdict: ACCEPTED

input
100000
66 87 90 67 93 89 57 29 34 4 8...

correct output
3044751906

user output
3044751906

Test 4

Group: 3

Verdict: ACCEPTED

input
100000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3039650753

user output
3039650753

Test 5

Group: 3

Verdict: ACCEPTED

input
100000
238907 151373 522599 885657 37...

correct output
3031155756

user output
3031155756

Test 6

Group: 3

Verdict: ACCEPTED

input
100000
510510 510510 510510 510510 51...

correct output
0

user output
0

Test 7

Group: 3

Verdict: ACCEPTED

input
100000
999983 999983 999983 999983 99...

correct output
0

user output
0