#include <bits/stdc++.h>
#include "garden.h"
#include "gardenlib.h"
using namespace std;
const int N = 150010;
const int INF = 2e9;
vector<int> mn[N], G[2*N];
int nxt[2*N], deg[2*N], h[2*N], ph[2*N];
bool comp[2*N][3];
void expand(int u, int c, int s, int d=0, int p=-INF)
{
if (comp[u][c])
return;
if (deg[u] != 1 && (u == 2*s || u == 2*s+1))
p = 0;
h[u] = d, ph[u] = p;
comp[u][c] = true;
for (auto v : G[u])
expand(v, c, s, d+1, p+1);
}
void count_routes(int n, int m, int p, int R[][2], int q, int K[])
{
// find min and second-min
for (int i = 0; i < m; ++i) {
int u = R[i][0], v = R[i][1];
if (mn[u].size() < 2)
mn[u].push_back(v);
if (mn[v].size() < 2)
mn[v].push_back(u);
}
// completely second-min
for (int i = 0; i < n; ++i) {
if (mn[i].size() == 0) continue;
if (mn[i].size() == 1) mn[i].push_back(mn[i].back());
}
// find nxt of each node
for (int i = 0; i < n; ++i) {
if (mn[i].size() == 0) continue;
// normal node
int j = mn[i][0];
if (mn[j][0] != i) nxt[2*i] = 2*j;
else nxt[2*i] = 2*j+1;
// walked node
j = mn[i][1];
if (mn[j][0] != i) nxt[2*i+1] = 2*j;
else nxt[2*i+1] = 2*j+1;
}
// find degree and make reversed graph: G
for (int u = 0; u < 2*n; ++u) {
++deg[nxt[u]];
G[nxt[u]].push_back(u);
}
// find cycle
queue<int> Q;
for (int u = 0; u < 2*n; ++u) {
if (deg[u] == 0)
Q.push(u);
}
while (!Q.empty()) {
int u = Q.front();
Q.pop();
if (--deg[nxt[u]] == 0)
Q.push(nxt[u]);
}
// expand
expand(2*p, 1, p);
expand(2*p+1, 2, p);
// find cycle size for 2*p and 2*p+1 (just in case they're different)
int c1 = 0, u = 2*p;
if (deg[u] == 1) {
do {
++c1;
u = nxt[u];
} while (u != 2*p);
}
int c2 = 0;
u = 2*p+1;
if (deg[u] == 1) {
do {
++c2;
u = nxt[u];
} while (u != 2*p+1);
}
// answer each query
for (int i = 0; i < q; ++i) {
int k = K[i], cnt = 0;
for (int j = 0; j < n; ++j) {
int u = 2*j;
bool ans = ph[u] == K[i];
if (comp[u][1])
ans |= (c1 > 0 && k-h[u] >= 0 && (k-h[u])%c1 == 0);
if (comp[u][2])
ans |= (c2 > 0 && k-h[u] >= 0 && (k-h[u])%c2 == 0);
if (ans) ++cnt;
}
answer(cnt);
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
11108 KB |
Output is correct |
2 |
Correct |
13 ms |
11128 KB |
Output is correct |
3 |
Correct |
13 ms |
11128 KB |
Output is correct |
4 |
Correct |
12 ms |
10872 KB |
Output is correct |
5 |
Correct |
13 ms |
11000 KB |
Output is correct |
6 |
Correct |
13 ms |
11256 KB |
Output is correct |
7 |
Correct |
12 ms |
10876 KB |
Output is correct |
8 |
Correct |
13 ms |
11000 KB |
Output is correct |
9 |
Correct |
15 ms |
11128 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
11108 KB |
Output is correct |
2 |
Correct |
13 ms |
11128 KB |
Output is correct |
3 |
Correct |
13 ms |
11128 KB |
Output is correct |
4 |
Correct |
12 ms |
10872 KB |
Output is correct |
5 |
Correct |
13 ms |
11000 KB |
Output is correct |
6 |
Correct |
13 ms |
11256 KB |
Output is correct |
7 |
Correct |
12 ms |
10876 KB |
Output is correct |
8 |
Correct |
13 ms |
11000 KB |
Output is correct |
9 |
Correct |
15 ms |
11128 KB |
Output is correct |
10 |
Correct |
12 ms |
11000 KB |
Output is correct |
11 |
Correct |
26 ms |
13688 KB |
Output is correct |
12 |
Correct |
43 ms |
15724 KB |
Output is correct |
13 |
Incorrect |
81 ms |
33948 KB |
Output isn't correct |
14 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
11108 KB |
Output is correct |
2 |
Correct |
13 ms |
11128 KB |
Output is correct |
3 |
Correct |
13 ms |
11128 KB |
Output is correct |
4 |
Correct |
12 ms |
10872 KB |
Output is correct |
5 |
Correct |
13 ms |
11000 KB |
Output is correct |
6 |
Correct |
13 ms |
11256 KB |
Output is correct |
7 |
Correct |
12 ms |
10876 KB |
Output is correct |
8 |
Correct |
13 ms |
11000 KB |
Output is correct |
9 |
Correct |
15 ms |
11128 KB |
Output is correct |
10 |
Correct |
12 ms |
11000 KB |
Output is correct |
11 |
Correct |
26 ms |
13688 KB |
Output is correct |
12 |
Correct |
43 ms |
15724 KB |
Output is correct |
13 |
Incorrect |
81 ms |
33948 KB |
Output isn't correct |
14 |
Halted |
0 ms |
0 KB |
- |