oj.uz

Submission #884709

# Submission time Handle Problem Language Result Execution time Memory
884709 2023-12-08T07:17:20 Z parlimoos Uzastopni (COCI15_uzastopni) C++14
144 / 160
 5 ms 3812 KB 
//Be Name KHODA
#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;
#define pb push_back
#define pp pop_back
#define cl clear
#define bg begin
#define lb lower_bound
#define ub upper_bound
#define arr(x) array<int , x>
#define endl '\n'

int n;
int vs[10000];
vector<arr(2)> tr[10000];
bool ls[10000][100];
bool rs[10000][100];

void dfs(int v = 0){
    // cout << v << ">>\n" << flush;
    for(auto el : tr[v]) dfs(el[1]);
    ls[v][vs[v]] = 1 , rs[v][vs[v]] = 1;
    arr(2) d = {vs[v] , -1};
    int inx1 = int(lb(tr[v].bg() , tr[v].end() , d) - tr[v].bg());
    while(inx1 > 0){
        inx1--;
        bool flg = 0;
        for(int i = 99 ; i >= 1 ; i--){
            if(ls[v][i] and rs[tr[v][inx1][1]][i - 1]){
                flg = 1;
                break;
            }
        }
        if(flg) for(int i = 99 ; i >= 0 ; i--) if(ls[tr[v][inx1][1]][i]) ls[v][i] = 1;
    }
    d = {vs[v] , int(1e9)};
    int inx2 = int(ub(tr[v].bg() , tr[v].end() , d) - tr[v].bg());
    while(inx2 < (int)tr[v].size()){
        bool flg = 0;
        for(int i = 0 ; i < 99 ; i++){
            if(rs[v][i] and rs[tr[v][inx2][1]][i + 1]){
                flg = 1;
                break;
            }
        }
        if(flg) for(int i = 0 ; i <= 99 ; i++) if(rs[tr[v][inx2][1]][i]) rs[v][i] = 1;
        inx2++;
    }
}

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n;
    for(int i = 0 ; i < n ; i++){
        cin >> vs[i];
        vs[i]--;
    }
    for(int i = 1 ; i < n ; i++){
        int a , b;
        cin >> a >> b;
        tr[a - 1].pb({vs[b - 1] , b - 1});
    }
    for(int i = 0 ; i < n ; i++) sort(tr[i].bg() , tr[i].end());
    dfs();
    int o1 = 0 , o2 = 0;
    for(int i = 0 ; i < 100 ; i++){
        if(ls[0][i]) o1++;
        if(rs[0][i]) o2++;
    }
    cout << 1ll * o1 * o2;
}

Subtask #1
144.0 / 160.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2392 KB Output isn't correct
2 Correct 1 ms 2508 KB Output is correct
3 Correct 1 ms 2396 KB Output is correct
4 Correct 1 ms 2396 KB Output is correct
5 Correct 1 ms 2396 KB Output is correct
6 Correct 1 ms 2396 KB Output is correct
7 Correct 1 ms 2512 KB Output is correct
8 Correct 1 ms 2516 KB Output is correct
9 Correct 1 ms 2396 KB Output is correct
10 Correct 1 ms 2396 KB Output is correct
11 Correct 4 ms 2908 KB Output is correct
12 Incorrect 4 ms 2904 KB Output isn't correct
13 Correct 4 ms 2908 KB Output is correct
14 Correct 4 ms 3812 KB Output is correct
15 Correct 5 ms 3676 KB Output is correct
16 Correct 4 ms 3804 KB Output is correct
17 Correct 4 ms 2904 KB Output is correct
18 Correct 5 ms 2908 KB Output is correct
19 Correct 4 ms 2652 KB Output is correct
20 Correct 4 ms 2652 KB Output is correct