[题解]CSP2019 Solution - Part A

• 至于为什么是 \(\text{Part A}\) 而不是 \(\text{Day 1}\)
• 那是因为 Day1 T3 还没改
• （那这六题的 \(\text{solution}\) 就按难度顺序写吧）
• 感觉今年的画风和 \(\text{NOIP 2016}\) 有点像？

D1T1 code

Solution

• 直接模拟
• 如果 \(k<2^{n-1}\) 就输出 \(0\)
• 否则输出 \(1\) 并把 \(k\) 变成 \(2^n-1-k\)
• 然后 \(n\) 减掉 \(1\) 继续进行下去直到 \(n=0\) 为止
• 注意 \(k\) 要开 \(\text{unsigned long long}\)
• \(O(n)\)

Code

#include <bits/stdc++.h>

typedef unsigned long long ull;

int n;
ull k;

void solve(int n, ull k)
{
	if (!n) return;
	ull mid = 1ull << n - 1;
	if (k < mid) putchar('0'), solve(n - 1, k);
	else putchar('1'), solve(n - 1, mid - 1 - (k - mid));
}

int main()
{
	std::cin >> n >> k;
	solve(n, k);
	return puts(""), 0;
}

D1T2 brackets

Solution

• 令 \(cnt_u\) 表示根到 \(u\) 的路径组成的括号序列，以 \(u\) 为右端点的合法括号序列个数
• 那么 \(k_u\) 就等于根到 \(u\) 的路径上所有点的 \(cnt\) 之和
• 易得如果存在 \(u\) 的一个深度最大的祖先 \(v\) 使得 \(v\) 到 \(u\) 的路径组成的括号序列是合法括号序列
• 那么 \(cnt_u=cnt_{fa_v}+1\)
• 对于求这个 \(v\) ，可以维护一个栈
• 从根到 \(u\) ，如果是左括号则直接加入，如果是右括号且栈不空则弹栈
• 那么如果 \(u\) 为右括号，那么 \(v\) 为这次弹出的括号对应的点
• 而对于求出所有的 \(u\) ，可以在对树 \(\text{DFS}\) 的过程中维护这个栈，在 \(\text{DFS}\) 回溯时把栈操作也退回即可
• \(O(n)\)

Code

#include <bits/stdc++.h>

template <class T>
inline void read(T &res)
{
	res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	if (bo) res = ~res + 1;
}

typedef long long ll;

const int N = 5e5 + 5;

int n, fa[N], ecnt, nxt[N], adj[N], go[N], stk[N], top, cnt[N];
char s[N];
ll sum[N], ans;

void add_edge(int u, int v)
{
	nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v;
}

void dfs(int u)
{
	int tf = 0;
	if (s[u] == '(') stk[++top] = u;
	else if (top) cnt[u] = cnt[fa[tf = stk[top--]]] + 1;
	sum[u] = sum[fa[u]] + cnt[u];
	ans ^= sum[u] * u;
	for (int e = adj[u], v = go[e]; e; e = nxt[e], v = go[e])
		dfs(v);
	if (s[u] == '(') top--;
	else if (tf) stk[++top] = tf;
}

int main()
{
	int x;
	read(n);
	scanf("%s", s + 1); n = strlen(s + 1);
	for (int i = 2; i <= n; i++) read(x), add_edge(fa[i] = x, i);
	dfs(1);
	return std::cout << ans << std::endl, 0;
}

D2T1 meal

Solution

• 如果没有一半的限制，那么答案为
• \[\prod_{i=1}^n(1+\sum_{j=1}^ma_{i,j})-1
  \]

• 而出现次数超过一半的主要食材最多 \(1\) 种
• 故可以枚举超过一半的主要食材是哪种，并把对应的方案数从上式种扣掉即可
• 假设确定了一种食材 \(x\) ，考虑如何求这种食材出现超过一半的方案数
• 设 \(u_i=a_{i,x}\) ， \(v_i=\sum_{j=1,j\ne x}^ma_{i,j}\)
• 问题就转化成了有 \(n\) 个变量，对于第 \(i\) 个变量有 \(u_i\) 种方法使其为 \(1\) ， \(v_i\) 种方法使其为 \(-1\) ，\(1\) 种方法使其为 \(0\)，求有多少种方案使得 \(1\) 的个数严格大于 \(-1\) （所有变量的和严格大于 \(0\) ）
• 于是可以 \(\text{DP}\) ：设 \(f[i][j]\) 表示前 \(i\) 个变量和为 \(j\) 的方案数（\(j\) 可以为负），转移时枚举下一个变量的取值
• \(O(mn^2)\)

Code

#include <bits/stdc++.h>

template <class T>
inline void read(T &res)
{
	res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	if (bo) res = ~res + 1;
}

const int N = 105, E = 205, M = 2005, rqy = 998244353;

int n, m, a[N][M], f[N][E], sum[N], tmp[N], ans = 1;

inline void add(int &a, const int &b)
{
	a += b; if (a >= rqy) a -= rqy;
}

inline void sub(int &a, const int &b)
{
	a -= b; if (a < 0) a += rqy;
}

int main()
{
	read(n); read(m);
	for (int i = 1; i <= n; i++) sum[i] = 1;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++)
			read(a[i][j]), add(sum[i], a[i][j]);
	for (int i = 1; i <= n; i++) ans = 1ll * ans * sum[i] % rqy;
	sub(ans, 1);
	for (int i = 1; i <= m; i++)
	{
		for (int j = 1; j <= n; j++) tmp[j] = sum[j], sub(tmp[j], a[j][i]), sub(tmp[j], 1);
		for (int j = -n; j <= n; j++)
			for (int k = 0; k <= n; k++)
				f[k][j + n] = 0;
		f[0][n] = 1;
		for (int j = 1; j <= n; j++)
			for (int k = -n; k <= n; k++)
			{
				add(f[j][k + n], f[j - 1][k + n]);
				if (k > -n) add(f[j][k + n], 1ll * f[j - 1][k - 1 + n] * a[j][i] % rqy);
				if (k < n) add(f[j][k + n], 1ll * f[j - 1][k + 1 + n] * tmp[j] % rqy);
			}
		for (int j = 1; j <= n; j++) sub(ans, f[n][j + n]);
	}
	return std::cout << ans << std::endl, 0;
}