#include <bits/stdc++.h>
using namespace std;
const int nax = 2e5 + 5;
int region[nax];
vector<int> edges[nax], positions[nax];
int tin[nax], tout[nax], who[nax];
int timer = 0;
unordered_map<int, int> mp[25005];
void dfs (int u, int p) {
tin[u] = ++timer;
who[timer] = u;
positions[region[u]].push_back(timer);
for (int v : edges[u]) {
if (v != p) {
dfs (v, u);
}
}
tout[u] = timer;
}
void solve() {
int n, r, q;
cin >> n >> r >> q;
cin >> region[1];
for (int i = 2; i <= n; ++i) {
int p;
cin >> p >> region[i];
edges[i].push_back(p);
edges[p].push_back(i);
}
dfs (1, 0);
const int B = sqrt(n) + 5;
for (int reg = 1; reg <= r; ++reg) {
if (positions[reg].size() >= B) {
vector<int> pref(n + 2);
for (int i = 1; i <= n; ++i) {
if (region[who[i]] == reg) {
pref[i + 1]++;
pref[tout[who[i]] + 1]--;
}
}
// for (int i = 1; i <= n; ++i) {
// pref[i] += pref[i - 1];
// mp[reg][region[who[i]]] += pref[i];
// }
}
}
// while (q--) {
// int reg1, reg2;
// cin >> reg1 >> reg2;
// if (positions[reg1].size() >= B) {
// cout << mp[reg1][reg2] << endl;
// } else {
// int ans = 0;
// for (int i : positions[reg1]) {
// if (region[who[i]] == reg1) {
// auto it1 = upper_bound(positions[reg2].begin(), positions[reg2].end(), tout[who[i]]) - positions[reg2].begin();
// auto it2 = lower_bound(positions[reg2].begin(), positions[reg2].end(), i) - positions[reg2].begin();
// if (it1 != 0) {
// ans += (int) it1 - it2;
// }
// }
// }
// cout << ans << endl;
// }
// }
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
solve();
}
Compilation message
regions.cpp: In function 'void solve()':
regions.cpp:38:35: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'const int' [-Wsign-compare]
38 | if (positions[reg].size() >= B) {
| ~~~~~~~~~~~~~~~~~~~~~~^~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
13400 KB |
Unexpected end of file - int32 expected |
2 |
Incorrect |
3 ms |
13652 KB |
Unexpected end of file - int32 expected |
3 |
Incorrect |
3 ms |
13400 KB |
Unexpected end of file - int32 expected |
4 |
Incorrect |
3 ms |
13400 KB |
Unexpected end of file - int32 expected |
5 |
Incorrect |
3 ms |
13400 KB |
Unexpected end of file - int32 expected |
6 |
Incorrect |
4 ms |
13400 KB |
Unexpected end of file - int32 expected |
7 |
Incorrect |
3 ms |
13400 KB |
Unexpected end of file - int32 expected |
8 |
Incorrect |
3 ms |
13656 KB |
Unexpected end of file - int32 expected |
9 |
Incorrect |
4 ms |
13912 KB |
Unexpected end of file - int32 expected |
10 |
Incorrect |
5 ms |
13912 KB |
Unexpected end of file - int32 expected |
11 |
Incorrect |
6 ms |
14084 KB |
Unexpected end of file - int32 expected |
12 |
Incorrect |
8 ms |
14528 KB |
Unexpected end of file - int32 expected |
13 |
Incorrect |
8 ms |
14424 KB |
Unexpected end of file - int32 expected |
14 |
Incorrect |
9 ms |
14936 KB |
Unexpected end of file - int32 expected |
15 |
Incorrect |
11 ms |
17240 KB |
Unexpected end of file - int32 expected |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
25 ms |
17644 KB |
Unexpected end of file - int32 expected |
2 |
Incorrect |
24 ms |
17372 KB |
Unexpected end of file - int32 expected |
3 |
Incorrect |
27 ms |
19160 KB |
Unexpected end of file - int32 expected |
4 |
Incorrect |
9 ms |
14936 KB |
Unexpected end of file - int32 expected |
5 |
Incorrect |
12 ms |
16216 KB |
Unexpected end of file - int32 expected |
6 |
Incorrect |
18 ms |
16272 KB |
Unexpected end of file - int32 expected |
7 |
Incorrect |
31 ms |
17100 KB |
Unexpected end of file - int32 expected |
8 |
Incorrect |
34 ms |
21376 KB |
Unexpected end of file - int32 expected |
9 |
Incorrect |
44 ms |
20956 KB |
Unexpected end of file - int32 expected |
10 |
Incorrect |
65 ms |
25780 KB |
Unexpected end of file - int32 expected |
11 |
Incorrect |
57 ms |
22492 KB |
Unexpected end of file - int32 expected |
12 |
Incorrect |
62 ms |
22440 KB |
Unexpected end of file - int32 expected |
13 |
Incorrect |
58 ms |
23032 KB |
Unexpected end of file - int32 expected |
14 |
Incorrect |
67 ms |
23192 KB |
Unexpected end of file - int32 expected |
15 |
Incorrect |
59 ms |
26524 KB |
Unexpected end of file - int32 expected |
16 |
Incorrect |
55 ms |
31584 KB |
Unexpected end of file - int32 expected |
17 |
Incorrect |
57 ms |
30728 KB |
Unexpected end of file - int32 expected |