oj.uz

Submission #1114292

# Submission time Handle Problem Language Result Execution time Memory
1114292 2024-11-18T13:35:16 Z Dan4Life Synchronization (JOI13_synchronization) C++17
0 / 100
 108 ms 34760 KB 
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)
using ar2 = array<int,2>;
const int N = (int)2e5+10;
int n, m, q;
vector<int> adj[N];
vector<ar2> edges;
set<int> S[N];
int tim[N], dfs_timer;
int e[N], r[N], active[N];

void dfs(int s, int p){
	tim[s] = dfs_timer++;
	for(auto u : adj[s]){
		if(u==p) continue;
		dfs(u,s);
	}
}

int dfs2(int s, int p, int curT){
	int ans = 1;
	for(auto u : adj[s]){
		if(u==p) continue;
		auto itr = S[u].upper_bound(curT);
		if(itr==begin(S[u])) continue;
		itr--; ans+=dfs2(u,s,*itr);
	}
	return ans;
}

int32_t main(){
	ios_base::sync_with_stdio(false); cin.tie(0);
	cin >> n >> m >> q;
	for(int i = 1; i < n; i++){
		int a, b; cin >> a >> b;
		adj[a].pb(b); adj[b].pb(a);
		edges.pb({a,b});
	}
	for(int i = 1; i <= m; i++) cin >> e[i],e[i]--;
	for(int i = 1; i <= q; i++) cin >> r[i];
	dfs(r[1],0); S[r[1]].insert(m+1);
	for(int i = 1; i <= m; i++){
		auto [a,b] = edges[e[i]];
		if(tim[a]<tim[b])swap(a,b);
		active[e[i]]^=1;
		if(active[e[i]]) S[a].insert(i);
		else S[a].insert(i-1);
	}
	for(int i = 0; i < n-1; i++){
		auto [a,b] = edges[e[i]];
		if(tim[a]<tim[b])swap(a,b);
		if(active[e[i]]) S[a].insert(m);
	}
	cout << dfs2(r[1],-1,*prev(end(S[r[1]]))) << "\n";
}

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 15440 KB Output is correct
2 Correct 6 ms 15608 KB Output is correct
3 Incorrect 6 ms 15440 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 81 ms 34760 KB Output is correct
2 Correct 68 ms 34492 KB Output is correct
3 Incorrect 45 ms 34236 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 6 ms 15440 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 108 ms 33980 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 40.0

# Verdict Execution time Memory Grader output
1 Incorrect 5 ms 15440 KB Output isn't correct
2 Halted 0 ms 0 KB -