oj.uz

Submission #675225

# Submission time Handle Problem Language Result Execution time Memory
675225 2022-12-27T03:13:10 Z Do_you_copy Uzastopni (COCI15_uzastopni) C++17
160 / 160
 99 ms 18824 KB 
#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

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.0 / 160.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 596 KB Output is correct
2 Correct 1 ms 596 KB Output is correct
3 Correct 1 ms 596 KB Output is correct
4 Correct 1 ms 596 KB Output is correct
5 Correct 1 ms 724 KB Output is correct
6 Correct 2 ms 724 KB Output is correct
7 Correct 1 ms 724 KB Output is correct
8 Correct 1 ms 724 KB Output is correct
9 Correct 1 ms 724 KB Output is correct
10 Correct 1 ms 724 KB Output is correct
11 Correct 74 ms 17900 KB Output is correct
12 Correct 77 ms 17840 KB Output is correct
13 Correct 72 ms 17876 KB Output is correct
14 Correct 99 ms 18824 KB Output is correct
15 Correct 89 ms 18764 KB Output is correct
16 Correct 86 ms 18824 KB Output is correct
17 Correct 71 ms 17860 KB Output is correct
18 Correct 70 ms 17868 KB Output is correct
19 Correct 81 ms 17800 KB Output is correct
20 Correct 83 ms 17784 KB Output is correct