oj.uz

Submission #167643

# Submission time Handle Problem Language Result Execution time Memory
167643 2019-12-09T08:06:30 Z Flying_dragon_02 Mag (COCI16_mag) C++14
108 / 120
 557 ms 147956 KB 
#include <bits/stdc++.h>

using namespace std;

#define fi first
#define se second
#define pb push_back
#define mp make_pair

typedef pair< int, int > ii;

const int N = 1e6 + 5;

const long long infll = (long long)(1e18);

int n, f[N][2], x[N], pref[2];

vector< int > graph[N];

bool vis[N];

double mn;

long long X, Y;

double F(long long a, long long b) {
    if( b == 0 ) return infll;
    else return (double)( (double)( a ) / (double)( b ) );
}

void dfs(int u, int p) {
    f[u][ x[u] - 1 ] = 1;
    vis[u] = 1;
    for(int i = 0; i < (int)( graph[u].size() ); i++ ) {
        int v = graph[u][i];
        if( v == p || x[v] > 2 ) continue;
        dfs( v, u );
        if( x[u] == 1 ) {
            f[u][0] = max( f[u][0], f[v][0] + 1 );
            f[u][1] = max( f[u][1], f[v][1] + 1 );
        }
        else
            f[u][1] = max( f[u][1], f[v][0] + 1 );
    }
    pref[0] = pref[1] = 0;
    for(int i = 0; i < (int)( graph[u].size() ); i++ ) {
        int v = graph[u][i];
        if( v == p || x[v] > 2 ) continue;
        if( x[u] == 1 ) {
            if( f[v][0] && F( 1, f[v][0] + 1 + pref[0] ) < mn ) {
                X = 1;
                mn = F( 1, f[v][0] + 1 + pref[0] );
                Y = f[v][0] + 1 + pref[0];
            }
            if( f[v][1] && F( 2, f[v][1] + 1 + pref[0] ) < mn ) {
                X = 2;
                mn = F( 2, f[v][1] + 1 + pref[0] );
                Y = f[v][1] + 1 + pref[0];
            }
            if( f[v][0] && F( 2, f[v][0] + 1 + pref[1] ) < mn ) {
                X = 2;
                mn = F( 2, f[v][0] + 1 + pref[1] );
                Y = f[v][0] + 1 + pref[1];
            }
        }
        else {
            if( f[v][0] && F( 2, f[v][0] + 1 + pref[0] ) < mn ) {
                X = 2;
                mn = F( 2, f[v][0] + 1 + pref[0] );
                Y = f[v][0] + 1 + pref[0];
            }
        }
        for(int j = 0; j < 1; j++)
            pref[j] = max( pref[j], f[v][j] );
    }
}

int main() {
    cin.tie(0), ios_base::sync_with_stdio(0);
    cin >> n;
    for(int i = 1; i <= n - 1; i++) {
        int u, v;
        cin >> u >> v;
        graph[u].pb(v);
        graph[v].pb(u);
    }
    mn = infll;
    for(int i = 1; i <= n; i++) {
        cin >> x[i];
        if(x[i] < mn) {
            mn = x[i];
            X = x[i];
            Y = 1;
        }
    }
    for(int i = 1; i <= n; i++) {
        if(!vis[i] && x[i] < 3) {
            dfs( i, i );
        }
    }
    for(int i = 1; i <= n; i++) {
        if(!vis[i]) continue;
        //cout << i << " " << f[i][0] << " " << f[i][1] << "\n";
        if( F( 1, f[i][0] ) < mn ) {
            X = 1;
            mn = F( 1, f[i][0] );
            Y = f[i][0];
        }
        if( F( 2, f[i][1] ) < mn ) {
            X = 2;
            mn = F( 2, f[i][1] );
            Y = f[i][1];
        }
    }
    cout << X << "/" << Y;
}

Subtask #1
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 23800 KB Output is correct
2 Correct 23 ms 23900 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 23928 KB Output is correct
2 Correct 23 ms 23928 KB Output is correct

Subtask #3
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 418 ms 84656 KB Output isn't correct

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 23800 KB Output is correct
2 Correct 557 ms 145008 KB Output is correct

Subtask #5
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 520 ms 128356 KB Output is correct

Subtask #6
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 467 ms 67864 KB Output is correct
2 Correct 355 ms 67320 KB Output is correct

Subtask #7
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 451 ms 69064 KB Output is correct
2 Correct 92 ms 29944 KB Output is correct
3 Correct 542 ms 147956 KB Output is correct

Subtask #8
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 93 ms 29652 KB Output is correct
2 Correct 473 ms 68636 KB Output is correct
3 Correct 382 ms 54068 KB Output is correct

Subtask #9
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 429 ms 65464 KB Output is correct
2 Correct 466 ms 67320 KB Output is correct

Subtask #10
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 479 ms 68912 KB Output is correct
2 Correct 378 ms 47292 KB Output is correct