oj.uz

Submission #556371

# Submission time Handle Problem Language Result Execution time Memory
556371 2022-05-03T03:52:02 Z Olympia Mag (COCI16_mag) C++17
24 / 120
 991 ms 198152 KB 
#include <bits/stdc++.h>
using namespace std;
vector<int> val;
vector<int> parent;
vector<int> dp1, dp2, dp3;
//dp1 is the maximum path that goes down
//dp2 is the maximum path that does not necessarily go down
//dp3 is the maximum path that goes up (straight up)
int get_max (vector<int>& v) {
    pair<int,int> p = make_pair(-1, -1);
    for (int i: v) {
        if (i > p.first) {
            p = make_pair(i, 1);
        } if (i == p.first) {
            p.second++;
        }
    }
    if (p.second >= 2) {
        return p.first * 2;
    }
    int myMax = 0;
    for (int i: v) {
        if (i == p.first) {
            continue;
        }
        myMax = max(myMax, i);
    }
    return myMax + p.first;
}
void dfs (vector<vector<int> > &adj, int curNode, int prevNode) {
    dp1[curNode] = (val[curNode] == 1);
    for (int i: adj[curNode]) {
        if (i != prevNode) {
            dfs (adj, i, curNode);
            dp1[curNode] = max(dp1[curNode], dp1[i] + 1);
        }
    }
    if (val[curNode] != 1) {
        dp1[curNode] = 0;
    }
}
void dfs2 (vector<vector<int> > &adj, int curNode, int prevNode) {
    parent[curNode] = prevNode;
    dp2[curNode] = dp1[curNode];
    vector<int> v;
    for (int i: adj[curNode]) {
        if (i != prevNode) {
            dfs2(adj, i, curNode);
            v.push_back(dp1[i]);
        }
    }
    if (v.size() >= 2) {
        dp2[curNode] = max(dp2[curNode], get_max(v) + 1);
    }
    if (val[curNode] != 1) {
        dp2[curNode] = 0;
    }
}

void dfs3 (vector<vector<int> >&adj, int curNode, int prevNode) {
    if (curNode != prevNode) {
        dp3[curNode] = dp3[prevNode] + 1;
    } else {
        dp3[curNode] = 1;
    }
    if (val[curNode] != 1) {
        dp3[curNode] = 0;
    }
    for (int i: adj[curNode]) {
        if (i != prevNode) {
            dfs3(adj, i, curNode);
        }
    }
}

int main() {
    vector<vector<int> > adj;
    int n;
    cin >> n;
    adj.resize(n), dp1.resize(n), dp2.resize(n), dp3.resize(n), parent.resize(n);
    for (int i = 0; i < n - 1; i++) {
        int u, v;
        cin >> u >> v;
        u--, v--;
        adj[u].push_back(v), adj[v].push_back(u);
    }
    val.resize(n);
    int myMin = INT_MAX;
    for (int i = 0; i < n; i++) {
        cin >> val[i];
        myMin = min(myMin, val[i]);
    }
    if (myMin >= 2) {
        cout << myMin << "/" << 1;
        return 0;
    }
    dfs(adj, 0, 0);
    dfs2(adj, 0, 0);
    dfs3(adj, 0, 0);
    int myMax2 = 0;
    int myMax1 = 0;
    for (int i = 0; i < n; i++) {
        myMax1 = max(myMax1, dp2[i]);
        if (val[i] == 2) {
            vector<int> v;
            for (int j: adj[i]) {
                if (j == parent[i]) {
                    v.push_back(dp3[j]);
                } else {
                    v.push_back(dp1[j]);
                }
            }
            sort(v.begin(), v.begin());
            reverse(v.begin(), v.end());
            myMax2 = max(myMax2, get_max(v));
        }
    }
    vector<int> dum;
    dum.push_back(2); dum.push_back(2), dum.push_back(1);
    myMax2++;
    cout << 2 << "/" << myMax2 << '\n';

}

Subtask #1
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Incorrect 1 ms 340 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Incorrect 1 ms 340 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 774 ms 109844 KB Output is correct

Subtask #4
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Incorrect 982 ms 198152 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #5
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 991 ms 195004 KB Output is correct

Subtask #6
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 829 ms 74160 KB Output is correct
2 Incorrect 626 ms 55120 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #7
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 889 ms 78648 KB Output is correct
2 Incorrect 111 ms 7964 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #8
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 107 ms 7844 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 811 ms 70672 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 854 ms 75468 KB Output isn't correct
2 Halted 0 ms 0 KB -