Submission #864394

#	Time	Username	Problem	Language	Result	Execution time	Memory
864394		iskhakkutbilim	Paths (BOI18_paths)	C++17	23 / 100	3071 ms	17748 KiB

paths

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define ff first
#define ss second
#define all(a) a.begin(), a.end()
const int N = 3e5;
int n, m, col[N+1];
int k;
vector<int> g[N+1];
int ans;

int used[N+10], used_c[N+10];

void dfs(int v){
	used[v] = 1, used_c[col[v]] = 1;
	ans++;
	for(int to : g[v]){
		if(used[to] == 0 && used_c[col[to]] == 0){
			dfs(to);
		}	
	}
	used[v] = 0, used_c[col[v]] = 0;
}

main(){
   ios::sync_with_stdio(0);
   cin.tie(0); cout.tie(0);
	cin >> n >> m >> k;
	for(int i = 1;i <= n; i++){
		cin >> col[i];
	}
	for(int i = 0;i < m; i++){
		int a, b; cin >> a >> b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
	
	for(int i = 1;i <= n; i++){
		dfs(i);
		ans--;
		for(int j = 1;j <= n; j++){
			used[j] = 0;
		}
		for(int j = 0;j <= k; j++) used_c[j] = 0;
	}
	
	cout << ans;
	return 0;
}

Compilation message (stderr)

paths.cpp:27:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   27 | main(){
      | ^~~~

• Subtask 1: 23 / 23
• Subtask 2: 0 / 20
• Subtask 3: 0 / 27
• Subtask 4: 0 / 30