AtCoder Regular Contest 117 解题报告

本文同步发表在 YangTY’s Blog

ARC117A - God Sequence

傻逼构造

ARC117B - ARC Wrecker

题目。

注意到可以排序之后做，方便考虑。然后可以考虑做差分，不难发现直接把差分数组用乘法原理合并即可。

#include <cstdio>
#include <cctype>
#include <algorithm>
#include <functional>
#define FOR(i, a, b) for (int i = a; i <= b; ++i)
#define DEC(i, a, b) for (int i = a; i >= b; --i)

int read()
{
    int s = 0, x = 0;
    char c = getchar();
    while (!isdigit(c))
        x |= (c == '-'), c = getchar();
    while (isdigit(c))
        s = 10 * s + c - '0', c = getchar();
    return x ? -s : s;
}

const int mod = 1e9 + 7, maxn = 1e5 + 5;

int n, a[maxn], ans;

int main()
{
    n = read();
    FOR(i, 1, n) a[i] = read();
    std::sort(a + 1, a + n + 1);
    ans = 1;
    FOR(i, 1, n)
        ans = 1ll * ans * (a[i] - a[i - 1] + 1ll) % mod;
    printf("%d\n", ans);
    return 0;
}

ARC117C - Tricolor Pyramid

题目

比较妙的题。首先把颜色分别看成 $0,1,2$，那么上一层的方块就由 $-(a_1+a_2)\mathrm{m}\mathrm{o}\mathrm{d}3$ 可以得到。把所有的一直累加上去就是杨辉三角的形式：

图源 AtCoder 官方题解。

所以求二项式系数即可，使用 Lucas 定理。

#include <cstdio>
#include <cstring>
#define FOR(i, a, b) for (int i = a; i <= b; ++i)

const int maxn = 4e5 + 5, mod = 3;

char s[maxn];
int n, c[maxn];

int calc(char c)
{
    return c == 'B' ? 0 : (c == 'W' ? 1 : 2);
}

int C(int n, int m)
{
    if (m > n) return 0;
    int factn = 1;
    FOR(i, 1, n) factn *= i;
    FOR(i, 1, m) factn /= i;
    FOR(i, 1, n - m) factn /= i;
    return factn;
}

int lucas(int n, int m)
{
    if (m == 0) return 1;
    return C(n % mod, m % mod) * lucas(n / mod, m / mod) % mod;
}

int main()
{
    scanf("%d", &n);
    scanf("%s", s + 1);
    FOR(i, 1, n) c[i] = calc(s[i]);
    int ans = 0;
    FOR(i, 1, n)
        ans = (ans + c[i] * lucas(n - 1, i - 1)) % mod;
    if (n % 2 == 0) ans = (-ans + mod) % mod;
    printf("%c\n", ans == 0 ? 'B' : (ans == 1 ? 'W' : 'R'));
    return 0;
}

ARC117D - Miracle Tree

给一棵树的每个顶点标号，满足 $E_i\ge 1$，且 $|E_i-E_j|\le \mathrm{dist}(i,j)$。寻找一种最大标号最小的方案并输出。

相当好的思维题。

首先考虑随便标一个图。

图源 AtCoder 官方题解。

不难发现我们标的号应该为： { 1 , 2 , 3 , 5 , 8 , 9 } \lbrace1, 2, 3, 5, 8, 9\rbrace {1,2,3,5,8,9} 是最优的。相当于就是遇到一个点，首先走一条子链，然后回溯回来的时候仍然需要统计回溯链长，然后加到下一条子链的贡献里面去（因为要 $|E_i-E_j|\le \mathrm{dist}(i,j)$）。我们自然希望的是尽可能回溯的少，所以做的时候从直径的一端开始走，然后遇到分叉就优先走非直径的边。

#include <cstdio>
#include <cctype>
#define FOR(i, a, b) for (int i = a; i <= b; ++i)
#define DEC(i, a, b) for (int i = a; i >= b; --i)
#define GO(u) for (int i = head[u], v = to[i]; i; i = nxt[i], v = to[i])

int read()
{
    int s = 0, x = 0;
    char c = getchar();
    while (!isdigit(c))
        x |= (c == '-'), c = getchar();
    while (isdigit(c))
        s = 10 * s + c - '0', c = getchar();
    return x ? -s : s;
}

const int maxn = 200000 + 5;
int head[maxn], to[maxn << 1], nxt[maxn << 1], cnte;
int n, e[maxn];

void add(int u, int v)
{
    to[++cnte] = v;
    nxt[cnte] = head[u];
    head[u] = cnte;
    return;
}

int dis[maxn], p1, p2, vis[maxn];

void dfs1(int u, int fa, int &p)
{
    if (dis[u] > dis[p]) p = u;
    GO(u)
    {
        if (v == fa) continue;
        dis[v] = dis[u] + 1;
        dfs1(v, u, p);
    }
    return;
}

bool dfs2(int u, int fa)
{
    if (u == p2)
    {
        vis[u] = 1;
        return 1;
    }
    GO(u)
    {
        if (v == fa) continue;
        if (dfs2(v, u))
        {
            vis[u] = 1;
            return 1;
        }
    }
    return 0;
}

int cnt = 0;

void dfs3(int u, int fa)
{
    e[u] = ++cnt;
    GO(u)
    {
        if (vis[v] || v == fa) continue;
        dfs3(v, u);
    }
    GO(u)
    {
        if (v == fa) continue;
        if (vis[v])
        {
            dfs3(v, u);
            return;
        }
    }
    ++cnt;
}

int main()
{
    n = read();
    FOR(i, 1, n - 1)
    {
        int u = read(), v = read();
        add(u, v), add(v, u);
    }
    dfs1(1, 0, p1);
    dis[p1] = 0;
    dfs1(p1, 0, p2);
    dfs2(p1, 0);
    dfs3(p1, 0);
    FOR(i, 1, n) printf("%d ", e[i]);
    return 0;
}