Submission #1324970

#TimeUsernameProblemLanguageResultExecution timeMemory
1324970annnRailway (BOI17_railway)C++20
36 / 100
74 ms32148 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long
#define endl "\n"
#define pb push_back
#define ff first
#define ss second
#define ii pair<int, int>
#define vi vector<int>
#define vii vector<pair<int, int>>
#define yes cout << "YES\n"
#define no cout << "NO\n"
#define mii map<int, int>
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define all(a, len) (a) + 1, (a) + len + 1
#define vall(a) (a).begin(), a.end()
const int INF = 4e18;
const int MOD = 1e9 + 7;

int n, m, k; const int mx = 1e5 + 3, LG = 17;
vector<ii> adj[mx];
int d[mx], ans[mx];
struct LCA {
	int par[mx], h[mx], id_par[mx];
	int up[mx][LG];
	void dfs(int u) {
		for (auto [v, id]: adj[u]) {
			if (v == par[u]) continue;
			h[v] = h[u] + 1;
			par[v] = u;
			id_par[v] = id;
			dfs(v);
		}
	};
	void preprocess() {
		dfs(1);
		for (int u = 1; u <= n; u++) up[u][0] = par[u];
		for (int j = 1; j < LG; j++) {
			for (int u = 1; u <= n; u++) {
				up[u][j] = up[up[u][j - 1]][j - 1];
			}
		}
	};
	int ancestor_k(int u, int k) {
		for (int j = 0; (1 << j) <= k; j++) {
			if ((k >> j & 1)) u = up[u][j];
		}
		return u;
	};
	int lca(int u, int v) {
		if (h[u] != h[v]) {
			if (h[u] < h[v]) swap(u, v);
			int k = h[u] - h[v];
			u = ancestor_k(u, k);
		}

		if (u == v) return u;
		
		int k = __lg(h[u]);
		for (int j = k; j >= 0; j--) {
			if (up[u][j] != up[v][j]) {
				u = up[u][j]; v = up[v][j];
			}
		}
		return up[u][0];
	};
} lca;

void dfs(int u, int p) {
	ans[u] = d[u];
	for (auto [v, id]: adj[u]) if (v != p) {
		dfs(v, u);
		ans[u] += ans[v];
	}
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);

    cin >> n >> m >> k;
    rep(i, 1, n-1) {
    	int u, v; cin >> u >> v;
    	adj[u].pb({v, i}); adj[v].pb({u, i});
    }
    lca.preprocess();
    
    while (m--) {
    	int s; cin >> s;
    	int u; cin >> u;
    	rep(i, 1, s-1) {
	    	int v; cin >> v;
	    	d[u]++; d[v]++;
	    	d[lca.lca(u, v)] -= 2;
	    	u = v;
    	}
    }
    dfs(1, 0);
    vi res;
    
    rep(i, 1, n) {
    	if (ans[i] >= k) res.pb(lca.id_par[i]);
    }
    sort(vall(res));
    cout << res.size() << endl;
    for (int u: res) cout << u << ' ';
    
    return 0;

}
#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...