제출 #716553

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
716553		AranMaster	Paths (BOI18_paths)	C++17	53 / 100	227 ms	31436 KiB

paths

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
vector <int> adj[maxn];
int dp[maxn][32];
int c[maxn];
int n, m , k;
int main(){
    cin >> n >> m >> k;
    for(int i = 1; i <= n; i++){
        cin >> c[i];
        c[i]--;
        dp[i][1 << c[i]] = 1;
    }

    for(int i = 1; i <= m; i++){
        int a, b; 
        cin >> a >> b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    long long ans = 0;
    for(int mask = 1; mask < (1 << k); mask++){
        for(int i = 1; i <= n; i++){
            if(mask & (1 << c[i])){
                for(int u : adj[i]){
                    dp[i][mask] += dp[u][mask ^ (1 << c[i])];
                }
                ans += dp[i][mask];
            }
        }
    }

    cout << ans - n << endl;
}

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