#include <bits/stdc++.h>
using namespace std;
const int mx = 3e5+5;
int n, m, k;
int a[mx];
long long b[6][mx];
long long ans = 0;
int 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];
}
ans += tmp * 2 + cnt;
}
cout << ans << "\n";
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
10844 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
36 ms |
10152 KB |
Output is correct |
2 |
Correct |
31 ms |
10696 KB |
Output is correct |
3 |
Correct |
67 ms |
15764 KB |
Output is correct |
4 |
Incorrect |
42 ms |
9784 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
10844 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
6488 KB |
Output is correct |
2 |
Incorrect |
11 ms |
13148 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |