Submission #1303894

#TimeUsernameProblemLanguageResultExecution timeMemory
1303894NonozeRailway (BOI17_railway)C++20
100 / 100
153 ms39980 KiB
/* * Author: Nonoze * Created: Saturday 20/12/2025 */ #include <bits/stdc++.h> using namespace std; #ifndef DEBUG #define dbg(...) #endif // #define cout cerr << "OUT: " #define endl '\n' #define endlfl '\n' << flush #define quit(x) return (void)(cout << x << endl) template<typename T> void read(T& x) { cin >> x; } template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second); } template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); } template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); } template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); } template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); } template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; } #define sz(x) (int)(x.size()) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define make_unique(v) sort(all(v)), v.erase(unique(all(v)), (v).end()) #define pb push_back #define mp(a, b) make_pair(a, b) #define fi first #define se second #define cmin(a, b) a = min(a, b) #define cmax(a, b) a = max(a, b) #define YES cout << "YES" << endl #define NO cout << "NO" << endl #define QYES quit("YES") #define QNO quit("NO") #define int long long #define double long double const int inf = numeric_limits<int>::max() / 4; mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); const int MOD = 1e9+7, LOG=20; void solve(); signed main() { ios::sync_with_stdio(0); cin.tie(0); int tt=1; // cin >> tt; while(tt--) solve(); return 0; } int n, k, m, q; vector<int> cnt, nb, par; vector<vector<pair<int, int>>> adj; vector<vector<int>> empl; map<int, int> dfs(int u, int p) { vector<map<int, int>> children; for (auto [v, id] : adj[u]) { if (v == p) continue; par[v]=u; children.pb(dfs(v, u)); } sort(rall(children), [](auto& a, auto& b) { return sz(a) < sz(b); }); map<int, int> curr; if (sz(children)) curr = move(children[0]); for (auto &x: empl[u]) { curr[x]++; if (curr[x] >= nb[x]) curr.erase(x); } for (int i=1; i<sz(children); i++) { for (auto &[key, val] : children[i]) { curr[key] += val; if (curr[key] >= nb[key]) curr.erase(key); } } cnt[u]=sz(curr); return curr; } void solve() { read(n, m, k); adj.assign(n, {}), cnt.assign(n, 0), empl.assign(n, {}), par.assign(n, -1); for (int i=0; i<n-1; i++) { int u, v, w; read(u, v); u--, v--; adj[u].pb({v, i}); adj[v].pb({u, i}); } for (int i=0; i<m; i++) { int sz; read(sz); nb.pb(sz); while (sz--) { int x; read(x); x--; empl[x].pb(i); } } dfs(0, -1); vector<int> ans; for (int u=0; u<n; u++) { for (auto &[v, id]: adj[u]) { if (cnt[v]>=k && par[v]==u) { ans.pb(id+1); } } } cout << sz(ans) << endl; sort(all(ans)); print(ans); }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...