Submission #1151297

#	Time	Username	Problem	Language	Result	Execution time	Memory
1151297		vladilius	Star Trek (CEOI20_startrek)	C++20	Compilation error	0 ms	0 KiB

startrek

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int, int>;
#define pb push_back
#define ff first
#define ss second
const int mod = 1e9 + 7;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    
    int n; ll d; cin>>n>>d;
    vector<int> g[n + 1];
    for (int i = 1; i < n; i++){
        int u, v; cin>>u>>v;
        g[u].pb(v);
        g[v].pb(u);
    }
    
    vector<vector<int>> p(n + 1, vector<int>(n + 1)), h(n + 1, vector<int>(n + 1));
    function<void(int, int, int&)> fill = [&](int v, int pr, int& u){
        p[u][v] = pr; h[u][v] = 1;
        for (int i: g[v]){
            if (i == pr) continue;
            fill(i, v, u);
            h[u][v] = min(h[u][v], h[u][i]);
        }
        h[u][v] = !h[u][v];
    };
    
    for (int i = 1; i <= n; i++){
        fill(i, 0, i);
    }
    
    bool f[n + 1][n + 1][2];
    for (int i = 1; i <= n; i++){
        for (int j = 1; j <= n; j++){
            f[i][j][0] = f[i][j][1] = 0;
        }
    }
    function<void(int, int&)> dfs = [&](int v, int& u){
        int s = 0, x = 0;
        for (int i: g[v]){
            if (i == p[u][v]) continue;
            if (!h[u][i]){
                s++;
                x = i;
            }
        }
        for (int i: g[v]){
            if (i == p[u][v]) continue;
            f[u][i][0] = f[u][v][1];
            if (!s || (s == 1 && i == x)){
                f[u][i][1] = f[u][v][0];
            }
            else {
                f[u][i][1] = f[u][v][1];
            }
            dfs(i, u);
        }
    };
    
    for (int i = 1; i <= n; i++){
        f[i][i][0] = 0; f[i][i][1] = 1;
        dfs(i, i);
    }

    vector<vector<int>> dp(n + 1, vector<int>(2));
    vector<int> c1(n + 1), c0(n + 1);
    for (int i = 1; i <= n; i++){
        dp[i][h[i][i]] = 1;
        for (int j = 1; j <= n; j++){
            if (f[i][j][1]){
                c1[i]++;
            }
            else {
                c0[i]++;
            }
        }
    }
    for (int i = 1; i <= d; i++){
        vector<vector<int>> dp1(n + 1, vector<int>(2));
        int s1 = 0, s0 = 0;
        for (int x = 1; x <= n; x++){
            s1 = (s1 + dp[x][1]) % mod;
            s0 = (s0 + dp[x][0]) % mod;
        }
        for (int v = 1; v <= n; v++){
            dp1[v][h[v][v]] = (dp1[v][h[v][v]] + 1LL * s1 * n) % mod;
            dp1[v][0] = (dp1[v][0] + 1LL * c0[v] * s0) % mod;
            dp1[v][1] = (dp1[v][1] + 1LL * c1[v] * s0) % mod;
        }
        dp = dp1;
    }
    cout<<dp[1][1]<<"\n";
}

Compilation message (stderr)

startrek.cpp: In lambda function:
startrek.cpp:55:13: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   55 |             f[u][i][0] = f[u][v][1];
      |             ^
startrek.cpp:55:13: note: because the array element type 'bool [(n + 1)][2]' has variable size
startrek.cpp:55:26: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   55 |             f[u][i][0] = f[u][v][1];
      |                          ^
startrek.cpp:55:26: note: because the array element type 'bool [(n + 1)][2]' has variable size
startrek.cpp:57:17: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   57 |                 f[u][i][1] = f[u][v][0];
      |                 ^
startrek.cpp:57:17: note: because the array element type 'bool [(n + 1)][2]' has variable size
startrek.cpp:57:30: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   57 |                 f[u][i][1] = f[u][v][0];
      |                              ^
startrek.cpp:57:30: note: because the array element type 'bool [(n + 1)][2]' has variable size
startrek.cpp:60:17: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   60 |                 f[u][i][1] = f[u][v][1];
      |                 ^
startrek.cpp:60:17: note: because the array element type 'bool [(n + 1)][2]' has variable size
startrek.cpp:60:30: sorry, unimplemented: capture of variably-modified type 'bool [(n + 1)][(n + 1)][2]' that is not an N3639 array of runtime bound
   60 |                 f[u][i][1] = f[u][v][1];
      |                              ^
startrek.cpp:60:30: note: because the array element type 'bool [(n + 1)][2]' has variable size