B_station解题报告(sgu148 枚举+堆).docVIP

试题翻译 SGU148 B-Station 试题描述 在离著名的国家Berland不远的地方，有一个水下工作站。这个工作站有N层。已知：是第i层装有Wi的水，最多可以容纳Li的水，恐怖分子炸毁第i层的代价是Pi。第i层一旦被炸毁，该层所有的水都将倾泻到第i+1层。如果某一层的水量超过了它的容量(即Li)，那么该层就将自动被毁坏，所有的水也会倾泻到下一层。 Pivland的恐怖分子想要用最少的钱毁掉第N层，现在他雇佣你来计算，需要炸毁哪些层。 输入 第一行有一个自然数N(1=n=15000)。接下来的N行，每行3个整数Wi, Li, Pi(0=Wi,Li,Pi=15000)。 输出 输出需要炸毁的层的编号。 样例 Input 3 1000 1000 1 0 1000 2 2 10 100 Output 1 2 解题报告 B-Station解题报告 【题意分析】 令Si=W1+W2+…+Wi（特别的S0=0）。 不妨设恐怖分子炸毁的第高层是第p层（第一层是最高层，第N层是最底层）。 因为恐怖分子的目标是毁灭第N层，所以水必须从第p层一直泻下去。如果存在一个i，满足Wp+Wp+1+…+Wi-1+Wi=Li，也就是说前面几层的水全部泄下来也无法把第i层自动冲毁，那么就必须要使用炸药把它炸开了。 所以恐怖分子需要炸毁的层的集合就是Bomb[p]={p}∪{i |ip Si-Sp-1=Li} 令Qi=Si-Li，那么： Bomb[p]={p}∪{i |ip Qi=Sp-1} 我们枚举p，然后看哪些层需要炸毁。这样就得到了一个O(n2)的算法。 【设计算法】 n=15000，O(n2)的算法效率太低。 注意到Sp是随着p的增加而递增的，观察两个集合： Bomb[p]={p}∪{i |ip Qi=Sp-1} Bomb[p-1]={p-1}∪{i |ip-1 Qi=Sp-2} 因为Sp-2=Sp-1，所以{i |ip-1 Qi=Sp-2}Bomb[p]。 也就是说，Bomb[p-1]是在Bomb[p]的基础上，通过以下操作得到的： 删除所有的i∈Bomb[p]，QiSp-2。 添加p-1。 针对这两种操作，我们可以使用大根堆(Heap)。从大到小枚举p，对于每一个p，执行： Step1. 如果堆的根i满足Qi Sp-2，那么删除根；否则转Step3。 Step2. 转Step1。 Step3. 添加p-1。 每层至多进堆一次、出堆一次，所以总的时间复杂度是O(nlogn)。对于n=15000的范围完全能够胜任了。