이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
using namespace std;
#define int long long
vector <int> color;
vector <vector <int>> g, dp;
signed main () {
int n, m, k;
cin >> n >> m >> k;
color.resize(n);
g.resize(n);
dp.resize(n, vector <int> (1 << k, 0));
for(int i = 0; i < n; i ++) {
cin >> color[i];
color[i] --;
}
for(int i = 0, u, v; i < m; i ++) {
cin >> u >> v;
u --, v --;
g[u].push_back(v);
g[v].push_back(u);
if(color[u] == color[v]) continue;
dp[v][(1 << color[u]) ^ (1 << color[v])] += 1;
dp[u][(1 << color[u]) ^ (1 << color[v])] += 1;
}
for(int mask = 0; mask < (1 << k); mask ++) {
for(int i = 0; i < n; i ++) {
if(dp[i][mask] == 0 || (!(mask & (1 << color[i])))) continue;
for(auto j : g[i]) {
if(mask & (1 << color[j])) continue;
dp[j][mask ^ (1 << color[j])] += dp[i][mask];
}
}
}
int ans = 0;
for(int mask = 0; mask < (1 << k); mask ++) {
for(int i = 0; i < n; i ++) {
ans += dp[i][mask];
}
}
cout << ans;
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |