Submission #675233

#	Time	Username	Problem	Language	Result	Execution time	Memory
675233		vjudge1	Uzastopni (COCI15_uzastopni)	C++17	160 / 160	109 ms	18808 KiB

uzastopni

#include <bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
#define faster ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
using ll = long long;
using ld = long double;
using pii = pair <int, int>;
mt19937_64 Rand(chrono::steady_clock::now().time_since_epoch().count());
const int maxN = 1e4 + 10;
const int maxM = 1e2 + 10;
const int Mod = 1e9 + 7;
//const int inf =
int n;

bitset <maxM> dp[maxN][maxM];
vector <int> adj[maxN];
bitset <maxM> B[maxM];
int a[maxN];
void dfs(int u){
    dp[u][a[u]][a[u]] = 1;
    for (int i: adj[u]){
        dfs(i);
    }
    for (int i = 1; i <= 100; ++i) B[i].reset();
    for (int i: adj[u]){
        for (int j = 1; j <= 100; ++j){
            B[j] |= dp[i][j];
        }
    }
    for (int i = 1; i <= 100; ++i){
        for (int j = i + 1; j <= 100; ++j){
            if (B[i][j - 1]) B[i] |= B[j];
        }
    }
    for (int i = 1; i <= a[u]; ++i){
        for (int j = a[u]; j <= 100; ++j){
            if ((i == a[u] || B[i][a[u] - 1]) && (j == a[u] || B[a[u] + 1][j])) dp[u][i][j] = 1;
        }
    }
}

void Init(){
    cin >> n;
    for (int i = 1; i <= n; ++i) cin >> a[i];
    for (int i = 1; i < n; ++i){
        int u, v;
        cin >> u >> v;
        adj[u].pb(v);
    }
    dfs(1);
    int ans = 0;
    for (int i = 1; i <= 100; ++i){
        ans += dp[1][i].count();
    }
    cout << ans << "\n";
}

#define debu
#define taskname "test"
signed main(){
    faster
    if (fopen(taskname".inp", "r")){
        freopen(taskname".inp", "r", stdin);
        freopen(taskname".out", "w", stdout);
    }
    int tt = 1;
    //cin >> tt;
    while (tt--){
        Init();
    }
    if (fopen("timeout.txt", "r")){
        ofstream timeout("timeout.txt");
        timeout << signed(double(clock()) / CLOCKS_PER_SEC * 1000);
        timeout.close();
        #ifndef debug
        cerr << "Time elapsed: " << signed(double(clock()) / CLOCKS_PER_SEC * 1000) << "ms\n";
        #endif // debug
    }
}

Compilation message (stderr)

uzastopni.cpp: In function 'int main()':
uzastopni.cpp:65:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   65 |         freopen(taskname".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
uzastopni.cpp:66:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   66 |         freopen(taskname".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 160 / 160