CFR-697D Cleaning the Phone 题解

题目大意

有n个物品，每个物品有自己的价值$a_i$，以及花费$b_i=1,2$，现在你需要至少m价值的物品，求最小花费。

解题思路

刚开始想的是贪心。

把花费不同的物品放到不同的vector里面。

然后不断进行比对。

结果发现样例都过不去。。

看了下题解，发现这还是老题型了。

先求前缀和，然后枚举一类物品的数量，然后二分搜索另一类物品的数量。

枚举之前需要排序。

完整代码

#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <queue>
#include <deque>
#include <cstring>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <list>
#include <string>
#include <vector>
#include <algorithm>
#include <sstream>
#include <unordered_map>
using namespace std;
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=a;i>=n;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define pll pair<ll,ll>
#define pii pair<int,int>
#define bg begin
#define rbg rbegin
#define ed end
#define dbg(x) cout << #x << "===" << x << endl
typedef double db;
typedef long long ll;
typedef unsigned long long ull;

const int N = 2e5 + 5;
ll num[N];
vector<ll> v1, v2;

int gcd(int a, int b) {
	return !b ? a : gcd(b, a % b);
}

ll qpow(ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1)	ans *= a;
		a <<= 1, b >>= 1;
	}
	return ans;
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	//freopen("D:\\测试用\\in.txt", "r", stdin);
	//freopen("D:\\测试用\\out.txt", "w+", stdout);
	int t; cin >> t;
	while (t--) {
		int n, m; cin >> n >> m;
		ll sum = 0;
		v1.clear(); v2.clear();
		rep(i, 1, n)	cin >> num[i], sum += num[i];
		v1.push_back(0); v2.push_back(0);
		rep(i, 1, n) {
			int x; cin >> x;
			if (x == 1)	v1.push_back(num[i]);
			else v2.push_back(num[i]);
		}
		sort(v1.begin() + 1, v1.end(), greater<ll>());
		sort(v2.begin() + 1, v2.end(), greater<ll>());
		int l1 = v1.size() - 1, l2 = v2.size() - 1;
		rep(i, 1, l1) v1[i] += v1[i - 1];
		rep(i, 1, l2) v2[i] += v2[i - 1];
		int pos;
		ll ans = 1e18;
		rep(i, 0, l1) {
			pos = lower_bound(v2.begin(), v2.end(), m - v1[i]) - v2.begin();
			if (pos != v2.size())	ans = min(ans, 1LL*i + pos * 2);
		}
		if (sum < m)	cout << -1 << endl;
		else cout << ans << endl;
	}
	return 0;
}