Codeforces 3B. Lorry

大神说考虑到数据范围不能使用背包求解,但是由于物品的大小只有两种就很好办。

一种方法是枚举大小为2的船一共占的大小,剩下的空间尽可能填1。

另一种方法是先全部填1,然后每次拿最小的两个出来检测,要是他们的和比最大的2小,那么替换即可。

 

居然 sort 反向了,交了一个WA。

#include 
using namespace std;
#define ll long long

pair t1[100005];
pair t2[100005];

bool cmp(pair &a, pair &b) {
	return a.first > b.first;
}
int main() {
	int n;
	ll v;

	while (cin >> n >> v) {
		ll cnt1 = 0;
		ll cnt2 = 0;
		for (int i = 0; i < n; i++) {
			int t, p;
			cin >> t >> p;
			if (t == 1) {
				t1[cnt1++] = { p, i + 1 };
			}
			else {
				t2[cnt2++] = { p, i + 1 };
			}
		}

		sort(t1, t1 + cnt1,cmp);
		sort(t2, t2 + cnt2,cmp);

		for (int i = 1; i < cnt1; i++) {
			t1[i].first += t1[i - 1].first;
		}
		for (int i = 1; i < cnt2; i++) {
			t2[i].first += t2[i - 1].first;
		}

		ll maxv = t1[min(cnt1 - 1, v - 1)].first;

		int p1= min(cnt1 - 1, v - 1), p2=-1;

		for (int i = 0; i < cnt2; i++) {
			ll tmpv = t2[i].first;
			ll s2 = (i + 1) * 2;
			if (s2 <= v) {
				ll s1 = v - s2;
				if (s1 > 0) 
					tmpv += t1[min(cnt1 - 1, s1 - 1)].first;

				if (tmpv > maxv) {
					maxv = tmpv;
					p1 = min(cnt1 - 1, s1 - 1);
					p2 = i;
				}
				
			}
		}

		cout << maxv << endl;
		vectorans;
		for (int i = 0; i <= p1; i++) {
			ans.push_back(t1[i].second);
		}
		for (int i = 0; i <= p2; i++) {
			ans.push_back(t2[i].second);
		}

		sort(ans.begin(), ans.end());

		for (auto i : ans) {
			cout << i <<" ";
		}
		cout << endl;
	}
}

 

你可能感兴趣的:(Codeforces 3B. Lorry)