Polynomial Round 2022 (Div. 1 + Div. 2, Rated, Prizes!)(E、B)

E. Two Chess Pieces

题意就是两个棋子都放在树根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;
}

B. Coloring

根据题意，需要构造连续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;
}