1. 程式人生 > >BZOJ 1555 KD之死 貪心 堆

BZOJ 1555 KD之死 貪心 堆

題意

  • 給你 n n 個盒子,每個盒子有重量 w w 和可以承受的最大重量 t
    t
    兩個屬性,有些盒子是必選的,你現在要在把所有的必選的盒子選定的基礎上,使選擇的盒子最多,最開始你有一輛能承重 v v 的車,如果不能選完必選的盒子就輸出 F o
    o l i s h   S D !
    Foolish \ SD!

首先對於兩個盒子 a a b b

如果 a a 放在 b b 上方 那麼承重為 b t a w b_t-a_w

如果 b b 放在 a a 上方 那麼承重為 a t b w a_t-b_w

如果 a a 放在 b b 上方更優 那麼 b t a w > a t b w b_t-a_w>a_t-b_w

a t + a w < b t + b w a_t+a_w<b_t+b_w

那麼我們可以貪心地按 a t + a w a_t+a_w 從小到大來從上到下確定盒子的順序

因為從上到下每個物品上面的重量和是定值 選擇沒有後效性

用一個堆來維護不是必選的盒子中質量最大值

每次加入一個必選的盒子一直刪除堆中的元素

直到剩餘重量可以放在這個盒子上 或者堆為空

如果加入一個不是必選的 能放就直接放

不能放的話比較它和堆頂元素誰更優即可

對於車我們把它當作一個必選的盒子插在拍完序的序列後面就可以了

複雜度 O ( n log n ) O(n\log n)

Codes

#include <bits/stdc++.h>

using namespace std;

const int N = 6e5 + 10;

int n, m, maxv, Sumw, ans;

struct node {
	int w, t, mst; 
	bool operator < (const node &T) const {
		return w + t < T.w + T.t;
	}
}A[N];

priority_queue<int> q; 

int main() {
#ifndef ONLINE_JUDGE
	freopen("1555.in", "r", stdin);
	freopen("1555.out", "w", stdout);
#endif

	scanf("%d%d%d", &n, &m, &maxv);
	for (int i = 1; i <= n; ++ i)
		scanf("%d%d", &A[i].w, &A[i].t);
	for (int i = 1, x; i <= m; ++ i) 
		scanf("%d", &x), A[x].mst = 1; 
	sort(A + 1, A + n + 1);
	A[++ n] = (node){0, maxv, 1};
	
	for (int i = 1; i <= n; ++ i) {
		if (A[i].mst) {
			while (Sumw > A[i].t) {
				if (q.empty()) return puts("Foolish SD!"), 0;
				Sumw -= q.top(), q.pop(), -- ans;
			}
			Sumw += A[i].w;
		}
		else {
			if (Sumw > A[i].t) {
				if (!q.empty() && Sumw - q.top() <= A[i].t && A[i].w < q.top())
					Sumw -= q.top(), q.pop(), q.push(A[i].w), Sumw += A[i].w;
				continue;
			}
			Sumw += A[i].w, q.push(A[i].w);
		}
		++ ans;
	}

	printf("%d\n", ans - 1);

	return 0;
}