BZOJ3728 PA2014Final Zarowki

BZOJ3728 PA2014Final Zarowki

贪心题目

然而我的贪心又是错的,但是这一次至少大体思路对了

我的错误思路:

把数组分别排序,然后一一对应,如果小就换,最后如果还有机会,就在剩下的大的选择差值最大的减去

正确思路:

将\(p\)数组排序,对于每个数,都找到小于等于他的最大的数,然后把他们的差值扔到堆里

找最大的数可以利用两个数组排序之后的单调性或者直接暴力上multiset去解决

如果最后还有超过\(k\)个房间没有被匹配,说明无解

否则就换成对应功率的灯泡

有过还有剩余机会

就把堆中的对应的数抹平

#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cctype>
#include<set>
#include<vector>
#include<ctime>
#include<cmath>
#define LL long long
#define pii pair<int,int>
#define mk make_pair
#define fi first
#define se second
using namespace std;
const int N = 5e5 + 3;
int p[N];
int w[N];
multiset <int> s;
priority_queue <int> q;
int n,k;
LL ans;
inline int read(){
	int v = 0,c = 1;char ch = getchar();
	while(!isdigit(ch)){
		if(ch == '-') c = -1;
		ch = getchar();
	}
	while(isdigit(ch)){
		v = v * 10 + ch - 48;
		ch = getchar();
	}
	return v * c;
}
int main(){
	multiset<int>::iterator it;
	n = read(),k = read();
	for(int i = 1;i <= n;++i) p[i] = read();
	for(int i = 1;i <= n;++i){
		w[i] = read();
		s.insert(w[i]);	
	}
	sort(p + 1,p + n + 1);
	for(int i = 1;i <= n;++i){
		it = s.upper_bound(p[i]);
		if(it == s.begin()) continue;
		it--;
		ans += p[i];
		q.push(p[i] - *it);
		s.erase(it);
	} 
	if(s.size() > k){printf("NIE\n");return 0;}
	for(it = s.begin();it != s.end();it++){
		ans += *it;
		k--;
		q.push(0);
	}
	while(k != 0){
		ans -= q.top();
		q.pop();
		k--;	
	}
	printf("%lld\n",ans);
	return 0;
}