oj.uz

Submission #1084236

# Submission time Handle Problem Language Result Execution time Memory
1084236 2024-09-05T16:46:32 Z serifefedartar Inside information (BOI21_servers) C++17
0 / 100
 345 ms 32180 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
 
#define fast ios::sync_with_stdio(0);cin.tie(0)
typedef long long ll;
#define f first
#define s second
#define LOGN 21
const ll MOD = 1e9 + 7;
const ll MAXN = 2e5 + 100;
 
tree<pair<int,int>, null_type, less<pair<int,int>>, rb_tree_tag, tree_order_statistics_node_update> os;
vector<vector<pair<int,int>>> graph;
vector<array<int,3>> query;
vector<int> query_ids[MAXN];
int N, K, sz[MAXN], ans[MAXN], marked[MAXN], color[MAXN];
int ttt[MAXN], out_smallest[MAXN], out_greatest[MAXN], in_greatest[MAXN];
 
int get_sz(int node, int parent) {
    sz[node] = 1;
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f])
            sz[node] += get_sz(u.f, node);
    }
    return sz[node];
}
 
int find_centro(int node, int parent, int n) {
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f] && sz[u.f] * 2 >= n)
            return find_centro(u.f, node, n);
    }
    return node;
}
 
vector<int> V;
void dfs(int node, int parent, int minimum_edge, int last_edge) {
    V.push_back(node);
    out_smallest[node] = minimum_edge;
    out_greatest[node] = last_edge;
    for (auto u : graph[node]) {
        if (u.f == parent || marked[u.f])
            continue;
        if (u.s > last_edge)
            dfs(u.f, node, minimum_edge, u.s);
    }
}
 
void dfs2(int node, int parent, int minimum_edge, int maximum_edge) {
    V.push_back(node);
    in_greatest[node] = maximum_edge;
    for (auto u : graph[node]) {
        if (u.f == parent || marked[u.f])
            continue;
        if (u.s < minimum_edge)
            dfs2(u.f, node, u.s, maximum_edge);
    }
}
 
void dfs3(int node, int parent) {
    if (ttt[node] > 0 && in_greatest[node] <= ttt[node])
        ans[ttt[node]] += os.order_of_key({ttt[node], 1e9}) + 1;
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f])
            dfs3(u.f, node);
    }
}
 
void init(int node, int parent, int cc) {
    color[node] = cc;
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f])
            init(u.f, node, cc);
    }
}
 
 
void activate(int node, int parent) {
    if (out_smallest[node] != -1)
        os.insert({out_smallest[node], node});
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f])
            activate(u.f, node);
    }
}
 
void deactivate(int node, int parent) {
    if (out_smallest[node] != -1)
        os.erase({out_smallest[node], node});
    for (auto u : graph[node]) {
        if (u.f != parent && !marked[u.f])
            deactivate(u.f, node);
    }
}
 
void decompose(int node) {
    int n = get_sz(node, node);
    int centro = find_centro(node, node, n);
 
    marked[centro] = true;
    color[centro] = -1;
    int cc = 0;
    os.clear();
    for (auto u : graph[centro]) {
        if (!marked[u.f]) {
            init(u.f, node, ++cc);
            dfs(u.f, node, u.s, u.s);
            dfs2(u.f, node, u.s, u.s);
            activate(u.f, node);
        }
    }
 
    for (auto id : query_ids[centro]) {
        array<int,3> Q = query[id];
        if (Q[0] == centro)
            ans[Q[2]] += (in_greatest[Q[1]] != -1 && in_greatest[Q[1]] < Q[2]);
        else if (Q[1] == centro)
            ans[Q[2]] += (out_greatest[Q[0]] != -1 && out_greatest[Q[0]] < Q[2]);
    }
 
    for (auto u : V) {
        for (auto id : query_ids[u]) {
            array<int,3> Q = query[id];
            ans[Q[2]] += (color[Q[1]] != color[Q[0]] && out_smallest[Q[0]] != -1 && in_greatest[Q[1]] != -1 
                            && in_greatest[Q[1]] < out_smallest[Q[0]] && out_greatest[Q[0]] < Q[2]);
        }
    }
 
    int deact = 0;
    for (auto u : graph[centro]) {
        if (!marked[u.f]) {
            while (deact + 1 <= graph[centro].size() && graph[centro][deact].s <= u.s) {
                deactivate(graph[centro][deact].f, node);
                deact++;
            }
            dfs3(u.f, node);
        }
    }
 
    for (auto u : V)
        in_greatest[u] = out_smallest[u] = out_greatest[u] = -1;
    V = vector<int>();
 
    for (auto u : graph[centro]) {
        if (!marked[u.f])
            decompose(u.f);
    }
}
 
int main() {
    fast;
    memset(in_greatest, -1, sizeof(in_greatest));
    memset(out_smallest, -1, sizeof(out_smallest));
    memset(out_greatest, -1, sizeof(out_greatest));
    cin >> N >> K;
 
    graph = vector<vector<pair<int,int>>>(N+1, vector<pair<int,int>>());
    for (int i = 1; i <= N + K - 1; i++) {
        char ch;
        cin >> ch;
        if (ch == 'S') {
            int a, b;
            cin >> a >> b;
            graph[a].push_back({b, i});
            graph[b].push_back({a, i});
        } else if (ch == 'Q') {
            int a, b;
            cin >> a >> b;
            query_ids[a].push_back(query.size());
            query_ids[b].push_back(query.size());
            query.push_back({a, b, i});
        } else {
            int a;
            cin >> a;
            ttt[a] = i;
            query.push_back({a, -1, i});
        }
    }
    for (int i = 1; i <= N; i++) {
        sort(graph[i].begin(), graph[i].end(), [&](pair<int,int> A, pair<int,int> B) {
            return A.s < B.s;
        });
    }
    decompose(1);
 
    for (auto u : query) {
        if (u[1] != -1)
            cout << (ans[u[2]] || u[0] == u[1] ? "yes\n" : "no\n");
        else
            cout << ans[u[2]] << "\n";
    }
}

Compilation message

servers.cpp: In function 'void decompose(int)':
servers.cpp:134:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  134 |             while (deact + 1 <= graph[centro].size() && graph[centro][deact].s <= u.s) {
      |                    ~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 2.5

# Verdict Execution time Memory Grader output
1 Incorrect 29 ms 12036 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 2.5

# Verdict Execution time Memory Grader output
1 Incorrect 29 ms 12036 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 2.5

# Verdict Execution time Memory Grader output
1 Correct 23 ms 12044 KB Output is correct
2 Incorrect 178 ms 32180 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 2.5

# Verdict Execution time Memory Grader output
1 Correct 23 ms 12044 KB Output is correct
2 Incorrect 178 ms 32180 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #5
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 12040 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 12040 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 12048 KB Output is correct
2 Incorrect 345 ms 28460 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 12048 KB Output is correct
2 Incorrect 345 ms 28460 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #9
0 / 12.5

# Verdict Execution time Memory Grader output
1 Incorrect 25 ms 12036 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 12.5

# Verdict Execution time Memory Grader output
1 Incorrect 25 ms 12036 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #11
0 / 17.5

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 12160 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #12
0 / 17.5

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 12160 KB Output isn't correct
2 Halted 0 ms 0 KB -