Submission #951747

#	Time	Username	Problem	Language	Result	Execution time	Memory
951747		mocha	Paths (BOI18_paths)	C++14	20 / 100	77 ms	16920 KiB

paths

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mx = 3e5+5;

int n, m, k;
int a[mx];
long long b[6][mx];
long long ans = 0;

signed main() {
	cin.tie(0);ios::sync_with_stdio(0);

	cin >> n >> m >> k;
	for (int i=1;i<=n;i++) {
		cin >> a[i];
	}
	for (int i=1;i<=m;i++) {
		int u, v;
		cin >> u >> v;
		b[a[v]][u]++;
		b[a[u]][v]++;
	}
	for (int i=1;i<=n;i++) {
		long long tmp = 1;
		long long cnt = 0;
		for (int j=1;j<=k;j++) {
			if (a[i] == j) continue;
			tmp *= b[j][i];
			cnt += b[j][i];
		}
		if (k==1 or k==2) tmp = 0;
		ans += tmp * 2 + cnt;
	}
	cout << ans << "\n";
}

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