oj.uz

Submission #872188

# Submission time Handle Problem Language Result Execution time Memory
872188 2023-11-12T13:14:45 Z overwatch9 Regions (IOI09_regions) C++17
1 / 100
 1 ms 596 KB 
#include <iostream>
#include <unordered_map>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 10;
int visit[maxn], finish[maxn], region[maxn];
vector <int> adj[maxn];
vector <int> region_members[maxn];
int t = 0;
void dfs(int s, int p) {
    visit[s] = t++;
    region_members[region[s]].push_back(s);
    for (auto i : adj[s])
        dfs(i, s);
    finish[s] = t++;
}
bool is_ancestor(int a, int b) {
    return visit[a] <= visit[b] && finish[a] >= finish[b];
}
int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    int n, r, q;
    cin >> n >> r >> q;
    cin >> region[1];
    for (int i = 2; i <= n; i++) {
        int x;
        cin >> x;
        adj[x].push_back(i);
        cin >> region[i];
    }
    dfs(1, 1);
    while (q--) {
        int r1, r2;
        cin >> r1 >> r2;
        int ans = 0;
        vector <int> starts(region_members[r2].size());
        for (int j = 0; j < region_members[r2].size(); j++)
            starts[j] = visit[region_members[r2][j]];
        sort(starts.begin(), starts.end());
        for (int i = 0; i < region_members[r1].size(); i++) {
            auto l = upper_bound(starts.begin(), starts.end(), visit[region_members[r1][i]]);
            auto r = upper_bound(starts.begin(), starts.end(), finish[region_members[r1][i]]);
            if (r == starts.begin())
                continue;
            r--;
            int id1 = l - starts.begin();
            int id2 = r - starts.begin();
            ans += max(0, (id2 - id1 + 1));
        }
        cout << ans << endl;
    }
}

Compilation message

regions.cpp: In function 'int main()':
regions.cpp:40:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |         for (int j = 0; j < region_members[r2].size(); j++)
      |                         ~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
regions.cpp:43:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |         for (int i = 0; i < region_members[r1].size(); i++) {
      |                         ~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
1.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Runtime error 0 ms 344 KB Execution killed with signal 11
3 Runtime error 0 ms 344 KB Execution killed with signal 11
4 Runtime error 0 ms 344 KB Execution killed with signal 11
5 Runtime error 0 ms 344 KB Execution killed with signal 11
6 Runtime error 1 ms 344 KB Execution killed with signal 11
7 Runtime error 0 ms 344 KB Execution killed with signal 11
8 Runtime error 0 ms 344 KB Execution killed with signal 11
9 Runtime error 1 ms 344 KB Execution killed with signal 11
10 Runtime error 1 ms 344 KB Execution killed with signal 11
11 Runtime error 1 ms 344 KB Execution killed with signal 11
12 Runtime error 0 ms 344 KB Execution killed with signal 11
13 Runtime error 1 ms 344 KB Execution killed with signal 11
14 Runtime error 1 ms 344 KB Execution killed with signal 11
15 Runtime error 1 ms 344 KB Execution killed with signal 11

Subtask #2
0 / 85.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 344 KB Execution killed with signal 11
2 Runtime error 0 ms 344 KB Execution killed with signal 11
3 Runtime error 1 ms 344 KB Execution killed with signal 11
4 Runtime error 0 ms 344 KB Execution killed with signal 11
5 Runtime error 0 ms 344 KB Execution killed with signal 11
6 Runtime error 1 ms 344 KB Execution killed with signal 11
7 Runtime error 1 ms 344 KB Execution killed with signal 11
8 Runtime error 1 ms 344 KB Execution killed with signal 11
9 Runtime error 0 ms 344 KB Execution killed with signal 11
10 Runtime error 1 ms 344 KB Execution killed with signal 11
11 Runtime error 1 ms 344 KB Execution killed with signal 11
12 Runtime error 1 ms 344 KB Execution killed with signal 11
13 Runtime error 1 ms 344 KB Execution killed with signal 11
14 Runtime error 0 ms 344 KB Execution killed with signal 11
15 Runtime error 0 ms 596 KB Execution killed with signal 11
16 Runtime error 0 ms 344 KB Execution killed with signal 11
17 Runtime error 1 ms 344 KB Execution killed with signal 11