打卡信奥刷题（3413）用C++实现信奥题 P10134 [USACO24JAN] Cowmpetency S

P10134 [USACO24JAN] Cowmpetency S

题目描述

Farmer John 正在为他的奶牛们雇用一位新的牛群领队。为此，他面试了 $N$（$2\le N\le 10^5$）头奶牛来担任该职位。在面试第 $i$ 个候选牛后，他会为候选牛分配一个 $1$ 到 $C$（$1\le C\le 10^9$）范围内的整数「牲任力」分数 $c_i$，与她们的领导能力相关。

由于 Farmer John 面试了如此多的奶牛，他没能记得所有奶牛的牲任力分数。然而，他确实记得 $Q$（$1\le Q<N$）对数字 $(a_j,h_j)$，其中奶牛 $h_j$ 是第一头比奶牛 $1$ 到 $a_j$ 拥有严格更高牲任力分数的奶牛（所以 $1\le a_j<h_j\le N$）。

Farmer John 现在告诉你序列 $c_1,\dots ,c_N$
（其中 $c_i=0$ 表示他忘记了奶牛 $i$ 的牲任力分数）和 $Q$ 对 $(a_j,h_j)$。帮助他求出与此信息一致的最小字典序的牲任力分数序列，或者判断不存在这样的序列！如果一个分数序列比另一个分数序列于这两个序列不同的第一个位置上的奶牛分配了更小的分数，则这个分数序列的字典序更小。

每个测试点包含 $T$（$1\le T\le 20$）个独立的测试用例。输入保证所有测试用例的 $N$ 之和不超过 $3\cdot 10^5$。

输入格式

输入的第一行包含 $T$，独立的测试用例的数量。每个测试用例描述如下：

1. 第一行包含 $N$，$Q$ 和 $C$。
2. 第二行包含序列 $c_1,\dots ,c_N$（$0\le c_i\le C$）。
3. 最后 $Q$ 行，每行包含一个数对 $(a_j,h_j)$。输入保证同一测试用例内的所有 $a_j$ 各不相同。

输出格式

对于每个测试用例，输出一行，包含与 Farmer John 记忆一致的最小字典序牲任力分数序列，或者如果这样的序列不存在时输出 $-1$。

输入输出样例 #1

输入 #1

1
7 3 5
1 0 2 3 0 4 0
1 2
3 4
4 5

输出 #1

1 2 2 3 4 4 1

输入输出样例 #2

输入 #2

5
7 6 10
0 0 0 0 0 0 0
1 2
2 3
3 4
4 5
5 6
6 7
8 4 9
0 0 0 0 1 6 0 6
1 3
6 7
4 7
2 3
2 1 1
0 0
1 2
10 4 10
1 2 0 2 1 5 8 6 0 3
4 7
1 2
5 7
3 7
10 2 8
1 0 0 0 0 5 7 0 0 0
4 6
6 9

输出 #2

1 2 3 4 5 6 7
1 1 2 6 1 6 7 6
-1
1 2 5 2 1 5 8 6 1 3
-1

说明/提示

样例解释 1

我们可以看到给定的输出满足所有 Farmer John 记得的数对。

• $\max (c_1)=1$，$c_2=2$ 且 $1<2$，故第一个数对是满足的
• $\max (c_1,c_2,c_3)=2$，$c_4=3$ 且 $2<3$，故第二个数对是满足的
• $\max (c_1,c_2,c_3,c_4)=3$，$c_5=4$ 且 $3<4$，故第三个数对是满足的

存在一些其他的序列与 Farmer John 的记忆相一致，如

$1223541$
$1223445$

然而，其中没有序列比给定的输出字典序更小。

样例解释 2

在测试用例 $3$ 中，由于 $C=1$，唯一可能的序列是

$11$

然而，在这种情况下，奶牛 $2$ 的分数并不比奶牛 $1$ 高，因此我们无法满足条件。

在测试用例 $5$ 中，$a_1$ 和 $h_1$ 告诉我们奶牛 $6$ 是第一头分数严格高于奶牛 $1$ 到 $4$ 的奶牛。因此，奶牛 $1$ 到 $6$ 的最高得分是奶牛 $6$，为 $5$。由于奶牛 $7$ 的分数为 $7$，所以奶牛 $7$ 是第一头比奶牛 $1$ 到 $6$ 得分更高的奶牛。因此，第二个陈述「奶牛 $9$ 是第一头比奶牛 $1$ 到 $6$ 得分更高的奶牛」不能成立。

测试点性质

• 测试点 $3$：$N\le 10$ 且 $Q,C\le 4$。
• 测试点 $4-8$：$N\le 1000$。
• 测试点 $9-12$：没有额外限制。

C++实现

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
struct Node { int u, v; } b[N];
int n, q, c, a[N], lst[N], dsu[N], tag[N], pre[N];
void Solve() {
	scanf("%d%d%d", &n, &q, &c);
	for(int i = 1; i <= n; ++i) scanf("%d", &a[i]), lst[i] = a[i] ? lst[i - 1] : i;
	for(int i = 1, u, v; i <= q; ++i) scanf("%d%d", &u, &v), b[i] = {u, v - 1};
	for(int i = 1; i <= n; ++i) dsu[i] = i + 1;
	for(int i = 1; i <= q; ++i) dsu[b[i].u] = b[i].v + 1, tag[b[i].v + 1] = 1;
	pre[0] = 1;
	for(int i = 1; i <= n; ++i) { // 求出每个点的 pre
		pre[i] = pre[i - 1] + tag[i];
		for(int j = i; j < dsu[i]; ++j) pre[i] = max(pre[i], a[j]);
		for(int j = i; j < dsu[i]; ++j) pre[j] = pre[i];
		i = dsu[i] - 1;
	}
	bool isOK = 1;
	for(int i = 1; i <= n; ++i) { // 根据 pre 求出 a
		if(!a[i]) a[i] = pre[i - 1] == pre[i] ? 1 : pre[i];
		else if(pre[i] != pre[i - 1] && pre[i] > a[i])  {
			if(lst[i]) a[lst[i]] = pre[i];
		}
		if(a[i] > c) isOK = 0;
		pre[i] = max({pre[i], pre[i - 1], a[i]});
		tag[i] = 0;
	}
	for(int i = 1; i <= n; ++i) pre[i] = max(pre[i - 1], a[i]);
	for(int i = 1; i <= q; ++i) 
		if(pre[b[i].u] < pre[b[i].v] || pre[b[i].u] == pre[b[i].v + 1]) isOK = 0;
	if(!isOK) printf("-1\n");
	else for(int i = 1; i <= n; ++i)
		printf("%d%c", a[i], i == n ? '\n' : ' ');
}
int main() {
	int t; scanf("%d", &t);
	while(t--) Solve();
	return 0;
}

后续

接下来我会不断用C++来实现信奥比赛中的算法题、GESP考级编程题实现、白名单赛事考题实现，记录日常的编程生活、比赛心得，感兴趣的请关注，我后续将继续分享相关内容