UOJ Logo Universal Online Judge

UOJ

#899. 【IOI2024】尼罗河船运

附件下载 统计

你想通过尼罗河来运输 N 件手工艺品。这些手工艺品从 0N1 编号。第 i0i<N)件手工艺品的重量是 W[i]

为了运输这些手工艺品,你使用了特制的船。每艘船最多可以运两件手工艺品。

  • 如果你决定将单件手工艺品放在一艘船上,那么这件手工艺品的重量可以是任意的。
  • 如果你想把两件手工艺品一起放在同一艘船上,你必须保证这艘船的平衡。具体来说,如果手工艺品 pq0p<q<N)的重量差的绝对值不超过 D,即满足 |W[p]W[q]|D,那么你可以将它们一起放在同一艘船上。

你必须付费来运一件手工艺品,其运费取决于同一艘船上所运载的手工艺品数量。手工艺品 i0i<N)的运费是:

  • A[i],如果你把手工艺品 i 单独放在船上,或者
  • B[i],如果你把手工艺品 i 和另一件手工艺品一起放在船上。

注意在第二种情况中,你要为船上两件手工艺品都支付运费。具体来说,如果你决定用同一艘船运输手工艺品 pq0p<q<N),你需要支付 B[p]+B[q]

一件手工艺品单独用一艘船运输的费用,总是比与其他手工艺品合用一艘船时的费用要高,所以对任意满足 0i<Ni,都有 B[i]<A[i]

麻烦的是,由于尼罗河变化莫测,导致 D 的值经常改变。你的任务是回答 Q 个问题,从 0Q1 编号。这些问题用一个长度为 Q 的数组 E 来描述。问题 j0j<Q)的答案,是在 D 的值等于 E[j] 时运输所有 N 件手工艺品的最小总代价。

实现细节

你需要实现以下函数。

std::vector<long long> calculate_costs(
    std::vector<int> W, std::vector<int> A, 
    std::vector<int> B, std::vector<int> E)
  • WAB:长度均为 N 的整数数组,分别给出手工艺品的重量和运费。
  • E:长度为 Q 的整数数组,给出每个问题中的 D 值。
  • 该函数应该返回一个包含 Q 个整数的数组 R,给出运输手工艺品的最小总代价,其中 R[j] 对应 D 等于 E[j](对每个满足 0j<Qj)时的运费。
  • 对于每个测试用例,该函数恰好被调用一次。

约束条件

  • 1N100000
  • 1Q100000
  • 对每个满足 0i<Ni,都有 1W[i]109
  • 对每个满足 0i<Ni,都有 1B[i]<A[i]109
  • 对每个满足 0j<Qj,都有 1E[j]109

子任务

子任务 分数 额外的约束条件
1 6 Q5N2000;对每个满足 0i<Ni,都有 W[i]=1
2 13 Q5;对每个满足 0i<Ni,都有 W[i]=i+1
3 17 Q5;对每个满足 0i<Ni,都有 A[i]=2B[i]=1
4 11 Q5N2000
5 20 Q5
6 15 对每个满足 0i<Ni,都有 A[i]=2B[i]=1
7 18 没有额外的约束条件。

例子

考虑以下调用。

calculate_costs([15, 12, 2, 10, 21],
                [5, 4, 5, 6, 3],
                [1, 2, 2, 3, 2],
                [5, 9, 1])

在该例子中,我们有 N=5 件手工艺品和 Q=3 个问题。

在第一个问题中,D=5。你可以把手工艺品 0 和手工艺品 3 放在同一艘船上(因为 |1510|5),而其他手工艺品都各自放在不同的船上。这使得运输所有手工艺品的总代价最小,即 1+4+5+3+3=16

在第二个问题中,D=9。你可以把手工艺品 0 和手工艺品 1 放在同一艘船上(因为 |1512|9),而把手工艺品 2 和手工艺品 3 放在同一艘船上(因为 |210|9)。剩下的手工艺品单独用一艘船运输。这使得运输所有手工艺品的总代价最小,即 1+2+2+3+3=11

在最后一个问题中,D=1。你需要把每件手工艺品都单独用一艘船运输。这使得运输所有手工艺品的总代价最小,即 5+4+5+6+3=23

因此,该函数应该返回 [16,11,23]

评测程序示例

输入格式:

N
W[0] A[0] B[0]
W[1] A[1] B[1]
...
W[N-1] A[N-1] B[N-1]
Q
E[0]
E[1]
...
E[Q-1]

输出格式:

R[0]
R[1]
...
R[S-1]

这里,Scalculate_costs 所返回的数组 R 的长度。

时间限制:2s

空间限制:2048MB