oj.uz

답안 #245931

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
245931 2020-07-07T20:06:56 Z Vladikus004 Mag (COCI16_mag) C++17
108 / 120
 651 ms 215676 KB 
#include <bits/stdc++.h>
//#define int long long
#define inf 2e9
#define all(v) v.begin(), v.end()
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair <int, int> pii;

const int N = 1000000 + 3;
int n, z[N], p[N], udp[N], ddp[N];
vector <vector <int> > v, mx;

void udfs(int x){
    p[x] = 1;
//    cout << x << " " << v[x].size() << "\n";
//    cout << p[v[x][0]] << " " << p[v[x][1]] << "!!!\n";
//    if ( x == 432518) {
//        exit(0);
//    }
    for (auto u: v[x]){
        if (p[u]) continue;
        udfs(u);
        mx[x].push_back(udp[u]);
        udp[x] = max(udp[x], udp[u]);
    }
    if (z[x] == 1) udp[x]++;
    else udp[x] = 0;
    if (mx[x].size()){
    sort(all(mx[x]));
    reverse(all(mx[x]));
    }
}

int ans ;
int e_ans ;
int mn = inf;

void ddfs(int x, int way){
    p[x] = 1;
    mn = min(mn, z[x]);
    e_ans = max(e_ans, way);
    if (z[x] == 2){
        if (!mx[x].empty()){
        vector <int> buff;
        buff.push_back(way);
        buff.push_back(mx[x][0]);
        if (mx[x].size() > 1)
        buff.push_back(mx[x][1]);
        sort(all(buff));
//        cout << x << " : ";
//        for (auto u: v[x]) cout << u << " "; cout << "\n";
        ans = max(ans, buff[(buff.size() - 1) / 2] * 2 + 1);
        }
    }
    for (auto u: v[x]){
        if (p[u]) continue;
        if (z[x] != 1) {
            ddfs(u, 0);
            continue;
        }
        e_ans = max(e_ans, udp[u]);
        if (udp[u] == mx[x][0]){
            if (mx[x].size() > 1)
            ddfs(u, max(way + 1, mx[x][1]));
            else ddfs(u, way + 1);
        }else{
            ddfs(u, max(way + 1, mx[x][0]));
        }
    }
    if (mx[x].size()){
    int t = mx[x][0];
    if (mx[x].size() > 1) t += mx[x][1];
    if (z[x] == 1) e_ans = max(e_ans, t + 1);
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    #ifdef LOCAL
        freopen("input.txt", "r", stdin);
    #endif // LOCAL
    cin >> n;
    v.resize(n);
    mx.resize(n);
    for (int i = 0; i < n - 1; i++){
        int a, b;
        cin >> a >> b;
        --a; --b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    for (int i = 0; i < n; i++) cin >> z[i];
    udfs(0);
//    return 0;
    memset(p, 0, sizeof p);
    ddfs(0, 0);
//    cout << ans <<"!\n";
    if (ans){
        if (e_ans * 2 < ans)
            cout << "2/"<<ans;
        else
            cout << "1/"<<e_ans;
    }else
    if (e_ans){
        cout << "1/"<<e_ans;
    }else
        cout << mn<<"/1";
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 4352 KB Output is correct
2 Correct 7 ms 4352 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 4352 KB Output is correct
2 Correct 7 ms 4352 KB Output is correct

Subtask #3
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 474 ms 136016 KB Output isn't correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 4224 KB Output is correct
2 Correct 651 ms 213760 KB Output is correct

Subtask #5
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 582 ms 210436 KB Output is correct

Subtask #6
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 539 ms 120824 KB Output is correct
2 Correct 380 ms 90104 KB Output is correct

Subtask #7
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 514 ms 112748 KB Output is correct
2 Correct 96 ms 14584 KB Output is correct
3 Correct 594 ms 215676 KB Output is correct

Subtask #8
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 98 ms 14584 KB Output is correct
2 Correct 521 ms 121468 KB Output is correct
3 Correct 432 ms 56056 KB Output is correct

Subtask #9
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 505 ms 109172 KB Output is correct
2 Correct 547 ms 119160 KB Output is correct

Subtask #10
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 543 ms 122104 KB Output is correct
2 Correct 456 ms 56160 KB Output is correct