Submission #716439

#	Time	Username	Problem	Language	Result	Execution time	Memory
716439		Iliya	Paths (BOI18_paths)	C++17	100 / 100	585 ms	95940 KiB

paths

#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef long long ll;
const int N = 3e5 + 10;
vector<int> Adj[N];
int n, m, k, ans, C[N], dp[N][1 << 5];
signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0), cout.tie(0);
	cin >> n >> m >> k;
	for (int i = 1; i <= n; i++) {
		cin >> C[i]; C[i]--;
		dp[i][(1 << C[i])] = 1;
	}
	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		Adj[u].push_back(v);
		Adj[v].push_back(u);
	}
	for (int mask = 1; mask < (1 << 5); mask++) {
		if (__builtin_popcount(mask) != 1) {
			for (int i = 1; i <= n; i++) {
				if (mask & (1 << C[i])) {
					for (int u : Adj[i]) {
						dp[i][mask] += dp[u][mask ^ (1 << C[i])];
					}
					ans += dp[i][mask];
				}
			}
		}
	}
	cout << ans << '\n';
}

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