本文同步发表在 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 0,1,2 0,1,2,那么上一层的方块就由 − ( a 1 + a 2 ) m o d 3 -(a_1 + a_2) \bmod 3 −(a1+a2)mod3 可以得到。把所有的一直累加上去就是杨辉三角的形式:
图源 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 ≥ 1 E_i \ge 1 Ei≥1,且 ∣ E i − E j ∣ ≤ dist ( i , j ) |E_i - E_j|\le \operatorname{dist}(i, j) ∣Ei−Ej∣≤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 ∣ ≤ dist ( i , j ) |E_i - E_j|\le \operatorname{dist}(i, j) ∣Ei−Ej∣≤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;
}