题意就是两个棋子都放在树根1,两个棋子间距不超过d的情况下走完各自必须要走到的结点然后回到树根,求最短距离
首先贪心地做,一个结点和另一个结点距离尽可能的大的完成所有任务,也就是让结点走最少的路完成任务,所以要先求出这棵树的DFS序,找到a或b中某点的第d层祖先,表示这个点一定要走,放在对方的必走路线里,最后就可以从根dfs整棵树,分别计算a和b走了哪些结点,这里因为有来回所以如果经过了某点必须路径+2,最后得出的答案减去4,因为在最后遍历的时候a和b都走了树根1两遍,这是不存在的距离
AC代码:
#include <bits/stdc++.h> using namespace std; using LL = long long; int st[200010]; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int n, d; cin >> n >> d; vector<vector<int>> G(n + 1); for (int i = 0; i < n - 1; i++) { int u, v; cin >> u >> v; G[u].push_back(v); G[v].push_back(u); } int ma, mb; cin >> ma; vector<bool> visa(n + 1), visb(n + 1); for (int i = 0; i < ma; i++) { int z; cin >> z; visa[z] = true; } cin >> mb; for (int i = 0; i < mb; i++) { int z; cin >> z; visb[z] = true; } int ans = 0, tot = 0; function<void(int, int)> dfs1 = [&](int u, int fa) { st[++tot] = u; if (visa[u] and tot > d) { visb[st[tot - d]] = true; } if (visb[u] and tot > d) { visa[st[tot - d]] = true; } for (auto v : G[u]) { if (v != fa) { dfs1(v, u); } } tot--; }; function<void(int, int)> dfs2 = [&](int u, int fa) { for (auto v : G[u]) { if (v != fa) { dfs2(v, u); if (visa[v]) { visa[u] = true; } if (visb[v]) { visb[u] = true; } } } if (visa[u]) { ans += 2; } if (visb[u]) { ans += 2; } }; dfs1(1, 0); dfs2(1, 0); cout << ans - 4 << '\n'; return 0; }
根据题意,需要构造连续k个数之间染不同的颜色,所以需要循环染(n+k-1)/k个区间,可知当a[i]>(n+k-1)/k,一定存在某个循环中出现两个一样的颜色,当a[i]<(n+k-1)/k,一定存在某个区间不存在这个颜色,当a[i]=(n+k-1)/k,所有区间都存在这个颜色,所以要进行讨论,当a[i]=(n+k-1)/k的个数小于等于最后一个区间的长度的时候是符合题意的,如果最后一个区间的长度为0,即k整除n,a[i]=(n+k-1)/k的个数应小于等于k才符合题意,否则一定存在区间内有至少两个的颜色相同
AC代码:
#include <bits/stdc++.h> using namespace std; using LL = long long; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int T; cin >> T; while (T--) { int n, m, k; cin >> n >> m >> k; vector<int> a(m); bool ok = true; int cnt = 0; for (int i = 0; i < m; i++) { cin >> a[i]; if (a[i] > (n + k - 1) / k) { ok = false; } if (a[i] == (n + k - 1) / k) { cnt++; } } if (ok) { if (cnt <= n % k) { cout << "YES\n"; } else if (n % k == 0 and cnt <= k) { cout << "YES\n"; } else { cout << "NO\n"; } } else { cout << "NO\n"; } } return 0; }
Polynomial Round 2022 (Div. 1 + Div. 2, Rated, Prizes!)(E、B)
于 2022-12-20 17:11:35 首次发布