const int LG = 21;
const int N = 200005;
const long long MOD = 1e9 + 7;
const long long INF = 1e9;
const long long INFLL = 1e18;
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;
typedef vector<vector<int> > vvi;
typedef vector<ll> vll;
#define forn(i, n) for (int (i) = 0; (i) != (n); (i)++)
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
#define popcount(x) __builtin_popcount(x)
#define popcountll(x) __builtin_popcountll(x)
#define fi first
#define se second
#define re return
#define pb push_back
#define uniq(x) sort(all(x)); (x).resize(unique(all(x)) - (x).begin())
#ifdef LOCAL
#define dbg(x) cerr << __LINE__ << " " << #x << " " << x << endl
#define ln cerr << __LINE__ << endl
#else
#define dbg(x) void(0)
#define ln void(0)
#endif // LOCAL
void solve_line(int n, int m, int q)
{
vector<int> le(n), ri(n);
iota(all(le), 0);
iota(all(ri), 0);
vector<int> act(n - 1);
for (int i = 0; i < m; i++)
{
int z;
cin >> z;
z--;
// connects (z) and (z + 1)
if (act[z])
{
act[z] = 0;
continue;
}
act[z] = 1;
int r = z;
while (r < n - 1 && act[r]) r++;
int l = z;
while (l >= 0 && act[l]) l--;
// line from l + 1 to r
l++;
for (int j = l; j <= r; j++) ri[j] = ri[z + 1];
for (int j = l; j <= r; j++) le[j] = le[z];
}
vector<int> ans(n);
for (int i = 0; i < n; i++)
{
ans[i] = ri[i] - le[i] + 1;
}
for (int i = 0; i < q; i++)
{
int v;
cin >> v;
v--;
cout << ans[v] << "\n";
}
}
int dfs(int v, int p, int t, vvi &graph, vi &x, vi &y, vector<vector<pii> > &act)
{
int res = 1;
for (auto i : graph[v])
{
int u = x[i] + y[i] - v;
if (u != p)
{
int L = -1, R = act[i].size();
while (L + 1 < R)
{
int M = (L + R) / 2;
if (act[i][M].first <= t)
L = M;
else
R = M;
}
if (L == -1) continue;
int T = min(act[i][L].second, t);
res += dfs(u, v, T, graph, x, y, act);
}
}
return res;
}
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n, m, q;
cin >> n >> m >> q;
vector<int> x(n - 1), y(n - 1);
vector<vector<int> > graph(n);
int line = 1;
for (int i = 0; i < n - 1; i++)
{
cin >> x[i] >> y[i];
x[i]--, y[i]--;
if (x[i] != i || y[i] != i + 1) line = 0;
graph[x[i]].push_back(i);
graph[y[i]].push_back(i);
}
if (line)
{
solve_line(n, m, q);
return 0;
}
vector<int> fr(n - 1, -1);
vector<vector<pii> > act(n - 1);
for (int i = 0; i < m; i++)
{
int z;
cin >> z;
z--;
if (fr[z] == -1)
{
fr[z] = i;
}
else
{
act[z].push_back({fr[z], i});
fr[z] = -1;
}
}
for (int i = 0; i < n - 1; i++)
{
if (fr[i] != -1)
{
act[i].push_back({fr[i], m});
}
}
for (int i = 0; i < q; i++)
{
int v;
cin >> v;
v--;
cout << dfs(v, v, m, graph, x, y, act) << "\n";
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
0 ms |
384 KB |
Output is correct |
5 |
Correct |
0 ms |
384 KB |
Output is correct |
6 |
Correct |
1 ms |
512 KB |
Output is correct |
7 |
Correct |
6 ms |
1536 KB |
Output is correct |
8 |
Correct |
7 ms |
1536 KB |
Output is correct |
9 |
Correct |
6 ms |
1536 KB |
Output is correct |
10 |
Correct |
104 ms |
12280 KB |
Output is correct |
11 |
Correct |
91 ms |
12280 KB |
Output is correct |
12 |
Correct |
67 ms |
8228 KB |
Output is correct |
13 |
Correct |
75 ms |
12528 KB |
Output is correct |
14 |
Correct |
98 ms |
12536 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
70 ms |
19316 KB |
Output is correct |
2 |
Correct |
66 ms |
16244 KB |
Output is correct |
3 |
Execution timed out |
8016 ms |
7800 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
1 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
0 ms |
384 KB |
Output is correct |
5 |
Correct |
0 ms |
384 KB |
Output is correct |
6 |
Correct |
1 ms |
384 KB |
Output is correct |
7 |
Correct |
8 ms |
1152 KB |
Output is correct |
8 |
Correct |
81 ms |
8440 KB |
Output is correct |
9 |
Correct |
91 ms |
8440 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
89 ms |
8440 KB |
Output is correct |
2 |
Execution timed out |
8076 ms |
7672 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
1 ms |
384 KB |
Output is correct |
5 |
Correct |
3 ms |
512 KB |
Output is correct |
6 |
Correct |
96 ms |
1536 KB |
Output is correct |
7 |
Correct |
4673 ms |
12600 KB |
Output is correct |
8 |
Correct |
82 ms |
8440 KB |
Output is correct |
9 |
Execution timed out |
8067 ms |
12692 KB |
Time limit exceeded |
10 |
Halted |
0 ms |
0 KB |
- |