题目内容

某地面光伏电站按区域划分为多个子阵。每个子阵的发电能力有明确的约束：

• 发电能力上限（$max\_capacity$）
• 发电能力下限（$min\_capacity$）
• 基准发电量（$base\_generation$）

系统需要根据这些约束，预测整个电站的发电量。预测策略是：

1. 每个子阵有一个基准发电量
2. 如果总预测发电量超过电站总容量，需要按比例缩减
3. 如果总预测发电量低于总容量下限，需要按比例放大
4. 最终预测发电量向上取整为整数

调整策略

缩减场景

当总发电量超过电站容量时，按各子阵的可用缩减空间比例分配缩减量

可用缩减空间 $=$ $base\_generation$ $-$ $min\_capacity$；表示该子阵还能减少多少发电量而不低于下限

缩减量按此比例分配：$缩减量_i$ = 总缩减量 × ($可用缩减空间_i$ / 总可用缩减空间)

放大场景

当总发电量低于总下限时，按各子阵的可用放大空间比例分配增加量

可用放大空间 = $max\_capacity$ - $base\_generation$；表示该子阵还能增加多少发电量而不超过上限

增加量按此比例分配：$增加量_i$ = 总增加量 × ($可用放大空间_i$ / 总可用放大空间)

补充说明

输入

$sub\_arrays$: 子阵列表，每个子阵是一个三元组[$max\_capacity$, $min\_capacity$, $base\_generation$]，均为整数

$station\_capacity$: 电站总容量上限（整数）

输出

返回一维数组，每个元素是对应子阵的预测发电量（向上整数） 如果无法在约束范围内满足总容量要求，返回全 $0$ 数组

特殊情况处理

1. 如果所有子阵都达到下限，但仍超过电站容量：返回全 $0$ 数组
2. 如果所有子阵都达到上限，但仍低于总下限：返回全 $0$ 数组

要求

最终预测发电量必须满足：$min\_capacity \le predicted\_generation \le max\_capacity$

1. 如果总预测发电量 $>$ $station\_capacity$，需要按比例缩减
2. 如果总预测发电量 $<$ 总下限（各子阵$min\_capacity$之和），需要按比例放大
3. 最终结果必须向上取整为整数（如使用$math.ceil$）

样例1

输入

[[1000, 800, 900],[1200, 900, 1100],[800, 600, 700]],2800

输出

[900,1100,700]

样例2

输入

[[1000, 800, 800],[1200, 900, 900],[800, 600, 600]],2600

输出

[800,900,600]