이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll dp[300005][1<<5], val[300005];
int main() {
int n, m, k, a, b;
cin >> n >> m >> k;
for(int i=0; i<n; i++) cin >> val[i], val[i]--;
vector<int> graph[n];
for(int i=0; i<m; i++) {
cin >> a >> b;
graph[a-1].push_back(b-1);
graph[b-1].push_back(a-1);
}
for(int i=0; i<n; i++) dp[i][1<<val[i]] = 1;
for(int s=0; s<(1<<k); s++) {
for(int i=0; i<n; i++) {
if((s & (1 << val[i])) == 0) continue;
for(int &j : graph[i]) {
if((s & (1 << val[j])) == 0 || val[j] == val[i]) continue;
dp[i][s] += dp[j][s^(1<<val[i])];
}
}
}
ll ans = 0;
for(int i=0; i<n; i++)
for(int j=0; j<(1<<k); j++)
if(__builtin_popcount(j) > 1) ans += dp[i][j];
cout << ans << '\n';
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... |