oj.uz

Submission #927164

# Submission time Handle Problem Language Result Execution time Memory
927164 2024-02-14T10:48:40 Z vjudge1 Paths (BOI18_paths) C++17
70 / 100
 372 ms 61056 KB 
#include<bits/stdc++.h>

using namespace std;


const int N = (int)3e5+3;
const int MOD = (int)1e6+3;
const int B1 = 450;
const int B2 = 450;

int n, m, k, a[N], kol[6];
vector<int> g[N];
bool used[6];
long long ans;

void dfs(int v) {
	used[a[v]] = 1;
	ans++;
	for(auto to : g[v]) {
		if(!used[a[to]]) {
			dfs(to);
		}
	}
	used[a[v]] = 0;
}

int cnt[N][32];
int bb[N][6];
int oo[6][32];
int cc[6][6];

int main() {
    ios_base::sync_with_stdio(NULL);
    cin.tie(0);
	cin >> n >> m >> k;
	for(int i = 1; i <= n; ++i) {
		cin >> a[i];
		a[i]--;
	}
	for(int i = 1; i <= m; ++i) {
		int u, v;
		cin >> u >> v;
		if(a[u] == a[v]) {
			continue;
		}
		ans += 2;
		bb[u][a[v]]++;
		bb[v][a[u]]++;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	for(int i = 1; i <= n; ++i) {
		int cur = 0;
		memset(kol, 0, sizeof(kol));
		for(auto x : g[i]) {
			ans += 2*(cur-kol[a[x]]);
			cur++;	
			kol[a[x]]++;
		}
	}
	for(int i = 1; i <= n; ++i) {
		for(int mask = 0; mask < (1 << k); ++mask) {
			for(int j : g[i]) {
				if(!(mask >> a[j] & 1)) {
					continue;
				}		
				cnt[i][mask]++;
			}
		}
	}
	for(int i = 1; i <= n; ++i) {
		memset(oo, 0, sizeof(oo));
		memset(kol, 0, sizeof(kol));
		for(int j : g[i]) {
			kol[a[j]]++;
			for(int mask = 0; mask < (1 << k); ++mask) {
				oo[a[j]][mask] += cnt[j][mask];
			}
		}
		for(int c = 0; c < k; ++c) {
			if(c == a[i]) {
				continue;
			}
			for(int d = 0; d < k; ++d) {
				if(d == a[i] || d == c) {
					continue;
				}	
				int can = (1 << k)-1;
				can ^= (1 << a[i]);
		        can ^= (1 << c);
		        can ^= (1 << d);
		        ans += 1ll*oo[c][can]*kol[d];
			}
		}
	}

	for(int i = 1; i <= n; ++i) {
		memset(oo, 0, sizeof(oo));
		memset(kol, 0, sizeof(kol));
		memset(cc, 0, sizeof(cc));
		for(int j : g[i]) {
			kol[a[j]]++;
			for(int c = 0; c < k; ++c) {
				cc[a[j]][c] += bb[j][c];
			}
			for(int mask = 0; mask < (1 << k); ++mask) {
				oo[a[j]][mask] += cnt[j][mask];
			}
		}
		for(int c = 0; c < k; ++c) {
			if(c == a[i]) {
				continue;
			}
			for(int d = 0; d < k; ++d) {
				if(d == a[i] || d == c) {
					continue;
				}	
				for(int e = 0; e < k; ++e) {
					if(e == d || e == c || e == a[i]) {
						continue;
					}
					int can = (1 << k)-1;
					can ^= (1 << a[i]);
					can ^= (1 << c);
					can ^= (1 << d);
					can ^= (1 << e);
					ans += 1ll*cc[c][e]*kol[d]*oo[d][can];
				}
			}
		}
	}
	cout << ans;
    return 0;
}

Subtask #1
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11096 KB Output is correct
2 Correct 2 ms 10844 KB Output is correct
3 Correct 3 ms 10844 KB Output is correct
4 Correct 3 ms 10844 KB Output is correct
5 Correct 2 ms 8796 KB Output is correct
6 Correct 2 ms 10844 KB Output is correct
7 Correct 2 ms 10840 KB Output is correct
8 Correct 2 ms 10840 KB Output is correct
9 Correct 2 ms 10844 KB Output is correct
10 Correct 2 ms 10844 KB Output is correct

Subtask #2
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 57 ms 13660 KB Output is correct
2 Correct 62 ms 12924 KB Output is correct
3 Correct 323 ms 60216 KB Output is correct
4 Correct 73 ms 19616 KB Output is correct
5 Correct 42 ms 8796 KB Output is correct
6 Correct 233 ms 48360 KB Output is correct
7 Correct 319 ms 60236 KB Output is correct
8 Correct 313 ms 61056 KB Output is correct

Subtask #3
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11096 KB Output is correct
2 Correct 2 ms 10844 KB Output is correct
3 Correct 3 ms 10844 KB Output is correct
4 Correct 3 ms 10844 KB Output is correct
5 Correct 2 ms 8796 KB Output is correct
6 Correct 2 ms 10844 KB Output is correct
7 Correct 2 ms 10840 KB Output is correct
8 Correct 2 ms 10840 KB Output is correct
9 Correct 2 ms 10844 KB Output is correct
10 Correct 2 ms 10844 KB Output is correct
11 Correct 57 ms 13660 KB Output is correct
12 Correct 62 ms 12924 KB Output is correct
13 Correct 323 ms 60216 KB Output is correct
14 Correct 73 ms 19616 KB Output is correct
15 Correct 42 ms 8796 KB Output is correct
16 Correct 233 ms 48360 KB Output is correct
17 Correct 319 ms 60236 KB Output is correct
18 Correct 313 ms 61056 KB Output is correct
19 Correct 57 ms 13652 KB Output is correct
20 Correct 49 ms 12724 KB Output is correct
21 Correct 308 ms 60240 KB Output is correct
22 Correct 70 ms 19888 KB Output is correct
23 Correct 49 ms 8796 KB Output is correct
24 Correct 187 ms 48204 KB Output is correct
25 Correct 286 ms 60244 KB Output is correct
26 Correct 313 ms 60968 KB Output is correct
27 Correct 68 ms 14512 KB Output is correct
28 Correct 80 ms 15944 KB Output is correct
29 Correct 372 ms 60708 KB Output is correct
30 Correct 207 ms 39236 KB Output is correct
31 Correct 233 ms 39504 KB Output is correct
32 Correct 372 ms 60664 KB Output is correct
33 Correct 2 ms 10844 KB Output is correct
34 Correct 2 ms 10840 KB Output is correct
35 Correct 3 ms 10844 KB Output is correct
36 Correct 2 ms 10844 KB Output is correct
37 Correct 2 ms 8796 KB Output is correct
38 Correct 2 ms 10844 KB Output is correct
39 Correct 2 ms 10844 KB Output is correct
40 Correct 2 ms 10844 KB Output is correct
41 Correct 2 ms 10844 KB Output is correct
42 Correct 2 ms 10844 KB Output is correct

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 10840 KB Output is correct
2 Incorrect 38 ms 11864 KB Output isn't correct
3 Halted 0 ms 0 KB -