洛谷 P2240 部分背包问题(贪心)

题目链接
洛谷P2240

题目描述
阿里巴巴走进了装满宝藏的藏宝洞。藏宝洞里面有 N(N≤100) 堆金币,第 i 堆金币的总重量和总价值分别是 mi,vi(1≤mi,vi≤100)。阿里巴巴有一个承重量为T(T≤1000) 的背包,但并没办法将全部的金币都装进去。他想装走尽可能多价值的金币。所有金币都可以随意分割,分割完的金币重量价值比(也就是单位价格)不变。请问阿里巴巴最多可以拿走多少价值的金币?

输入格式
第一行两个整数 N、T
接下来 N 行,每行两个整数 mi,vi

输出格式
一个整数表示答案,输出两位小数

题目分析
金币可以分割,所以可以取某一堆金币的一部分。利用贪心算法,贪心策略为选择金币单位价格最高的装入背包。所以需要根据金币单价进行快速排序,排序之后进行贪心选择。

#include 
#include 
#include 
#include 
using namespace std;

int getIndex(vector&m, vector&v, vector&w, int low, int high) {
	double wtemp = w[low], mtemp = m[low], vtemp = v[low];
	while (low < high) {
		while (low < high && w[high] >= wtemp) {
			high--;
		}
		w[low] = w[high];
		m[low] = m[high];
		v[low] = v[high];
		//low++;
		while (low < high && w[low] <= wtemp) {
			low++;
		}
		w[high] = w[low];
		m[high] = m[low];
		v[high] = v[low];
	}
	w[low] = wtemp;
	m[low] = mtemp;
	v[low] = vtemp;
	return low;
}

void quickSort(vector&m, vector&v, vector&w, int low, int high) {
	if (low < high) {
		int index = getIndex(m, v, w, low, high);

		quickSort(m, v, w, low, index - 1);
		quickSort(m, v, w, index + 1, high);
	}
}

int main()
{
	int n;
	double t,c,cv,tm,tv;
	vectorm;
	vectorv;
	vectorw;
	cin >> n >> t;
	c = 0;//能取的最大重量
	cv = 0;//背包中金币价值
	int i;
	for (i = 0; i < n; i++) {
		cin >> tm >> tv;
		m.push_back(tm);
		v.push_back(tv);
		w.push_back(v[i] / m[i]); //计算单位价格
	}
	//按单位价格快速排序
	quickSort(m, v, w, 0, n - 1);
	/*
	for (i = 0; i < n; i++) {
		cout << "m:" << m[i] << " " << "v:" << v[i] << " " << "w:" << w[i] << endl;
	}
	*/
	for (i = n - 1; i >= 0; i--) {
		if ((t - c) > 0) {//背包还有空间
			if ((t - c) >= m[i]) {//背包空间能放下当前堆所有金币
				c = c + m[i];
				cv = cv + v[i];
			}
			else {//背包放当前堆一部分金币
				cv = cv + w[i] * (t - c);
				c = t;
			}
		}
		else if (t - c == 0) break;
	}
	cout << setiosflags(ios::fixed);
	cout << setprecision(2) << cv << endl;
	//printf("%.2f", cv);
	return 0;
}

你可能感兴趣的:(洛谷 P2240 部分背包问题(贪心))