제출 #1324099

#제출 시각아이디문제언어결과실행 시간메모리
1324099edoPaths (BOI18_paths)C++20
100 / 100
556 ms92764 KiB
#include <bits/stdc++.h>

using namespace std;
using ll = long long;

const int nax = 3e5 + 2;
ll dp[nax][32];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m, k;
    cin >> n >> m >> k;

    int c[n + 1];
    for(int i = 0; i < n; i++) {
        cin >> c[i], c[i]--;
        dp[i][1 << c[i]] = 1;
    }

    vector<vector<int>> g(n);
    for(int i = 0, u, v; i < m; i++) {
        cin >> u >> v, --v, --u;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    for(int j = 0; j < 32; j++) {
        for(int i = 0; i < n; i++) {
            for(int q : g[i]) {
                if(!(j >> c[q] & 1))
                    dp[q][j ^ (1 << c[q])] += dp[i][j];
            }
        }
    }

    ll ans = 0;
    
    for(int i = 0; i < n; i++)
        for(int j = 0; j < 32; j++)
            if(j & (j - 1)) ans += dp[i][j];

    cout << ans << "\n";

    return 0;
}

#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...