QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#321631 | #4584. Not One | ssuyu | WA | 0ms | 30320kb | C++17 | 2.1kb | 2024-02-05 02:03:55 | 2024-02-05 02:03:56 |
Judging History
answer
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;
#define ssuyu \
ios_base::sync_with_stdio(0); \
cin.tie(0)
#define pb push_back
#define F first
#define S second
#define pii pair<int, int>
#define int long long
#define PI acos(-1)
#define all(x) (x).begin(), (x).end()
const int N = 1e6 + 5;
int a[N];
int prime[N] = {0}, f[N] = {0};
vector<int> primes;
int Find(int a) {
if (f[a] <= 0) {
return a;
} else
return f[a] = Find(f[a]);
}
int Union(int a, int b) {
int fa = Find(a);
int fb = Find(b);
if (fa == fb)
return -f[fa];
if (f[fa] > f[fb])
swap(fa, fb);
f[fa] += f[fb];
f[fb] = fa;
return -f[fa];
}
void getprimes(int n) {
for (int i = 2; i <= n; i++) {
if (!prime[i]) {
primes.pb(i);
}
for (auto u : primes) {
if (u * i > n)
break;
prime[u * i] = 1;
if (!(i % u))
break;
}
}
}
struct edge {
int u;
int v;
int w = 1;
} edge[N];
signed main() {
int n, m, x, y;
int mx = 0, ans = 0;
vector<int> affect;
map<int, vector<pii>> mp;
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> a[i];
mx = max(mx, a[i]);
}
getprimes(mx);
for (int i = 0; i < n - 1; i++) {
cin >> x >> y;
f[x] = f[y] = -1;
int w = __gcd(a[x], a[y]);
for (int j = 0; i < primes.size() && primes[j] * primes[j] < w; j++) {
if (w % primes[j] == 0) {
mp[primes[j]].pb({x, y});
}
while (w % primes[j] == 0)
w /= primes[j];
}
}
for (auto p : mp) {
for (auto u : p.S) {
ans = max(Union(u.F, u.S), ans);
affect.pb(u.F);
affect.pb(u.S);
}
for (auto u : affect)
f[u] = -1;
affect.clear();
}
cout << ans;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 0ms
memory: 30320kb
input:
7 10 5 8 6 10 6 4 1 2 2 3 2 4 4 5 4 6 4 7
output:
0
result:
wrong answer 1st lines differ - expected: '4', found: '0'