PAT 1020 月饼

PAT 1020 月饼

#include <bits/stdc++.h>

using namespace std;

const int N = 1e3 + 10;
// 注意：题目说了库存量和总价是整数，并没有说是正整数，所以要小心！ 
struct moon {
	double stock;  // 库存量
	double total;  // 总价
	double avg;    // 单价
} m[N];

bool cmp(moon a, moon b) {
	return a.avg > b.avg;
}

int main() {

	int n, need;
	cin >> n >> need;
	for (int i = 0; i < n; i++) {
		cin >> m[i].stock;
	} 
	for (int i = 0; i < n; i++) {
		cin >> m[i].total;
		m[i].avg = m[i].total / m[i].stock; 
	}
	
	sort(m, m + n, cmp);
	
	double ans = 0;
	int i = 0;
	for (int i = 0; i < n; i++) {
		if (need >= m[i].stock) {
			ans += m[i].total;
			need -= m[i].stock;
		} else {
			ans += m[i].avg * need;
			break;
		}
	}	
	printf("%.2f", ans);

	return 0;
}