#include <bits/stdc++.h> using namespace std; template <typename T> using Grid = array<vector<T>, 2>; struct TestCase { size_t N, K; Grid<size_t> rows; }; TestCase read_test_case(); void solve_test_case(const TestCase&); int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); solve_test_case(read_test_case()); } TestCase read_test_case() { TestCase tc; cin >> tc.N >> tc.K; for (size_t i = 0; i < 2; i++) { tc.rows[i].resize(tc.N); for (auto& v : tc.rows[i]) { cin >> v; v--; } } return tc; } void solve_small_n(const TestCase& tc); void solve_test_case(const TestCase& tc) { solve_small_n(tc); } size_t count_areas(const TestCase& tc, size_t low, size_t high); void solve_small_n(const TestCase& tc) { vector<size_t> cnt(tc.K + 1); for (size_t high = 0; high < tc.N * 2; high++) for (size_t low = 0; low <= high; low++) { size_t areas = count_areas(tc, low, high); if (areas <= tc.K) cnt[areas]++; } for (size_t i = 1; i <= tc.K; i++) cout << cnt[i] << " "; cout << "\n"; } void visit_dfs( const TestCase& tc, Grid<bool>& visited, size_t low, size_t high, size_t i, size_t j); size_t count_areas(const TestCase& tc, size_t low, size_t high) { size_t cnt = 0; Grid<bool> visited; visited[0].resize(tc.N); visited[1].resize(tc.N); for (size_t i = 0; i < 2; i++) for (size_t j = 0; j < tc.N; j++) if (!visited[i][j] && low <= tc.rows[i][j] && tc.rows[i][j] <= high) { visit_dfs(tc, visited, low, high, i, j); cnt++; } return cnt; } void visit_dfs( const TestCase& tc, Grid<bool>& visited, size_t low, size_t high, size_t i, size_t j) { if (visited[i][j]) return; if (tc.rows[i][j] < low || tc.rows[i][j] > high) return; visited[i][j] = true; visit_dfs(tc, visited, low, high, (i + 1) % 2, j); visit_dfs(tc, visited, low, high, i, (j + 1) % tc.N); visit_dfs(tc, visited, low, high, i, (j + tc.N - 1) % tc.N); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | #include <bits/stdc++.h> using namespace std; template <typename T> using Grid = array<vector<T>, 2>; struct TestCase { size_t N, K; Grid<size_t> rows; }; TestCase read_test_case(); void solve_test_case(const TestCase&); int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); solve_test_case(read_test_case()); } TestCase read_test_case() { TestCase tc; cin >> tc.N >> tc.K; for (size_t i = 0; i < 2; i++) { tc.rows[i].resize(tc.N); for (auto& v : tc.rows[i]) { cin >> v; v--; } } return tc; } void solve_small_n(const TestCase& tc); void solve_test_case(const TestCase& tc) { solve_small_n(tc); } size_t count_areas(const TestCase& tc, size_t low, size_t high); void solve_small_n(const TestCase& tc) { vector<size_t> cnt(tc.K + 1); for (size_t high = 0; high < tc.N * 2; high++) for (size_t low = 0; low <= high; low++) { size_t areas = count_areas(tc, low, high); if (areas <= tc.K) cnt[areas]++; } for (size_t i = 1; i <= tc.K; i++) cout << cnt[i] << " "; cout << "\n"; } void visit_dfs( const TestCase& tc, Grid<bool>& visited, size_t low, size_t high, size_t i, size_t j); size_t count_areas(const TestCase& tc, size_t low, size_t high) { size_t cnt = 0; Grid<bool> visited; visited[0].resize(tc.N); visited[1].resize(tc.N); for (size_t i = 0; i < 2; i++) for (size_t j = 0; j < tc.N; j++) if (!visited[i][j] && low <= tc.rows[i][j] && tc.rows[i][j] <= high) { visit_dfs(tc, visited, low, high, i, j); cnt++; } return cnt; } void visit_dfs( const TestCase& tc, Grid<bool>& visited, size_t low, size_t high, size_t i, size_t j) { if (visited[i][j]) return; if (tc.rows[i][j] < low || tc.rows[i][j] > high) return; visited[i][j] = true; visit_dfs(tc, visited, low, high, (i + 1) % 2, j); visit_dfs(tc, visited, low, high, i, (j + 1) % tc.N); visit_dfs(tc, visited, low, high, i, (j + tc.N - 1) % tc.N); } |