Submission #1009407

#	Time	Username	Problem	Language	Result	Execution time	Memory
1009407		alexdd	Paths (BOI18_paths)	C++17	100 / 100	403 ms	141396 KiB

paths

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int color[300005];
long long dp[5][32][300005];
vector<int> con[300005];
signed main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>color[i];
        color[i]--;
        dp[0][(1<<color[i])][i]=1;
    }
    int u,v;
    for(int i=1;i<=m;i++)
    {
        cin>>u>>v;
        con[u].push_back(v);
        con[v].push_back(u);
    }
    long long rez=0;
    for(int pas=1;pas<k;pas++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int mask=(1<<color[i]);mask<(1<<k);mask++)
            {
                if(!(mask&(1<<color[i])))
                    continue;
                for(auto x:con[i])
                    dp[pas][mask][i] += dp[pas-1][mask-(1<<color[i])][x];
                rez += dp[pas][mask][i];
            }
        }
    }
    cout<<rez;
    return 0;
}

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