QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#64914#4229. GCD HarmonyThree_Sannin#WA 42ms199640kbC++2.3kb2022-11-25 21:31:202022-11-25 21:31:23

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2022-11-25 21:31:23]
  • 评测
  • 测评结果:WA
  • 用时:42ms
  • 内存:199640kb
  • [2022-11-25 21:31:20]
  • 提交

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;
}

Details

Tip: Click on the bar to expand more detailed information

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'