Compile Error
foo.cc: In function 'void shelby()': foo.cc:34:20: error: use of 'auto' in lambda parameter declaration only available with '-std=c++14' or '-std=gnu++14' 34 | auto dfs = [&](auto &&self, int u, int par)-> ordered_set<int> { | ^~~~ foo.cc: In lambda function: foo.cc:40:38: error: expression cannot be used as a function 40 | ordered_set<int> o = self(self, v, u); | ~~~~^~~~~~~~~~~~ foo.cc: In function 'void shelby()': foo.cc:48:8: error: no match for call to '(shelby()::<lambda(int&&, int, int)>) (shelby()::<lambda(int&&, int, int)>&, int, int)' 48 | dfs(dfs, 1, -1); | ~~~^~~~~~~~~~~~ foo.cc:34:16: note: candidate: 'shelby()::<lambda(int&&, int, int)>' 34 | auto dfs = [&](auto &&self, int u, int par)-> ordered_set<int> { | ^ foo.cc:34:16: note: no known conversion for argument 1 from 'shelby()::<lambda(int&&, int, int)>' to 'int&&'
Code
#pragma GCC target ("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#ifndef LOCAL
#include <bits/stdc++.h>
#define debug(...)
#endif
using namespace std;
#define cinv(v) for (auto &it:v) cin>>it;
#define coutv(v) for (auto &it:v) cout<< it<<' '; cout<<'\n';
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
template<typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
const int N = 1e5 + 5;
int arr[N], ans[N], sz[N];
vector<int> adj[N];
void shelby() {
int n;
cin >> n;
for (int i = 1; i <= n; ++i) cin >> arr[i], adj[i].clear();
for (int i = 1; i < n; ++i) {
int u, v;
cin >> u >> v;
adj[u].push_back(v);
adj[v].push_back(u);
}
auto dfs = [&](auto &&self, int u, int par)-> ordered_set<int> {
ordered_set<int> ret;
ret.insert(arr[u]);
sz[u] = 1;
for (auto &v: adj[u]) {
if (v == par) continue;
ordered_set<int> o = self(self, v, u);
if (o.size() > ret.size()) swap(ret, o);
for (auto &it: o) ret.insert(it);
sz[u] += sz[v];
}
ans[u] = sz[u] - (ret.order_of_key(sz[u] + 1));
return ret;
};
dfs(dfs, 1, -1);
int q;
cin >> q;
while (q--) {
int x;
cin >> x;
cout << ans[x] << '\n';
}
}
signed main() {
cin.tie(0)->sync_with_stdio(0);
int t = 1;
cin >> t;
for (int _ = 1; _ <= t; ++_) {
// cout << "Case " << _ << ": ";
shelby();
}
}
Information
- Submit By
- Type
- Submission
- Problem
- P1157 Roy and Tree Permutation
- Language
- C++11 (G++ 13.2.0)
- Submit At
- 2025-01-12 20:35:25
- Judged At
- 2025-01-12 20:35:25
- Judged By
- Score
- 0
- Total Time
- 0ms
- Peak Memory
- 0 Bytes