uoj#P1001. 【IOI2025】节日游戏
【IOI2025】节日游戏
节日上 Nayra 正在玩一个游戏,大奖是一次去红湖(Laguna Colorada)的旅行。游戏中玩家使用代币购买礼券。每购买一张礼券都有可能会获得额外的代币。游戏的目标是获得尽可能多的礼券。
开始时她有 $A$ 枚代币。游戏中一共有 $N$ 张礼券,从 $0$ 到 $N-1$ 编号。Nayra 需要支付 $P[i]$ 枚代币($0 \leq i < N$)来购买礼券 $i$(购买前她至少要有 $P[i]$ 枚代币)。每张礼券最多只能购买一次。
此外,每张礼券 $i$($0 \leq i < N$)都指定了类型,记为 $T[i]$,其值为 $1$ 到 $4$ 之间的整数。当 Nayra 购买礼券 $i$ 后,她剩余的代币数量将乘以 $T[i]$。形式化地,如果她在游戏的某个时刻有 $X$ 枚代币,并购买了礼券 $i$(要求 $X \geq P[i]$),那么购买后她将有 $(X - P[i]) \cdot T[i]$ 枚代币。
你的任务是确定 Nayra 应该购买哪些礼券以及按什么顺序来购买,使她最终拥有的礼券数量最大化。如果有多种购买序列能达成该目标,你可以回答其中任意一种。
实现细节
你要实现以下函数:
std::vector<int> max_coupons(int A, std::vector<int> P, std::vector<int> T)
- $A$: Narya 初始拥有的代币数量。
- $P$: 长度为 $N$ 的数组,表示礼券的价格。
- $T$: 长度为 $N$ 的数组,表示礼券的类型。
- 对每个测试用例,该函数恰好被调用一次。
该函数应返回一个数组 $R$,按以下规则表示 Narya 的购买计划:
- 数组 $R$ 的长度应等于她最多可以购买的礼券数量。
- 数组中的元素为她购买的礼券编号,按购买的顺序排列。也就是说,她首先购买礼券 $R[0]$,然后购买礼券 $R[1]$,以此类推。
- $R$ 中所有的元素互不相同。
如果无法购买任何礼券,则 $R$ 应为空数组。
输入格式
输入格式:
N A P[0] T[0] P[1] T[1] ... P[N-1] T[N-1]
输出格式
输出格式:
S R[0] R[1] ... R[S-1]
其中,$S$ 是 max_coupons 返回的数组 $R$ 的长度。
输入输出样例 #1
输入 #1
4 13 4 1 500 3 8 3 14 4
输出 #1
3 2 3 0
输入输出样例 #2
输入 #2
2 9 6 2 5 3
输出 #2
2 0 1
输入输出样例 #3
输入 #3
3 1 2 4 5 3 7 1
输出 #3
0
说明/提示
例 1
考虑以下调用。
max_coupons(13, [4, 500, 8, 14], [1, 3, 3, 4])
Narya 起初有 $A = 13$ 枚代币。她可以按以下顺序购买 $3$ 张礼券:
| 购买的礼券 | 礼券价格 | 礼券类型 | 购买后的代币数量 |
|---|---|---|---|
| $2$ | $8$ | $3$ | $(13 - 8) \cdot 3 = 15$ |
| $3$ | $14$ | $4$ | $(15 - 14) \cdot 4 = 4$ |
| $0$ | $4$ | $1$ | $(4 - 4) \cdot 1 = 0$ |
在这个例子中,Narya 不可能购买多于 $3$ 张的礼券,并且上述购买顺序是她购买这 $3$ 张礼券的唯一方式。因此,该函数应返回 $[2, 3, 0]$。
例 2
考虑以下调用。
max_coupons(9, [6, 5], [2, 3])
在这个例子中,Narya 可以以任意顺序购买两张礼券。因此,该函数可以返回 $[0, 1]$ 或 $[1,0]$。
例 3
考虑以下调用。
max_coupons(1, [2, 5, 7], [4, 3, 1])
在这个例子中,Narya 有 $1$ 枚代币,不足以购买任何一张礼券。因此,该函数应返回 $[\ ]$ (空数组)。
约束条件
- $1 \leq N \leq 200\,000$
- $1 \leq A \leq 10^{9}$
- 对每个满足 $0 \leq i < N$ 的 $i$,都有 $1 \leq P[i] \leq 10^{9}$。
- 对每个满足 $0 \leq i < N$ 的 $i$,都有 $1 \leq T[i] \leq 4$。
子任务
| 子任务 | 分数 | 额外的约束条件 |
|---|---|---|
| 1 | $5$ | 对每个满足 $0 \leq i < N$ 的 $i$,都有 $T[i] = 1$。 |
| 2 | $7$ | $N \leq 3000$;对每个满足 $0 \leq i < N$ 的 $i$,都有 $T[i] \leq 2$。 |
| 3 | $12$ | 对每个满足 $0 \leq i < N$ 的 $i$,都有 $T[i] \leq 2$。 |
| 4 | $15$ | $N \leq 70$ |
| 5 | $27$ | Nayra 可以购买所有 $N$ 张礼券(以某种顺序)。 |
| 6 | $16$ | 对每个满足 $0 \leq i < N$ 的 $i$,都有 $(A - P[i]) \cdot T[i] < A$。 |
| 7 | $18$ | 没有额外的约束条件。 |