#include <bits/stdc++.h>
#define int long long
using namespace std;
const int nmax = 200000;
const int block = 1000;
long long n, r, q;
int region[nmax + 5], cntRegion[nmax + 5];
int bigRegion[nmax + 5], cntBigRegion, bigIdx[nmax + 5];
int smallRegion[nmax + 5], cntSmallRegion, smallIdx[nmax + 5];
long long cntAbove[nmax + 5], sol[205][25005], sumePartiale[nmax + 5];
long long precalc[205][25005];
int cntTimp, newtime;
int stiva[nmax + 5];
vector<pair<int, int>> lista[25005];
pair<int, int> timp[nmax + 5];
vector<int> nodesFromRegion[nmax + 5];
vector<int> g[nmax + 5];
void dfs(int fiu, int tata)
{
newtime++;
lista[region[fiu]].push_back(make_pair(fiu, newtime));
timp[fiu].first = ++cntTimp;
/// cntAbove[i] -> numarul de noduri de regiunea i(care este o regiune de tip BIG), care sunt stramosi ai nodului curent
if(bigIdx[region[fiu]] != 0)
{
cntAbove[bigIdx[region[fiu]]] ++;
}
for(int i = 1; i <= 200; i ++)
{
sol[i][region[fiu]] += 1LL * cntAbove[i];
}
for(auto it : g[fiu])
{
if(it == tata)
{
continue;
}
dfs(it, fiu);
}
if(bigIdx[region[fiu]] != 0)
{
cntAbove[bigIdx[region[fiu]]] ++;
}
newtime++;
lista[region[fiu]].push_back(make_pair(fiu, newtime));
timp[fiu].second = cntTimp;
}
signed main()
{
cin >> n >> r >> q;
cin >> region[1];
cntRegion[region[1]] ++;
for(int i = 2; i <= n; i ++)
{
int p;
cin >> p >> region[i];
g[p].push_back(i);
g[i].push_back(p);
cntRegion[region[i]] ++;
nodesFromRegion[region[i]].push_back(i);
}
for(int i = 1; i <= r; i ++)
{
if(cntRegion[i] >= block)
{
bigRegion[++cntBigRegion] = i;
bigIdx[i] = cntBigRegion;
}
else
{
smallRegion[++cntSmallRegion] = i;
smallIdx[i] = cntSmallRegion;
}
}
/// in dfs calculez pentru tipurile 1 si 2, cand regiunea de sus este de tip BIG
dfs(1, 0);
/// imi propun sa calculez pentru tipul 3, cand regiunea de jos este de tip BIG, iar regiunea de sus este de tip SMALL
for(long long i = 1; i <= r; i ++)
{
if(bigIdx[i] == 0)
{
continue;
}
for(auto it : nodesFromRegion[i])
{
sumePartiale[timp[it].first] ++;
}
for(long long j = 1; j <= n; j ++)
{
sumePartiale[j] += 1LL * sumePartiale[j - 1];
}
for(long long j = 1; j <= cntSmallRegion; j ++)
{
for(auto it : nodesFromRegion[smallRegion[j]])
{
precalc[bigIdx[i]][j] += 1LL * (sumePartiale[timp[it].second] - sumePartiale[timp[it].first - 1]);
}
}
for(long long j = 1; j <= n; j ++)
{
sumePartiale[j] = 0;
}
}
while(q--)
{
long long a, b;
cin >> a >> b;
if(bigIdx[a] != 0)
{
/// cand regiunea de sus e de tip BIG
cout << sol[bigIdx[a]][b] << '\n';
}
else
{
if(bigIdx[b] != 0)
{
/// cand regiunea de jos e BIG, dar cea de sus e SMALL
///cout << precalc[bigIdx[b]][smallIdx[a]] << '\n';
}
else
{
long long i = 0, j = 0, vf = 0;
long long deschise = 0, answer = 0;
while(i < lista[a].size() && j < lista[b].size())
{
pair<int, int> frst = lista[a][i];
pair<int, int> scnd = lista[b][j];
if(frst.second < scnd.second)
{
if(vf && frst.first == stiva[vf])
{
vf--;
deschise--;
}
else
{
stiva[++vf] = frst.first;
deschise++;
}
i++;
}
else
{
if(vf && scnd.first == stiva[vf])
{
vf--;
}
else
{
answer += 1LL * deschise;
stiva[++vf] = scnd.first;
}
j++;
}
}
cout << answer << '\n';
}
}
}
return 0;
}
Compilation message
regions.cpp: In function 'int main()':
regions.cpp:135:25: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
135 | while(i < lista[a].size() && j < lista[b].size())
| ~~^~~~~~~~~~~~~~~~~
regions.cpp:135:48: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
135 | while(i < lista[a].size() && j < lista[b].size())
| ~~^~~~~~~~~~~~~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
6 ms |
11216 KB |
Output is correct |
| 2 |
Correct |
6 ms |
11216 KB |
Output is correct |
| 3 |
Correct |
8 ms |
11216 KB |
Output is correct |
| 4 |
Correct |
11 ms |
11216 KB |
Output is correct |
| 5 |
Correct |
15 ms |
11344 KB |
Output is correct |
| 6 |
Correct |
29 ms |
11728 KB |
Output is correct |
| 7 |
Correct |
36 ms |
11600 KB |
Output is correct |
| 8 |
Correct |
40 ms |
11728 KB |
Output is correct |
| 9 |
Correct |
54 ms |
12752 KB |
Output is correct |
| 10 |
Correct |
90 ms |
13008 KB |
Output is correct |
| 11 |
Correct |
137 ms |
13340 KB |
Output is correct |
| 12 |
Correct |
132 ms |
14612 KB |
Output is correct |
| 13 |
Correct |
164 ms |
14928 KB |
Output is correct |
| 14 |
Correct |
154 ms |
15440 KB |
Output is correct |
| 15 |
Correct |
289 ms |
20232 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Execution timed out |
76 ms |
21420 KB |
Time limit exceeded (wall clock) |
| 2 |
Execution timed out |
78 ms |
21304 KB |
Time limit exceeded (wall clock) |
| 3 |
Execution timed out |
98 ms |
24940 KB |
Time limit exceeded (wall clock) |
| 4 |
Correct |
309 ms |
21484 KB |
Output is correct |
| 5 |
Correct |
418 ms |
25600 KB |
Output is correct |
| 6 |
Correct |
924 ms |
28464 KB |
Output is correct |
| 7 |
Correct |
1280 ms |
35528 KB |
Output is correct |
| 8 |
Correct |
1199 ms |
43980 KB |
Output is correct |
| 9 |
Correct |
2088 ms |
55080 KB |
Output is correct |
| 10 |
Correct |
2969 ms |
76980 KB |
Output is correct |
| 11 |
Correct |
2787 ms |
73756 KB |
Output is correct |
| 12 |
Execution timed out |
525 ms |
58760 KB |
Time limit exceeded (wall clock) |
| 13 |
Execution timed out |
494 ms |
60768 KB |
Time limit exceeded (wall clock) |
| 14 |
Execution timed out |
583 ms |
69528 KB |
Time limit exceeded (wall clock) |
| 15 |
Execution timed out |
562 ms |
80140 KB |
Time limit exceeded (wall clock) |
| 16 |
Execution timed out |
547 ms |
89932 KB |
Time limit exceeded (wall clock) |
| 17 |
Execution timed out |
581 ms |
84240 KB |
Time limit exceeded (wall clock) |
