QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#64914 | #4229. GCD Harmony | Three_Sannin# | WA | 42ms | 199640kb | C++ | 2.3kb | 2022-11-25 21:31:20 | 2022-11-25 21:31:23 |
Judging History
answer
// Hey there.. I love you <3
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/numeric>
using namespace std;
using namespace __gnu_pbds;
using ll = long long;
using ld = long double;
using L = __int128;
#define fast cin.tie(0), cin.sync_with_stdio(0)
FILE *stream;
#define input(x) freopen_s(&stream, (x), "r", stdin)
#define output(x) freopen_s(&stream, (x), "w", stdout)
#define all(x) x.begin(), x.end()
#define sz(x) ((int)(x).size())
#define maxz(x, y) x = max(x, y)
#define minz(x, y) x = min(x, y)
#define modz(x) x = x % mod + mod, x -= (x >= mod) * mod
#define mp(x, y) make_pair(x, y)
#define f first
#define s second
#define pii pair<int, int>
#define ordered_set(x) tree<x, null_type,less<x>, rb_tree_tag,tree_order_statistics_node_update> //less_equal
mt19937 rnd(time(nullptr));
const int N = 5e3 + 5, M = N * N, mod = 1e9 + 7;
const ld pi = acos(-1), eps = 1e-6;
int prime[N];
vector<int> prs[N];
void sieve() {
for (int i = 2; i < N; ++i) prime[i] = 1;
for (int i = 2; i < N; ++i) if (prime[i]) for (int j = 2 * i; j < N; j += i) prime[j] = 0, prs[j].push_back(i);
}
int n, p[N], a[N], dp[N][N], done[N][N];
vector<int> e[N];
int solve(int u, int g);
void help(int u) {
for (int i = 2; i < N; ++i) {
int sum = 0;
for (auto v: e[u]) if (v != p[u]) p[v] = u, sum += solve(v, i);
done[u][i] = i + sum;
}
}
int solve(int u = 0, int g = 0) {
auto &me = dp[u][g];
if (~me) return me;
me = 1e4;
if (gcd(a[u], g) != 1) {
int sum = 0;
for (auto v: e[u]) if (v != p[u]) p[v] = u, sum += solve(v, a[u]);
minz(me, sum);
}
if (!~done[u][2]) help(u);
for (int i = 2; i <= 2 * sz(e[u]); ++i) if (gcd(i, g) != 1) minz(me, done[u][i]);
return me;
}
void hi() {
cin >> n;
for (int i = 0; i < n; ++i) cin >> a[i];
for (int i = 0; i < n - 1; ++i) {
int u, v;
cin >> u >> v, e[--u].push_back(--v), e[v].push_back(u);
}
cout << solve();
}
void init() {
fast;
memset(dp, -1, sizeof dp), memset(done, -1, sizeof done);
sieve();
}
signed main() {
init();
int T = 1;
//cin >> T;
for (int tc = 1; tc <= T; ++tc) hi();
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 42ms
memory: 199508kb
input:
6 5 6 3 4 9 12 1 2 1 3 1 4 1 6 3 5
output:
6
result:
ok single line: '6'
Test #2:
score: 0
Accepted
time: 28ms
memory: 199640kb
input:
3 1 2 3 3 1 2 3
output:
4
result:
ok single line: '4'
Test #3:
score: 0
Accepted
time: 20ms
memory: 199580kb
input:
13 2 5 5 5 5 5 5 3 3 3 3 3 3 1 2 1 3 1 4 1 5 1 6 1 7 1 8 1 9 1 10 1 11 1 12 1 13
output:
15
result:
ok single line: '15'
Test #4:
score: 0
Accepted
time: 23ms
memory: 199608kb
input:
9 2 5 5 5 5 3 3 3 3 1 2 1 3 1 4 1 5 1 6 1 7 1 8 1 9
output:
14
result:
ok single line: '14'
Test #5:
score: -100
Wrong Answer
time: 39ms
memory: 199588kb
input:
8 13 13 13 2 13 13 13 13 1 2 1 3 1 4 1 5 1 6 1 7 1 8
output:
14
result:
wrong answer 1st lines differ - expected: '13', found: '14'