oj.uz

Submission #251800

# Submission time Handle Problem Language Result Execution time Memory
251800 2020-07-22T10:24:33 Z Vimmer Uzastopni (COCI15_uzastopni) C++14
64 / 160
 142 ms 8320 KB 
#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")

#define F first
#define S second
#define sz(x) int(x.size())
#define pb push_back
#define pf push_front
#define N 100010
#define M ll(10007)
#define inf 1e9 + 1e9

using namespace std;
//using namespace __gnu_pbds;

typedef long double ld;
typedef long long ll;
typedef unsigned long long ull;
typedef short int si;
typedef array <int, 3> a3;
typedef array <int, 4> a4;
typedef pair <int, int> pt;
//typedef tree <int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;


int n, ans, from[N], a[N], siz[N], h[N], l, r;

bool mk[105];

vector <int> val[105];

set <int> st[N];

vector <int> g[N];

bool bad;

void dfs_rev(int v)
{
    st[v].insert(a[v]);

    int lst = -1;

    for (auto it : st[v])
    {
        if (lst != -1 && it - lst != 1) {bad = 1; return;}

        lst = it;
    }

    if (v != 1)
    {
        int nxt = from[v];

        if (nxt == 0) {bad = 1; return;}

        if (sz(st[nxt]) < sz(st[v])) swap(st[v], st[nxt]);

        for (auto it : st[v]) st[nxt].insert(it);
    }
    else if (r - l + 1 != sz(st[v])) bad = 1;

}

bool cmp(int x, int y) {return h[x] > h[y];}

void dfs(int v)
{
    siz[v] = 1;

    for (auto it : g[v])
    {
        h[it] = h[v] + 1;

        dfs(it);

        siz[v] += siz[it];
    }
}

int main()
{
    //freopen("mining.in", "r", stdin); freopen("mining.out", "w", stdout);

    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    cin >> n;

    for (int i = 1; i <= n; i++) {cin >> a[i]; val[a[i]].pb(i); mk[a[i]] = 1;}

    for (int i = 1; i < n; i++)
    {
        int x, y;

        cin >> x >> y;

        g[x].pb(y);

        from[y] = x;
    }

    dfs(1);

    for (l = 1; l <= 100; l++)
      if (mk[l])
        for (r = l; r <= 100; r++)
        {
            if (!mk[r]) break;

            if (!(l <= a[1] && a[1] <= r)) continue;

            bad = 0;

            vector <int> gr; gr.clear();

            for (int i = 1; i <= n; i++) st[i].clear();

            for (int i = l; i <= r; i++) gr.pb(val[i].back());

            sort(gr.begin(), gr.end(), cmp);

            for (auto it : gr) {dfs_rev(it); if (bad) break;}

            if (!bad) ans++;
            else
            {
                bad = 0;

                gr.clear();

                for (int i = 1; i <= n; i++) st[i].clear();

                for (int i = l; i <= r; i++) gr.pb(val[i][0]);

                sort(gr.begin(), gr.end(), cmp);

                for (auto it : gr) {dfs_rev(it); if (bad) break;}

                if (!bad) ans++;
            }
        }

    cout << ans << endl;
}

Subtask #1
64.0 / 160.0

# Verdict Execution time Memory Grader output
1 Incorrect 5 ms 7424 KB Output isn't correct
2 Incorrect 4 ms 7424 KB Output isn't correct
3 Incorrect 4 ms 7424 KB Output isn't correct
4 Incorrect 6 ms 7424 KB Output isn't correct
5 Incorrect 6 ms 7424 KB Output isn't correct
6 Correct 18 ms 7424 KB Output is correct
7 Correct 28 ms 7424 KB Output is correct
8 Correct 9 ms 7424 KB Output is correct
9 Correct 5 ms 7424 KB Output is correct
10 Correct 5 ms 7424 KB Output is correct
11 Incorrect 9 ms 7808 KB Output isn't correct
12 Incorrect 9 ms 7808 KB Output isn't correct
13 Correct 9 ms 7808 KB Output is correct
14 Incorrect 92 ms 8320 KB Output isn't correct
15 Incorrect 66 ms 8320 KB Output isn't correct
16 Incorrect 142 ms 8320 KB Output isn't correct
17 Correct 8 ms 7808 KB Output is correct
18 Correct 10 ms 7808 KB Output is correct
19 Incorrect 97 ms 7800 KB Output isn't correct
20 Incorrect 102 ms 7680 KB Output isn't correct