| Task: | Tekijät |
| Sender: | mango_lassi |
| Submission time: | 2020-11-08 20:02:26 +0200 |
| Language: | C++ (C++11) |
| Status: | READY |
| Result: | 100 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 12 |
| #2 | ACCEPTED | 35 |
| #3 | ACCEPTED | 53 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.04 s | 1, 2, 3 | details |
| #2 | ACCEPTED | 0.04 s | 1, 2, 3 | details |
| #3 | ACCEPTED | 0.04 s | 2, 3 | details |
| #4 | ACCEPTED | 0.05 s | 3 | details |
| #5 | ACCEPTED | 0.08 s | 3 | details |
| #6 | ACCEPTED | 0.05 s | 3 | details |
| #7 | ACCEPTED | 0.05 s | 3 | details |
Code
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int H = 1e6;
int cou[H+1];
int mob[H+1];
vector<int> primes;
int div_ind[H+1];
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
for (int i = 0; i <= H; ++i) div_ind[i] = -1;
mob[1] = 1;
for (int i = 2; i <= H; ++i) {
if (div_ind[i] == -1) {
div_ind[i] = primes.size();
primes.push_back(i);
mob[i] = -1;
}
for (int j = 0; j <= div_ind[i]; ++j) {
ll t = (ll) i * primes[j];
if (t > H) break;
div_ind[t] = j;
if (j < div_ind[i]) mob[t] = -mob[i];
else mob[t] = 0;
}
}
int n;
cin >> n;
// \sum_{d | n} \mu(d) = [n == 1]
// \sum_{i = 1}^{H} \sum_{j = 1}^{H} [gcd(i, j) == 1] w_i w_j
// = \sum_{i = 1}^{H} \sum_{j = 1}^{H} \sum_{d | gcd(i, j)} \mu(d) w_i w_j
// = \sum_{d = 1}^{H} \mu d (\sum_{d | i} w_i) (\sum_{d | j} w_j)
for (int i = 0; i < n; ++i) {
int x;
cin >> x;
++cou[x];
}
ll res = -cou[1];
for (int i = 1; i <= H; ++i) {
for (int j = 2*i; j <= H; j += i) {
cou[i] += cou[j];
}
}
// for (int v = 1; v <= 20; ++v) cout << cou[v] << ' '; cout << '\n';
// for (int v = 1; v <= 20; ++v) cout << mob[v] << ' '; cout << '\n';
for (int d = 1; d <= H; ++d) {
res += mob[d] * (ll)cou[d] * cou[d];
}
cout << res / 2 << '\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 |
