#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include using namespace std; using Int = long long; template ostream &operator<<(ostream &os, const pair &a) { return os << "(" << a.first << ", " << a.second << ")"; }; template ostream &operator<<(ostream &os, const vector &as) { const int sz = as.size(); os << "["; for (int i = 0; i < sz; ++i) { if (i >= 256) { os << ", ..."; break; } if (i > 0) { os << ", "; } os << as[i]; } return os << "]"; } template void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; } template bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; } template bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; } #define COLOR(s) ("\x1b[" s "m") struct Functional { int n; vector par; int cyclesLen; vector lens; vector> cycles; // cycle id or -1 vector on; // forest vector> graph; int zeit; vector dis, fin, dep; // root is cycle[k][l] vector ks, ls; Functional() {} Functional(const vector &par_) : par(par_) { n = par.size(); for (int u = 0; u < n; ++u) { assert(0 <= par[u]); assert(par[u] < n); } cycles.clear(); vector vis(n, -1); for (int s = 0; s < n; ++s) { int u = s; for (; !~vis[u]; u = par[u]) { vis[u] = s; } if (vis[u] == s) { vector cycle; for (int v = u; ; ) { cycle.push_back(v); if ((v = par[v]) == u) break; } cycles.push_back(cycle); } } cyclesLen = cycles.size(); lens.resize(cyclesLen); on.assign(n, -1); for (int k = 0; k < cyclesLen; ++k) { lens[k] = cycles[k].size(); for (const int u : cycles[k]) { on[u] = k; } } graph.assign(n, {}); for (int u = 0; u < n; ++u) if (!~on[u]) { graph[par[u]].push_back(u); } zeit = 0; dis.assign(n, -1); fin.assign(n, -1); dep.assign(n, 0); ks.assign(n, -1); ls.assign(n, -1); for (int k = 0; k < cyclesLen; ++k) { for (int l = 0; l < lens[k]; ++l) { dfs(k, l, cycles[k][l]); } } } void dfs(int k, int l, int u) { dis[u] = zeit++; ks[u] = k; ls[u] = l; for (const int v : graph[u]) { dep[v] = dep[u] + 1; dfs(k, l, v); } fin[u] = zeit; } // min d s.t. f^d(u) = v, or -1 int dist(int u, int v) const { if (ks[u] != ks[v]) return -1; if (~on[v]) { int dl = ls[v] - ls[u]; if (dl < 0) dl += lens[ks[u]]; return dep[u] + dl; } return (dis[v] <= dis[u] && dis[u] < fin[v]) ? (dep[u] - dep[v]) : -1; }; }; //////////////////////////////////////////////////////////////////////////////// // \sum[u] f[u] + \sum[u] (f[u] xor f[P[u]] xor A[u]) = K int N, K; vector P; vector A; Functional F; using BS = bitset<2010>; BS dp[1010][2]; BS mul(BS fs, BS gs) { BS hs; for (int k = fs._Find_first(); k < (int)fs.size(); k = fs._Find_next(k)) { hs |= gs << k; } return hs; } void dfs(int u) { for (int x = 0; x < 2; ++x) { dp[u][x].reset(); dp[u][x].set(x); } for (const int v : F.graph[u]) { dfs(v); for (int x = 0; x < 2; ++x) { BS tmp; for (int y = 0; y < 2; ++y) { tmp |= mul(dp[u][x], dp[v][y] << (x ^ y ^ A[v])); } dp[u][x] = tmp; } } // for(int x=0;x<2;++x){cerr<<"dp["<