105332024-04-04 17:46:41111Kutyavetélkedő 2cpp17Time limit exceeded 21/1001.282s123696 KiB
#include<bits/stdc++.h>
using namespace std;

#define int long long

void scc_dfs(const auto&g,auto&v,auto&s,int i){
	if(v[i]){
		return;
	}
	v[i]=1;
	for(int j:g[i]){
		scc_dfs(g,v,s,j);
	}
	s.push_back(i);
}

int scc(const auto&g,const auto&gg,auto&c){
	int n=g.size();
	vector<int>v,s;
	v.assign(n,0);
	for(int i=0;i<n;i++){
		scc_dfs(g,v,s,i);
	}
	v.assign(n,0);
	for(int i=n-1;i>=0;i--){
		if(v[s[i]]){
			continue;
		}
		c.emplace_back();
		scc_dfs(gg,v,c.back(),s[i]);
	}
	return c.size();
}

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int N,K;
	cin>>N>>K;
	vector<int>v(N);
	for(int i=0;i<N;i++){
		cin>>v[i];
	}
	int M;
	cin>>M;
	vector<vector<int>>g(K+1),gg(K+1);
	for(int i=0;i<M;i++){
		int a,b;
		cin>>a>>b;
		g[a].push_back(b);
		gg[b].push_back(a);
	}
	vector<vector<int>>c;
	int O=scc(g,gg,c);
	vector<int>cc(K+1);
	for(int i=0;i<O;i++){
		for(int j:c[i]){
			cc[j]=i;
		}
	}
	vector<vector<int>>h(O);
	for(int i=0;i<O;i++){
		for(int j:c[i]){
			for(int k:g[j]){
				int l=cc[k];
				if(l!=i){
					h[i].push_back(l);
				}
			}
		}
	}
	for(int i=0;i<O;i++){
		sort(h[i].begin(),h[i].end());
		h[i].erase(unique(h[i].begin(),h[i].end()),h[i].end());
	}
	for(int i=1;i<=K;i++){
		sort(g[i].begin(),g[i].end());
	}
	int ans=2;
	int last=cc[v[0]];
	for(int i=0;i+1<N;i++){
		if(binary_search(g[v[i]].begin(),g[v[i]].end(),v[i+1])){
			ans+=2;
			continue;
		}
		int next=cc[v[i+1]];
		vector<int>w(O);
		deque<int>q;
		w[last]=1;
		q.push_back(last);
		while(!q.empty()&&last!=next){
			int i=q.front();
			q.pop_front();
			for(int j:h[i]){
				if(w[j]){
					continue;
				}
				w[j]=1;
				q.push_back(j);
				if(j==next){
					last=j;
					break;
				}
			}
		}
		if(last!=next){
			break;
		}
		ans+=1;
	}
	cout<<ans<<'\n';
	return 0;
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted3ms1700 KiB
2Accepted3ms1876 KiB
subtask221/21
3Accepted3ms2240 KiB
4Accepted3ms2452 KiB
5Accepted3ms2408 KiB
6Accepted3ms2560 KiB
7Accepted3ms2876 KiB
8Accepted3ms2952 KiB
9Accepted3ms2836 KiB
subtask30/45
10Accepted287ms16268 KiB
11Accepted180ms16496 KiB
12Time limit exceeded1.268s16364 KiB
13Time limit exceeded1.259s16520 KiB
14Time limit exceeded1.268s16356 KiB
15Time limit exceeded1.271s16400 KiB
16Time limit exceeded1.256s18420 KiB
17Time limit exceeded1.282s20444 KiB
18Time limit exceeded1.279s21644 KiB
19Time limit exceeded1.264s21648 KiB
20Accepted1.187s40512 KiB
subtask40/34
21Time limit exceeded1.261s29688 KiB
22Time limit exceeded1.24s39004 KiB
23Time limit exceeded1.266s48648 KiB
24Accepted342ms123696 KiB
25Time limit exceeded1.266s48672 KiB
26Time limit exceeded1.269s38920 KiB