762B - USB vs PS2 - CodeForces Solution

// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

int main() {
	int usb, ps2, both; cin >> usb >> ps2 >> both;
	int m; cin >> m;
	long long um[300000]; memset(um, 0, sizeof(um));
	long long pm[300000]; memset(pm, 0, sizeof(pm));
	int umNum = 0, pmNum = 0;
	for (int i = 0; i < m; i++) {
		long long p; string kind;
		cin >> p >> kind;
		if (kind == "USB") {
			um[umNum++] = p;
		} else {
			pm[pmNum++] = p;
		}
	}
	sort(um, um+umNum);
	sort(pm, pm+pmNum);
	int umInd = 0, pmInd = 0;
	int count = 0;
	long long cost = 0;
	while (umInd < min(umNum, usb)) {
		cost += um[umInd++];
		count++;
	}
	while (pmInd < min(pmNum, ps2)) {
		cost += pm[pmInd++];
		count++;
	}
	while ((pmInd < pmNum || umInd < umNum) && both > 0) {
		if (pmInd < pmNum && umInd < umNum) {
			if (um[umInd] < pm[pmInd]) {
				cost += um[umInd++];
			} else {
				cost += pm[pmInd++];
			}
			count++;
			both--;
			continue;
		}
		if (pmInd < pmNum) {
			cost += pm[pmInd++];
		} else {
			cost += um[umInd++];
		}
		count++;
		both--;
	}
	cout << count << " " << cost << endl;
}