답안 #241455

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
241455 2020-06-24T08:13:17 Z kartel Mag (COCI16_mag) C++14
120 / 120
702 ms 239224 KB
#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define in(x) freopen(x, "r", stdin)
#define out(x) freopen(x, "w", stdout)
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("-O3")
#define F first
#define S second
#define pb push_back
#define N +1000500
#define M ll(1e9 + 7)
#define sz(x) (int)x.size()
#define re return
#define oo ll(1e9)
#define el '\n'
#define pii pair <int, int>
#define err ld(1e-9)
#define last(x) x.back()
#define all(x) (x.begin(), x.end())
using namespace std;
//using namespace __gnu_pbds;
//typedef tree <int, null_type, less_equal <int> , rb_tree_tag, tree_order_statistics_node_update> ordered_set;
typedef long long ll;
typedef long double ld;

int f[N], ans, i, n, u, v;
ll c[N], a, b;
vector <int> g[N];

void dfs(int v, int pr)
{
    f[v] = (c[v] == 1);

    for (auto u : g[v])
    {
        if (u == pr) continue;

        dfs(u, v);

        if (f[v]) f[v] = max(f[u] + 1, f[v]);
    }
}

void DFS(int v, int pr, int to)
{
    vector <int> vc;
    vc.pb(0);
    vc.pb(to);

    for (auto u : g[v])
    {
        if (u == pr) continue;
        vc.pb(f[u]);
        sort(vc.begin(), vc.end());
        reverse(vc.begin(), vc.end());
        vc.resize(2);
    }

    ll A = c[v];
    ll B = vc[0] + vc[1] + 1;

//    cerr << v << ":" << A << " " << B << el;

    ll gcd = __gcd(A, B);

    A /= gcd;
    B /= gcd;

    if (A * b < a * B)
        a = A, b = B;

    for (auto u : g[v])
    {
        if (u == pr) continue;

        int nto = 0;

        if (c[v] == 1)
        {
            nto++;
            if (f[u] == vc[0]) nto += vc[1];
                          else nto += vc[0];
        }

        DFS(u, v, nto);
    }
}

int main()
{
    srand(time(0));
    ios_base::sync_with_stdio(0);
    iostream::sync_with_stdio(0);
    ios::sync_with_stdio(0);
    cin.tie(NULL);
    cout.tie(NULL);

//    in("input.txt");
//    out("output.txt");

    cin >> n;
    for (i = 1; i < n; i++)
    {
        cin >> u >> v;
        g[v].pb(u);
        g[u].pb(v);
    }

    a = 1e9;
    b = 1;

    for (i = 1; i <= n; i++) cin >> c[i], a = min(a, c[i]);

    dfs(1, -1);

    DFS(1, -1, 0);

    cout << a << "/" << b;
}

//
//00000
//00110
//00111
//00011
//00000
# 결과 실행 시간 메모리 Grader output
1 Correct 19 ms 23936 KB Output is correct
2 Correct 19 ms 23936 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 19 ms 23936 KB Output is correct
2 Correct 22 ms 23936 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 525 ms 127224 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 18 ms 23808 KB Output is correct
2 Correct 701 ms 237288 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 657 ms 233720 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 538 ms 66748 KB Output is correct
2 Correct 406 ms 55684 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 536 ms 67812 KB Output is correct
2 Correct 99 ms 28280 KB Output is correct
3 Correct 702 ms 239224 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 102 ms 28256 KB Output is correct
2 Correct 570 ms 68216 KB Output is correct
3 Correct 428 ms 45936 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 502 ms 65900 KB Output is correct
2 Correct 555 ms 66040 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 573 ms 67960 KB Output is correct
2 Correct 572 ms 46072 KB Output is correct