0x17 二叉堆

目录

poj1456

题目链接

题目

题解

代码

poj2422

题目链接

题目

题解

代码

---

poj1456

题目链接

　　http://poj.org/problem?id=1456

题目

　　有n个商品，每个商品有利润p和过期时间d，每天只能卖一个商品。过期的商品不能再卖，问如何安排每天卖的商品。

题解

　　首先，用一个小根堆来维护所有商品的利润p。

1. 将所有商品按过期时间排序；
2. 从头到尾扫描商品，如果该商品的过期时间t=堆的大小，则比较它与堆顶的大小，如果它大于堆顶的话，那么堆顶弹出，再将该商品加入堆；如果该商品的过期时间>堆的大小，直接push进去即可；（为什么不讨论<的情况呢，因为根本不会小于）
3. 对堆里所有元素求和，结果就是答案。

注意：priority_queue默认的是大根堆！！！！！！！！！

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <cmath>
using namespace std;

#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8
#define next next_
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x7fffffff;
const int inf = 0x3f3f3f3f;
const int maxn = 200005;
const int N = 105;

inline void read(int &x) {
    int f=1;x=0;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    x*=f;
}

/*inline void caltime(int tt) {
	tt = clock() - tt;
    cerr << (double)tt/CLOCKS_PER_SEC << " seconds!" << endl;
}*/

inline void print_matrix(int a[N][N],int n,int m,bool flag=0) {
	if(flag) {
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++)
				cout<<a[i][j]<<" ";
			cout<<endl;
		}
	}else {
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++)
				cout<<a[i][j]<<" ";
			cout<<endl;
		}
	}
}

inline void print_line(int *a,int n,bool flag=0) {
	if(flag) {
		for(int i=1;i<=n;i++)
			cout<<a[i]<<" ";
	}else {
		for(int i=0;i<n;i++)
			cout<<a[i]<<" ";
	}
	cout<<endl;
}

int n;
struct node{
	int p,d;
	bool operator<(const node&b)const{
		return d<b.d;
	}
}a[maxn];
priority_queue<int,vector<int>,greater<int> > q; //优先队列默认是大根堆！！！ 

ll get_sum() {
	ll res = 0;
	while(!q.empty()) {
		res += q.top();
		q.pop();
	}	
	return res;
}

int main()
{
	while(cin>>n) {
		for(int i=0;i<n;i++) {
			read(a[i].p),read(a[i].d);
		}
		sort(a,a+n);
		for(int i=0;i<n;i++) {
			if(a[i].d>q.size()) {
				q.push(a[i].p);
			}else if(a[i].d==q.size()) {
				if(q.top()<a[i].p) {
					q.pop();
					q.push(a[i].p);
				}
			}
		}
		ll ans = get_sum();
		cout<<ans<<endl;
	}
	return 0;
}

---

poj2422

题目链接

　　http://poj.org/problem?id=1456

题目

　　给定M个长度为N的序列，从每个序列中任取一个数求和，可以构成个和，求其中最小的N个和。

题解

　　首先将Ｍ组个数全部排序；

　　假设M=2，那么最小值一定是ａ[0]+b[0], 次小值就是min(a[0]+b[1],a[1]+b[0]) ,以此类推。

　　这样我们每次就有了两个扩展方案，但是可能会出现重复，比如a[1]+b[1]可以同时由a[0]+b[1]和a[1]+b[0]扩展而来；

　　于是就需要加上标记，如果之前扩展过b，那么下一次只能继续扩展b（可以模拟一下）；

　　这一步我们通过一个三元组(i,j,flag)来得到，其中flag表示之前是否扩展过b。

　　现在扩展到M组，扩展完前两组后，把新得到的N个数作为一组，和第三组去比较，依次类推，总共会生成M-1次。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <cmath>
using namespace std;

#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8
#define next next_
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x7fffffff;
const int inf = 0x3f3f3f3f;
const int maxn = 3005;
const int N = 105;

inline void read(int &x) {
    int f=1;x=0;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    x*=f;
}

/*inline void caltime(int tt) {
	tt = clock() - tt;
    cerr << (double)tt/CLOCKS_PER_SEC << " seconds!" << endl;
}*/

inline void print_matrix(int a[N][N],int n,int m,bool flag=0) {
	if(flag) {
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++)
				cout<<a[i][j]<<" ";
			cout<<endl;
		}
	}else {
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++)
				cout<<a[i][j]<<" ";
			cout<<endl;
		}
	}
}

inline void print_line(int *a,int n,bool flag=0) {
	if(flag) {
		for(int i=1;i<=n;i++)
			cout<<a[i]<<" ";
	}else {
		for(int i=0;i<n;i++)
			cout<<a[i]<<" ";
	}
	cout<<endl;
}

int t,n,m,f[maxn];
vector<int> e[maxn]; 
struct node {
	int i,j,sum;
	bool flag;
	bool operator<(const node&b) const{
		return sum>b.sum;
	}
};
priority_queue<node> q;

void solve(int k) {
	int cnt=0;
	while(!q.empty()) {
		node temp = q.top();
		q.pop();
		f[cnt++] =  temp.sum;
		if(temp.j+1<n) q.push({temp.i,temp.j+1,e[k-1][temp.i]+e[k][temp.j+1],1}); //注意不能超出范围
		if(!temp.flag&&temp.i+1<n)	q.push({temp.i+1,temp.j,e[k-1][temp.i+1]+e[k][temp.j],0});	
		if(cnt==n)	break;
	}
	for(int i=0;i<n;i++) {
		e[k][i] = f[i];
	}
}

void init() {while(!q.empty())	q.pop();}

int main() 
{
	cin>>t;
	while(t--) {
		read(m),read(n);
		int x;
		for(int i=0;i<m;i++) {
			e[i].clear();
			for(int j=0;j<n;j++) {
				read(x);
				e[i].push_back(x);
			}
			sort(e[i].begin(),e[i].end());
		}		
		for(int i=0;i<n;i++)	f[i] = e[0][i];		
		for(int i=1;i<m;i++) {
			init();
			q.push({0,0,e[i-1][0]+e[i][0],0});
			solve(i);
		}
		for(int i=0;i<n;i++)	{
			if(i==n-1)	cout<<f[i];
			else cout<<f[i]<<" ";
		}
		cout<<endl;
	}
	return 0;
}